arXiv:2602.06219cs.ROcs.AI2026-02

用内外世界模型联合训练,让机器人在图像空间高效完成复杂操作。

Coupled Local and Global World Models for Efficient First Order RL

  • 用大模型生成精确轨迹,小模型快速计算梯度。
  • 在推箱子任务中样本效率远超PPO算法。
  • 适合无物理引擎的视觉操控任务,无需手工建模。

世界模型为更真实地捕捉复杂动态(如接触、非刚性)和复杂感知信息(如视觉)提供了前景,尤其在传统模拟器难以处理的情况下。然而,这些模型评估计算量大,制约了主流强化学习方法在复杂运动任务中的应用,尤其是在操作类任务上表现不佳。本文提出一种完全绕过模拟器的方法:在从机器人与真实环境交互中学习到的世界模型内训练强化学习策略。核心是通过一种新型解耦的一阶梯度(FoG)方法,实现大规模扩散模型上的策略训练:全尺度世界模型生成准确的前向轨迹,轻量级潜在空间代理模型近似其局部动力学以实现高效的梯度计算。这种局部与全局世界模型的耦合,确保了高保真轨迹展开的同时具备可计算的微分能力。我们在推箱子操作任务中验证了该方法的有效性,其样本效率显著优于PPO。进一步在四足机器人进行的自中心物体操作任务中评估,结果表明,在无需依赖手工物理模拟器的前提下,基于数据驱动的世界模型内学习是解决图像空间中难建模强化学习任务的可行路径。

原文摘要 · Abstract (English)

World models offer a promising avenue for more faithfully capturing complex dynamics, including contacts and non-rigidity, as well as complex sensory information, such as visual perception, in situations where standard simulators struggle. However, these models are computationally complex to evaluate, posing a challenge for popular RL approaches that have been successfully used with simulators to solve complex locomotion tasks but yet struggle with manipulation. This paper introduces a method that bypasses simulators entirely, training RL policies inside world models learned from robots' interactions with real environments. At its core, our approach enables policy training with large-scale diffusion models via a novel decoupled first-order gradient (FoG) method: a full-scale world model generates accurate forward trajectories, while a lightweight latent-space surrogate approximates its local dynamics for efficient gradient computation. This coupling of a local and global world model ensures high-fidelity unrolling alongside computationally tractable differentiation. We demonstrate the efficacy of our method on the Push-T manipulation task, where it significantly outperforms PPO in sample efficiency. We further evaluate our approach through an ego-centric object manipulation task with a quadruped. Together, these results demonstrate that learning inside data-driven world models is a promising pathway for solving hard-to-model RL tasks in image space without reliance on hand-crafted physics simulators.

强化学习世界模型视觉操控扩散模型

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