用KL正则统一规划型强化学习,提升采样效率与性能
A KL-regularization Framework for Learning to Plan with Adaptive Priors
- 将规划器动作分布作为先验,通过KL正则化优化策略
- 在连续控制任务中实现更优的长期回报与采样效率
- 适用于追求高样本效率的模型预测控制场景
有效探索仍是基于模型的强化学习(MBRL)的核心挑战,尤其在高维连续控制任务中,样本效率至关重要。近期工作利用学习到的策略作为模型预测路径积分(MPPI)规划的提议分布。早期方法独立更新采样策略,通常通过确定性策略梯度最大化学习到的价值函数并结合熵正则化。然而,由于训练中遇到的状态依赖于MPPI规划器,使采样策略与规划器对齐可提高价值估计精度和长期性能。为此,最新方法通过最小化采样策略与规划器分布之间的KL散度,或引入规划器引导的正则化项来更新策略。本文提出统一框架PO-MPC,将基于MPPI的强化学习方法整合为一类KL正则化的MBRL方法,将规划器的动作分布作为策略优化中的先验。通过使学习策略与规划器行为对齐,PO-MPC在回报最大化与KL最小化之间提供更灵活的权衡。我们阐明了先前方法作为该族的特例,并探索了此前未研究的变体。实验表明,这些扩展配置显著提升性能,推动了基于MPPI的强化学习的前沿水平。
原文摘要 · Abstract (English)
Effective exploration remains a central challenge in model-based reinforcement learning (MBRL), particularly in high-dimensional continuous control tasks where sample efficiency is crucial. A prominent line of recent work leverages learned policies as proposal distributions for Model-Predictive Path Integral (MPPI) planning. Initial approaches update the sampling policy independently of the planner distribution, typically maximizing a learned value function with deterministic policy gradient and entropy regularization. However, because the states encountered during training depend on the MPPI planner, aligning the sampling policy with the planner improves the accuracy of value estimation and long-term performance. To this end, recent methods update the sampling policy by minimizing KL divergence to the planner distribution or by introducing planner-guided regularization into the policy update. In this work, we unify these MPPI-based reinforcement learning methods under a single framework by introducing Policy Optimization-Model Predictive Control (PO-MPC), a family of KL-regularized MBRL methods that integrate the planner's action distribution as a prior in policy optimization. By aligning the learned policy with the planner's behavior, PO-MPC allows more flexibility in the policy updates to trade off Return maximization and KL divergence minimization. We clarify how prior approaches emerge as special cases of this family, and we explore previously unstudied variations. Our experiments show that these extended configurations yield significant performance improvements, advancing the state of the art in MPPI-based RL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。