arXiv:2604.09035cs.AIcs.LG2026-04

用优势值引导扩散模型,提升强化学习采样效率与长期回报。

Advantage-Guided Diffusion for Model-Based Reinforcement Learning

  • 用优势估计指导扩散过程,让生成轨迹聚焦高回报区域。
  • 在多个MuJoCo任务上,样本效率和最终奖励提升2倍。
  • 兼容现有架构,无需修改训练目标,适合长序列决策场景。

基于模型的强化学习(MBRL)使用自回归世界模型时存在误差累积问题,而扩散世界模型通过联合生成轨迹段缓解此问题。然而,现有扩散引导方式或仅依赖策略(忽略价值信息),或基于奖励(短时程下易陷入短视)。本文提出优势引导扩散(AGD-MBRL),利用智能体的优势估计引导反向扩散过程,使采样集中于预期长期回报更高的轨迹,超越生成窗口。提出两种引导方法:Sigmoid优势引导(SAG)和指数优势引导(EAG)。理论证明,在标准假设下,经SAG或EAG引导的扩散模型可实现按优势重加权采样,促进策略改进。实验表明,相比无引导扩散模型,AGD-MBRL生成的轨迹对应更高价值策略。该方法可无缝集成至PolyGRAD类架构,仅引导状态分量,动作生成仍由策略条件控制,无需修改扩散训练目标。在MuJoCo控制任务(HalfCheetah、Hopper、Walker2D、Reacher)中,AGD-MBRL在样本效率和最终回报上优于PolyGRAD(在线Diffuser风格奖励引导)及模型无关基线(PPO/TRPO),部分任务提升达2倍。结果表明,优势感知引导是解决扩散模型MBRL短时程短视问题的简单有效方案。

原文摘要 · Abstract (English)

Model-based reinforcement learning (MBRL) with autoregressive world models suffers from compounding errors, whereas diffusion world models mitigate this by generating trajectory segments jointly. However, existing diffusion guides are either policy-only, discarding value information, or reward-based, which becomes myopic when the diffusion horizon is short. We introduce Advantage-Guided Diffusion for MBRL (AGD-MBRL), which steers the reverse diffusion process using the agent's advantage estimates so that sampling concentrates on trajectories expected to yield higher long-term return beyond the generated window. We develop two guides: (i) Sigmoid Advantage Guidance (SAG) and (ii) Exponential Advantage Guidance (EAG). We prove that a diffusion model guided through SAG or EAG allows us to perform reweighted sampling of trajectories with weights increasing in state-action advantage-implying policy improvement under standard assumptions. Additionally, we show that the trajectories generated from AGD-MBRL follow an improved policy (that is, with higher value) compared to an unguided diffusion model. AGD integrates seamlessly with PolyGRAD-style architectures by guiding the state components while leaving action generation policy-conditioned, and requires no change to the diffusion training objective. On MuJoCo control tasks (HalfCheetah, Hopper, Walker2D and Reacher), AGD-MBRL improves sample efficiency and final return over PolyGRAD, an online Diffuser-style reward guide, and model-free baselines (PPO/TRPO), in some cases by a margin of 2x. These results show that advantage-aware guidance is a simple, effective remedy for short-horizon myopia in diffusion-model MBRL.

强化学习扩散模型轨迹生成策略优化

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