跳到正文
arXiv · cs.RO· Fenglong Song·· 3 天前精选AI 评分70

CuPIQP:面向机器人学习和控制的GPU批处理可二次规划求解器

CuPIQP: A Batched and Differentiable QP Solver on GPUs for Robot Learning and Control

AI 导读

Fenglong Song提出CuPIQP,一种基于近端内点法的GPU批处理可微分二次规划求解器,支持密集、稀疏和多阶段问题。基准测试显示,每批次最多16384个QPs的求解速度比Moreau快18.8倍,比八核PIQP快4.26倍。在端到端学习任务中,CuPIQP比QPAX加速4.6倍,比Moreau加速7.8倍,为基于模型优化的机器人学习和控制提供高效可重用基础设施。

推荐理由

通过GPU批处理和可微分优化,将机器人控制求解速度提升18.8倍,为大规模机器人学习提供高效基础设施。

来源:arXiv · cs.RO · arxiv.org