用扩散模型统一搜索与策略优化,解决世界模型长时预测偏差问题。
Scaling World-Model Reinforcement Learning Through Diffusion Policy Optimization

- 将策略优化转为潜空间轨迹的扩散过程,隐式建模能量函数。
- 在多任务离线预训练中,模型越大性能越强,呈现单调提升。
- 适合大规模世界模型强化学习,尤其擅长长序列决策任务。
基于模型的强化学习可通过世界模型实现有效扩展,但实际应用中仍面临根本性限制。常见挑战包括模型偏差和误差累积,导致长时序预测性能下降。此外,我们识别出一个更关键却未被充分关注的瓶颈:现有世界模型方法中搜索与价值学习之间的结构不匹配。具体表现为,策略优化常依赖于独立非搜索策略诱导的价值函数,造成训练不一致,最终导致次优学习。为此,我们提出基于模型的扩散策略优化(MBDPO),通过扩散策略表示统一搜索与策略优化,从而释放世界模型在可扩展策略学习中的潜力。不再显式构建基于学习的世界模型的规划器,而是将策略优化重构为潜空间中搜索轨迹的扩散过程。在此框架下,从数据集中提取隐式能量函数作为策略锚点,使MBDPO能够优化得分场以指导策略改进,同时缓解结构性错配。我们在多种设置下评估了MBDPO,包括多任务离线预训练、在线学习以及离线到在线微调。在离线场景中,通过在大规模数据集上预训练,观察到随着模型容量增加,性能持续且单调提升。
原文摘要 · Abstract (English)
Model-based reinforcement learning (RL) can be effectively supported at scale through the use of world models. However, in practice, scaling such approaches remains fundamentally limited. A commonly recognized challenge is model bias and error compounding, which degrade long-horizon predictions. Beyond these issues, we identify a more critical yet underexplored bottleneck: a structural misalignment between search and value learning in existing world model approaches. In particular, policy improvement often relies on value functions induced by a separate, non-search policy, resulting in training inconsistency and ultimately suboptimal learning. To address this limitation, we propose Model-Based Diffusion Policy Optimization (MBDPO) in world models, a framework that unifies search and policy optimization through diffusion policy representations, thereby unlocking the potential of world models for scalable policy learning. Instead of constructing an explicit planner over a learned world model, we reformulate policy optimization as a diffusion process over searched trajectories in latent world models. In this view, we extract an implicit energy function from the collected dataset that anchors the policy, enabling MBDPO to refine the score field for policy optimization while mitigating misalignment. We evaluate MBDPO across a wide range of settings, including multi-task offline pretraining, online learning, and offline-to-online fine-tuning. In the offline regime, we further investigate its scaling behavior by pretraining on large-scale datasets, observing consistent and monotonic performance gains with increasing model capacity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。