用WASP导数替代有限差分,让机器人控制更快更稳
Robust and Efficient MuJoCo-based Model Predictive Control via Web of Affine Spaces Derivatives
- 用WASP导数复用历史计算信息,加速梯度求解
- 在高自由度系统中实现最高2倍速度提升
- 适合需要实时响应的复杂机器人控制任务
MuJoCo 是广泛应用于机器人领域的高效物理仿真器,常通过模型预测控制(MPC)实现实时响应。现有开源库 MJPC 依赖有限差分(FD)计算导数,成为高自由度或复杂场景下的性能瓶颈。本文提出将最近发展的 Web of Affine Spaces(WASP)导数作为 FD 的直接替代方案。WASP 通过复用先前相关导数计算的信息,显著加速并稳定新导数的生成,特别适合 MPC 的迭代更新需求。我们在多种机器人形态的 MJPC 任务上评估,结果表明 WASP 能无缝集成,表现稳定可靠,在 iLQG 等基于导数的规划器中实现最高 2× 速度提升;同时,其性能优于原生的随机采样规划器,兼具效率与可靠性。我们已开源集成了 WASP 导数的 MJPC 版本,支持后续研究与应用。
原文摘要 · Abstract (English)
MuJoCo is a powerful and efficient physics simulator widely used in robotics. One common way it is applied in practice is through Model Predictive Control (MPC), which uses repeated rollouts of the simulator to optimize future actions and generate responsive control policies in real time. To make this process more accessible, the open source library MuJoCo MPC (MJPC) provides ready-to-use MPC algorithms and implementations built directly on top of the MuJoCo simulator. However, MJPC relies on finite differencing (FD) to compute derivatives through the underlying MuJoCo simulator, which is often a key bottleneck that can make it prohibitively costly for time-sensitive tasks, especially in high-DOF systems or complex scenes. In this paper, we introduce the use of Web of Affine Spaces (WASP) derivatives within MJPC as a drop-in replacement for FD. WASP is a recently developed approach for efficiently computing sequences of accurate derivative approximations. By reusing information from prior, related derivative calculations, WASP accelerates and stabilizes the computation of new derivatives, making it especially well suited for MPC's iterative, fine-grained updates over time. We evaluate WASP across a diverse suite of MJPC tasks spanning multiple robot embodiments. Our results suggest that WASP derivatives are particularly effective in MJPC: it integrates seamlessly across tasks, delivers consistently robust performance, and achieves up to a 2$\mathsf{x}$ speedup compared to an FD backend when used with derivative-based planners, such as iLQG. In addition, WASP-based MPC outperforms MJPC's stochastic sampling-based planners on our evaluation tasks, offering both greater efficiency and reliability. To support adoption and future research, we release an open-source implementation of MJPC with WASP derivatives fully integrated.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。