arXiv:2605.30056cs.ROcs.LG2026-05中稿 · ICML

用批评网络引导扩散策略,提升强化学习的探索与利用平衡。

Sample-Efficient Diffusion-based Reinforcement Learning with Critic Guidance

论文配图:Sample-Efficient Diffusion-based Reinforcement Learning with Critic Guidance
图 1 · 摘自论文原文
  • 在去噪过程中引入批评网络指导动作生成,实现无训练引导
  • 5个MuJoCo任务上性能领先,真实机器人抓取任务首次成功应用
  • 兼顾探索多样性与收敛速度,适合需要高效样本的学习场景

近期强化学习进展得益于扩散策略的多模态特性与探索能力。现有方法分为采样式优化(探索强但利用弱,收敛慢)和梯度式优化(利用充分但易退化为单峰策略)。本文提出CGPO(Critic-Guided Policy Optimization),通过在扩散策略去噪过程中集成无训练的批评网络引导,将动作生成导向高价值区域,并以引导动作作为回归目标。该方法显著缩短高质量动作获取时间,提升最终性能,在5个MuJoCo运动任务中达到当前最优,且首次成功应用于真实世界机器人抓取任务(Franka机械臂)。官方页面已发布:https://dingsht.tech/cgpo-webpage。

原文摘要 · Abstract (English)

Recent advances in reinforcement learning (RL) have achieved great successes by leveraging the multimodality and exploration capability of diffusion policies. Among these approaches, one representative branch focuses on the sampling-based policy optimization. This design enables better exploration capability of the diffusion model, particularly at the beginning of training, but suffer from low exploitation in Q-value information, resulting in a slow policy convergence. Another branch pays attention to gradient-based policy optimization, which sufficiently exploits the gradient of the Q function yet tends to collapse into a unimodal policy with low diversity. To address this issue, we propose CGPO, \textbf{C}ritic-\textbf{G}uided diffusion \textbf{P}olicy \textbf{O}ptimization, which effectively balances exploration and exploitation with the training-free guidance technique integrated into the denoising process of diffusion policy. Concretely, CGPO steers action generation toward high-value regions defined by the critic network and uses the guided actions as regression objectives. In this manner, CGPO reduces the time required to obtain high-quality actions and improves final performance with better balance between the exploration-exploitation tradeoff. We validate the effectiveness of CGPO on 5 MuJoCo locomotion tasks, and CGPO achieves state-of-the-art performance compared with existing diffusion-based RL methods. Notably, CGPO is the first success to incorporate diffusion policy into real-world RL, with its superior performance on Franka robot arm grasping tasks. Our official page is released at https://dingsht.tech/cgpo-webpage.

强化学习扩散模型机器人控制

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