arXiv:2509.19080cs.ROcs.AI2025-09被引 32

用扩散模型模拟真实环境,让机械臂在虚拟世界中自我优化抓取策略。

World4RL: Diffusion World Models for Policy Refinement with Reinforcement Learning for Robotic Manipulation

  • 用扩散模型构建高保真虚拟环境,替代真实训练
  • 在虚拟环境中直接优化策略,成功率显著高于模仿学习
  • 适合需要高效安全训练的机器人抓取任务

机器人抓取策略通常通过模仿学习初始化,但受限于专家数据稀缺和覆盖范围窄。强化学习可进一步优化策略,但真实机器人训练成本高且不安全,而仿真训练又存在仿真到现实的差距。生成模型(尤其是扩散模型)在真实世界模拟方面展现出强大能力。本文提出World4RL框架,利用基于扩散模型的世界模型作为高保真模拟器,在纯虚拟环境中对预训练策略进行优化。不同于以往仅用于规划的世界模型,本框架实现端到端策略优化。核心设计包括:在多任务数据集上预训练捕捉多样动态的扩散世界模型,并在冻结的世界模型中完全进行策略优化,避免在线真实交互。我们还设计了针对机器人抓取的双热动作编码方案,并采用扩散主干网络提升建模精度。大量仿真与真实实验表明,World4RL实现了高保真环境建模,能持续提升策略性能,显著优于模仿学习及其他基线方法。

原文摘要 · Abstract (English)

Robotic manipulation policies are commonly initialized through imitation learning, but their performance is limited by the scarcity and narrow coverage of expert data. Reinforcement learning can refine polices to alleviate this limitation, yet real-robot training is costly and unsafe, while training in simulators suffers from the sim-to-real gap. Recent advances in generative models have demonstrated remarkable capabilities in real-world simulation, with diffusion models in particular excelling at generation. This raises the question of how diffusion model-based world models can be combined to enhance pre-trained policies in robotic manipulation. In this work, we propose World4RL, a framework that employs diffusion-based world models as high-fidelity simulators to refine pre-trained policies entirely in imagined environments for robotic manipulation. Unlike prior works that primarily employ world models for planning, our framework enables direct end-to-end policy optimization. World4RL is designed around two principles: pre-training a diffusion world model that captures diverse dynamics on multi-task datasets and refining policies entirely within a frozen world model to avoid online real-world interactions. We further design a two-hot action encoding scheme tailored for robotic manipulation and adopt diffusion backbones to improve modeling fidelity. Extensive simulation and real-world experiments demonstrate that World4RL provides high-fidelity environment modeling and enables consistent policy refinement, yielding significantly higher success rates compared to imitation learning and other baselines.

机器人抓取扩散模型强化学习虚拟训练

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