arXiv:2603.04790cs.LGcs.RO2026-03被引 1

提出新方法让扩散策略高效用于强化学习,无需复杂计算。

Diffusion Policy through Conditional Proximal Policy Optimization

  • 将策略迭代与扩散过程对齐,只需计算简单高斯概率。
  • 在IsaacLab和MuJoCo上实现多模态行为,性能优于基线。
  • 天然支持熵正则化,适合需要多样化决策的任务。

强化学习广泛应用于游戏和机器人等决策问题。近期,扩散策略展现出建模多模态行为的潜力,相比传统高斯策略能生成更丰富灵活的动作。然而,将强化学习与扩散模型结合的关键挑战在于难以计算扩散模型下的动作对数似然,严重限制了其在在线策略强化学习中的应用。现有方法通常通过整个去噪过程计算或近似对数似然,导致内存与计算效率低下。为此,本文提出一种新颖高效的在线策略扩散策略训练方法,仅需评估简单的高斯概率即可实现。该方法通过将策略迭代与扩散过程对齐,形成区别于以往的新范式。此外,该公式可自然处理熵正则化,这在扩散策略中通常难以实现。实验表明,所提方法能生成多模态策略行为,并在IsaacLab和MuJoCo Playground多个基准任务上取得更优性能。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has been extensively employed in a wide range of decision-making problems, such as games and robotics. Recently, diffusion policies have shown strong potential in modeling multi-modal behaviors, enabling more diverse and flexible action generation compared to the conventional Gaussian policy. Despite various attempts to combine RL with diffusion, a key challenge is the difficulty of computing action log-likelihood under the diffusion model. This greatly hinders the direct application of diffusion policies in on-policy reinforcement learning. Most existing methods calculate or approximate the log-likelihood through the entire denoising process in the diffusion model, which can be memory- and computationally inefficient. To overcome this challenge, we propose a novel and efficient method to train a diffusion policy in an on-policy setting that requires only evaluating a simple Gaussian probability. This is achieved by aligning the policy iteration with the diffusion process, which is a distinct paradigm compared to previous work. Moreover, our formulation can naturally handle entropy regularization, which is often difficult to incorporate into diffusion policies. Experiments demonstrate that the proposed method produces multimodal policy behaviors and achieves superior performance on a variety of benchmark tasks in both IsaacLab and MuJoCo Playground.

扩散模型强化学习多模态策略

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