arXiv:2606.24039cs.ROcs.LG2026-06被引 1

GPU加速的可微模型预测控制,速度提升58倍

TurboMPC: Fast, Scalable, and Differentiable Model Predictive Control on the GPU

论文配图:TurboMPC: Fast, Scalable, and Differentiable Model Predictive Control on the GPU
图 1 · 摘自论文原文
  • GPU全端运行,结合SQP与ADMM求解器
  • 实测比现有最优解算器快58倍,支持8000个时间点规划
  • 适合机器人实时控制、强化学习与参数调优场景

机器人系统越来越多依赖GPU进行并行仿真、大规模学习和神经网络推理。为使模型预测控制(MPC)与这一趋势同步,求解器必须在该硬件上高效运行,同时保持快速、可微分,并兼容机器人领域常用的复杂MPC形式。我们提出TurboMPC,一个完全在GPU上运行的可微分MPC求解器,支持状态与控制不等式约束、隐式积分器、跨时间耦合代价函数以及松弛变量。TurboMPC结合序列二次规划(SQP)、ADMM内层求解器、隐式微分和定制化的JAX-CUDA实现,在仿真中验证了其在约束规划、人形机器人模仿学习及基于神经网络代价函数的强化学习任务中的有效性,相较当前最优的CPU与GPU可微求解器分别实现了最高15倍和58倍的速度提升。我们在全尺寸汽车上部署TurboMPC进行最短时间竞速,发现通过贝叶斯优化批量调优MPC参数可显著提升驾驶速度,优于人工调参基线。TurboMPC还能扩展至超过8000个节点的规划时域,同时保持车辆控制能力。项目已开源:https://github.com/ToyotaResearchInstitute/turbompc

原文摘要 · Abstract (English)

Robotics increasingly relies on GPUs for parallel simulation, large-scale learning, and neural-network inference. For model predictive control (MPC) to scale with this paradigm, solvers must run efficiently on this hardware while remaining fast, differentiable, and compatible with expressive MPC formulations used in robotics. We present TurboMPC, a differentiable MPC solver that runs entirely on the GPU and supports state and control inequality constraints, implicit integrators, cross-time-coupled costs, and slack variables. TurboMPC combines sequential quadratic programming (SQP), an alternating direction method of multipliers (ADMM) inner solver, implicit differentiation, and a co-designed JAX-CUDA implementation for efficiency and ease of use. In simulation, we validate TurboMPC on constrained planning, humanoid imitation learning, and reinforcement learning with neural-network cost function tasks, achieving up to $15\times$ and $58\times$ speedups over state-of-the-art CPU and GPU differentiable solvers, respectively. We deploy TurboMPC on a full-scale car for minimum-time racing and find that batched, GPU-accelerated tuning of MPC parameters via Bayesian optimization yields significantly faster driving than a hand-tuned baseline. TurboMPC also scales to planning horizons of over $8000$ knot points while maintaining control of the vehicle. We open-source TurboMPC at: https://github.com/ToyotaResearchInstitute/turbompc

模型预测控制GPU加速可微分优化机器人控制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。