提出DIPOLE算法,让扩散策略训练更稳定可控。
Dichotomous Diffusion Policy Optimization
- 将最优策略分解为收益最大化与最小化两类策略
- 推理时线性组合得分,灵活调节贪婪程度
- 适用于复杂任务如自动驾驶,支持端到端训练
基于扩散的策略因其强大的表达能力和推理阶段的可控生成而广泛应用于决策任务。然而,使用强化学习(RL)有效训练大型扩散策略仍具挑战性:现有方法或因直接最大化价值目标导致训练不稳定,或依赖粗糙的高斯似然近似引发计算开销,需大量微小去噪步骤。本文提出DIPOLE(Dichotomous Diffusion Policy Improvement),一种新型强化学习算法,实现稳定且可控的扩散策略优化。我们重新审视了带KL正则化的强化学习目标,其虽能提供理想的加权回归目标,但难以平衡贪婪性与稳定性。为此,我们设计了一种贪婪化策略正则化机制,自然地将最优策略分解为一对可稳定学习的二元策略:一个专注于奖励最大化,另一个聚焦于奖励最小化。推理时,通过线性组合两者的得分生成优化动作,从而灵活控制贪婪水平。在ExORL和OGBench的离线及离线到在线强化学习设置中评估表明本方法有效。此外,我们使用DIPOLE训练大型视觉-语言-动作(VLA)模型用于端到端自动驾驶,并在大规模真实世界基准NAVSIM上进行评估,验证其在复杂现实应用中的潜力。
原文摘要 · Abstract (English)
Diffusion-based policies have gained growing popularity in solving a wide range of decision-making tasks due to their superior expressiveness and controllable generation during inference. However, effectively training large diffusion policies using reinforcement learning (RL) remains challenging. Existing methods either suffer from unstable training due to directly maximizing value objectives, or face computational issues due to relying on crude Gaussian likelihood approximation, which requires a large amount of sufficiently small denoising steps. In this work, we propose DIPOLE (Dichotomous diffusion Policy improvement), a novel RL algorithm designed for stable and controllable diffusion policy optimization. We begin by revisiting the KL-regularized objective in RL, which offers a desirable weighted regression objective for diffusion policy extraction, but often struggles to balance greediness and stability. We then formulate a greedified policy regularization scheme, which naturally enables decomposing the optimal policy into a pair of stably learned dichotomous policies: one aims at reward maximization, and the other focuses on reward minimization. Under such a design, optimized actions can be generated by linearly combining the scores of dichotomous policies during inference, thereby enabling flexible control over the level of greediness.Evaluations in offline and offline-to-online RL settings on ExORL and OGBench demonstrate the effectiveness of our approach. We also use DIPOLE to train a large vision-language-action (VLA) model for end-to-end autonomous driving (AD) and evaluate it on the large-scale real-world AD benchmark NAVSIM, highlighting its potential for complex real-world applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。