arXiv:2505.18763cs.LG2025-05NeurIPS被引 21

首次将扩散策略成功融入在线强化学习,实现高效机器人训练。

GenPO: Generative Diffusion Models Meet On-Policy Reinforcement Learning

  • 提出新框架GenPO,用双重虚拟动作实现扩散策略的可逆映射。
  • 在8个IsaacLab任务上超越现有基线,尤其在复杂运动与操作任务中表现优异。
  • 适合追求高效、多模态机器人策略训练的研究者与工程师。

近期强化学习进展表明,基于扩散模型的策略具备强大探索能力与多模态特性。尽管离线和异策略强化学习已取得显著进展,但将扩散策略集成到如PPO等在线策略框架中仍鲜有研究。这一空白尤为关键,因为大规模并行GPU加速仿真器(如IsaacLab)专为在线算法优化,可快速训练复杂机器人任务。核心挑战在于扩散策略下状态-动作对数似然难以计算,流模型因前向-反向过程不可逆及离散化误差(如Euler-Maruyama近似)而无法求解。为此,我们提出GenPO,通过精确扩散反演构建可逆动作映射。GenPO引入新颖的双重虚拟动作机制,借助交替更新实现可逆性,突破对数似然计算障碍。同时利用动作对数似然实现无偏熵与KL散度估计,支持在线更新中的自适应学习率与熵正则化。在八个IsaacLab基准测试中(包括仿人行走(Ant, Humanoid, Anymal-D, Unitree H1, Go2)、灵巧操作(Shadow Hand)、空中控制(Quadcopter)及机械臂任务(Franka))的大量实验表明,GenPO优于现有强化学习基线。值得注意的是,GenPO是首个成功将扩散策略融入在线强化学习的方法,释放其在大规模并行训练与真实机器人部署中的潜力。

原文摘要 · Abstract (English)

Recent advances in reinforcement learning (RL) have demonstrated the powerful exploration capabilities and multimodality of generative diffusion-based policies. While substantial progress has been made in offline RL and off-policy RL settings, integrating diffusion policies into on-policy frameworks like PPO remains underexplored. This gap is particularly significant given the widespread use of large-scale parallel GPU-accelerated simulators, such as IsaacLab, which are optimized for on-policy RL algorithms and enable rapid training of complex robotic tasks. A key challenge lies in computing state-action log-likelihoods under diffusion policies, which is straightforward for Gaussian policies but intractable for flow-based models due to irreversible forward-reverse processes and discretization errors (e.g., Euler-Maruyama approximations). To bridge this gap, we propose GenPO, a generative policy optimization framework that leverages exact diffusion inversion to construct invertible action mappings. GenPO introduces a novel doubled dummy action mechanism that enables invertibility via alternating updates, resolving log-likelihood computation barriers. Furthermore, we also use the action log-likelihood for unbiased entropy and KL divergence estimation, enabling KL-adaptive learning rates and entropy regularization in on-policy updates. Extensive experiments on eight IsaacLab benchmarks, including legged locomotion (Ant, Humanoid, Anymal-D, Unitree H1, Go2), dexterous manipulation (Shadow Hand), aerial control (Quadcopter), and robotic arm tasks (Franka), demonstrate GenPO's superiority over existing RL baselines. Notably, GenPO is the first method to successfully integrate diffusion policies into on-policy RL, unlocking their potential for large-scale parallelized training and real-world robotic deployment.

扩散模型强化学习机器人在线训练

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