用世界模型离线优化扩散策略,大幅减少真实机器人交互次数。
DiWA: Diffusion Policy Adaptation with World Models
- 借助世界模型模拟环境,实现扩散策略的离线强化学习微调。
- 在CALVIN基准上仅用少量离线数据即提升8个任务性能。
- 相比传统方法减少数百万次真实交互,更安全高效,适合实际机器人学习。
使用强化学习微调扩散策略面临巨大挑战:每个动作预测的长去噪序列阻碍有效奖励传播;且标准RL方法需数百万次真实环境交互,成为实用化瓶颈。尽管已有研究将扩散策略的去噪过程建模为马尔可夫决策过程以支持强化学习更新,但仍严重依赖环境交互,效率低下。为此,我们提出DiWA,一种全新框架,利用世界模型实现完全离线的强化学习微调,无需真实交互。与需数百万次环境交互才能微调一整套机器人技能的无模型方法不同,DiWA仅需一次在数十万次离线游戏数据上训练的世界模型即可实现有效适应。这显著提升了样本效率,使方法更实用、更安全。在具有挑战性的CALVIN基准上,DiWA仅通过离线适应便在8项任务中提升性能,所需物理交互量比无模型基线少多个数量级。据我们所知,这是首次成功展示基于离线世界模型对真实机器人技能进行扩散策略微调。代码已公开于 https://diwa.cs.uni-freiburg.de。
原文摘要 · Abstract (English)
Fine-tuning diffusion policies with reinforcement learning (RL) presents significant challenges. The long denoising sequence for each action prediction impedes effective reward propagation. Moreover, standard RL methods require millions of real-world interactions, posing a major bottleneck for practical fine-tuning. Although prior work frames the denoising process in diffusion policies as a Markov Decision Process to enable RL-based updates, its strong dependence on environment interaction remains highly inefficient. To bridge this gap, we introduce DiWA, a novel framework that leverages a world model for fine-tuning diffusion-based robotic skills entirely offline with reinforcement learning. Unlike model-free approaches that require millions of environment interactions to fine-tune a repertoire of robot skills, DiWA achieves effective adaptation using a world model trained once on a few hundred thousand offline play interactions. This results in dramatically improved sample efficiency, making the approach significantly more practical and safer for real-world robot learning. On the challenging CALVIN benchmark, DiWA improves performance across eight tasks using only offline adaptation, while requiring orders of magnitude fewer physical interactions than model-free baselines. To our knowledge, this is the first demonstration of fine-tuning diffusion policies for real-world robotic skills using an offline world model. We make the code publicly available at https://diwa.cs.uni-freiburg.de.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。