用强化学习让机器人世界模型能稳定预测多步未来画面。
Persistent Robot World Models: Stabilizing Multi-Step Rollouts via Reinforcement Learning
- 用自回归回放训练世界模型,让其自我纠错。
- 多候选未来对比训练,提升高保真度预测。
- 多视角感知奖励融合,提升训练信号质量。
动作条件的机器人世界模型可根据动作序列生成场景未来视频帧,为难以用传统物理引擎建模的任务提供了新思路。但这类模型在自回归生成时会因误差累积导致画面快速劣化。本文提出四点改进:首先,引入强化学习后训练方案,让模型基于自身生成的自回归回放进行优化,采用近期扩散模型的对比强化学习目标,并证明其收敛性可直接迁移;其次,设计多候选变量长度未来生成与比较机制,强化高保真预测;第三,开发高效多视角视觉保真度奖励,结合不同摄像头视角的感知指标,在片段级别聚合,提供密集且低方差的训练信号;第四,在DROID数据集上验证,本方法在所有指标上均超越最强基线(如外部相机LPIPS降低14%,腕部相机SSIM提升9.1%),98%的成对比较胜出,盲测人类偏好率达80%。
原文摘要 · Abstract (English)
Action-conditioned robot world models generate future video frames of the manipulated scene given a robot action sequence, offering a promising alternative for simulating tasks that are difficult to model with traditional physics engines. However, these models are optimized for short-term prediction and break down when deployed autoregressively: each predicted clip feeds back as context for the next, causing errors to compound and visual quality to rapidly degrade. We address this through the following contributions. First, we introduce a reinforcement learning (RL) post-training scheme that trains the world model on its own autoregressive rollouts rather than on ground-truth histories. We achieve this by adapting a recent contrastive RL objective for diffusion models to our setting and show that its convergence guarantees carry over exactly. Second, we design a training protocol that generates and compares multiple candidate variable-length futures from the same rollout state, reinforcing higher-fidelity predictions over lower-fidelity ones. Third, we develop efficient, multi-view visual fidelity rewards that combine complementary perceptual metrics across camera views and are aggregated at the clip level for dense, low-variance training signal. Fourth, we show that our approach establishes a new state-of-the-art for rollout fidelity on the DROID dataset, outperforming the strongest baseline on all metrics (e.g., LPIPS reduced by 14% on external cameras, SSIM improved by 9.1% on the wrist camera), winning 98% of paired comparisons, and achieving an 80% preference rate in a blind human study.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。