arXiv:2503.18871cs.LGcs.AI2025-03ICLR被引 14

用MPC自我迭代优化策略,提升复杂控制任务的效率与稳定性。

Bootstrapped Model Predictive Control

  • 用MPC专家行为引导策略网络学习,实现自举式策略优化。
  • 在高维运动任务上数据效率显著提升,训练更稳定且性能更好。
  • 适合追求高效稳定强化学习的算法研究者和工业应用开发者。

模型预测控制(MPC)在连续控制任务中表现优异。当具备世界模型和价值函数时,提前规划动作序列可获得更优策略。现有方法通常以无模型方式获取价值函数及对应策略,但在复杂任务中表现不佳,导致策略学习差、价值估计不准。为此,本文提出自举式模型预测控制(BMPC),通过模仿MPC专家来学习网络策略,并利用该策略反向优化MPC过程。结合基于模型的TD学习,策略学习提升了价值估计精度,进一步增强MPC效率。我们还引入懒惰重分析机制,实现计算高效的模仿学习。实验表明,BMPC在多种连续控制任务上优于先前方法,尤其在高维运动任务中显著提升数据效率,同时改善渐近性能与训练稳定性,且训练时间相当、网络规模更小。代码已开源。

原文摘要 · Abstract (English)

Model Predictive Control (MPC) has been demonstrated to be effective in continuous control tasks. When a world model and a value function are available, planning a sequence of actions ahead of time leads to a better policy. Existing methods typically obtain the value function and the corresponding policy in a model-free manner. However, we find that such an approach struggles with complex tasks, resulting in poor policy learning and inaccurate value estimation. To address this problem, we leverage the strengths of MPC itself. In this work, we introduce Bootstrapped Model Predictive Control (BMPC), a novel algorithm that performs policy learning in a bootstrapped manner. BMPC learns a network policy by imitating an MPC expert, and in turn, uses this policy to guide the MPC process. Combined with model-based TD-learning, our policy learning yields better value estimation and further boosts the efficiency of MPC. We also introduce a lazy reanalyze mechanism, which enables computationally efficient imitation learning. Our method achieves superior performance over prior works on diverse continuous control tasks. In particular, on challenging high-dimensional locomotion tasks, BMPC significantly improves data efficiency while also enhancing asymptotic performance and training stability, with comparable training time and smaller network sizes. Code is available at https://github.com/wertyuilife2/bmpc.

强化学习模型预测控制策略学习

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