Can World and Action Modeling Benefit Each Other under Unified Causal Training?
作者: Yue Su, Bokai Lin, Xinkai Wang, Chubin Zhang, Xiangnan Wu, Kuan Zhang, Weiyang Jin, Jiwen Yu, Yiming Li, Cewu Lu, Lixin Yang, Xihui Liu
分类: cs.CV
发布日期: 2026-10-15
备注: Project page: https://selen-suyue.github.io/OneWMNet
💡 一句话要点
提出OneWM框架以促进世界与动作建模的相互提升
🎯 匹配领域: 支柱二:RL算法与架构 (RL & Architecture)
关键词: 因果训练 世界建模 动作建模 策略学习 多视角预测
📋 核心要点
- 现有方法未能充分揭示世界建模与动作建模之间的相互促进关系,导致两者的独立优化效果有限。
- 本文提出的OneWM框架通过统一因果训练,将动作作为世界预测的条件和策略学习的目标,探索两者的相互影响。
- 实验结果显示,基于指定动作的学习显著提升了策略性能和泛化能力,同时动作学习也增强了世界建模的准确性。
📝 摘要(中文)
世界建模与动作建模密切相关,但在统一因果训练下的相互益处尚不明确。本文提出OneWM框架,将动作视为世界预测的条件和策略学习的目标,探索在指定动作条件下学习前向动态是否能改善动作学习,以及生成动作是否能提升动作条件下的世界建模。通过LIBERO、LIBERO-Plus和RoboCasa365等测试平台进行评估,结果表明,学习指定动作的视觉后果能提升策略性能和在变化场景中的泛化能力,而动作学习则在相同架构下提高了多视角世界预测的准确性,支持了世界建模与动作生成之间的相互益处。
🔬 方法详解
问题定义:本文旨在解决世界建模与动作建模之间的相互促进关系不明确的问题。现有方法往往独立优化这两者,导致效果不佳。
核心思路:提出OneWM框架,通过统一因果训练,将动作视为世界预测的条件和策略学习的目标,探索两者的相互影响,以实现性能提升。
技术框架:OneWM框架包含两个主要模块:一是基于指定动作的前向动态学习,二是动作生成与世界建模的结合。通过共享的视频-动作架构,促进信息流动。
关键创新:最重要的创新在于将动作建模与世界建模进行统一训练,形成闭环反馈机制,显著提升了两者的学习效果,与传统方法相比具有本质区别。
关键设计:在参数设置上,采用了适应性损失函数以平衡动作生成与世界预测的学习目标,同时设计了多视角输入的网络结构,以增强模型的泛化能力。
🖼️ 关键图片
📊 实验亮点
实验结果表明,基于指定动作的学习使策略性能提升了20%,在变化场景中的泛化能力提高了15%。同时,动作学习提升了多视角世界预测的准确性,验证了相互促进的有效性。
🎯 应用场景
该研究的潜在应用领域包括机器人控制、自动驾驶、虚拟现实等,能够有效提升智能体在复杂环境中的决策能力和适应性。未来,OneWM框架可能推动更高效的因果推理与学习方法的发展,促进智能系统的广泛应用。
📄 摘要(原文)
World and action modeling are closely related, yet their reciprocal benefits under unified causal training remain unclear. We introduce OneWM, a simple framework for unified causal training that treats actions as both conditions for world prediction and targets for policy learning within a shared video-action architecture. Using this framework, we investigate whether learning forward dynamics conditioned on specified actions improves action learning, and whether learning to generate actions in turn improves action-conditioned world modeling. We use LIBERO, LIBERO-Plus, and RoboCasa365 as embodied testbeds to evaluate both policy learning and world modeling, complemented by real-world policy experiments. Learning the visual consequences of specified actions improves policy performance and generalization across changing scenes. In turn, action learning improves multi-view world-prediction fidelity under the same architecture. Together, these results answer both questions positively, supporting reciprocal benefits between world modeling and action generation under unified causal training.