WALL-WM:在事件节点处雕刻世界动作建模 WALL-WM: Carving World Action Modeling at the Event Joints
用语义事件而非固定时间块作为学习单元,解决语言-视觉-动作粒度失配问题
前置知识
Vision-Language-Action (VLA) 模型
VLA 模型是从视觉观察和自然语言指令预测可执行动作的具身基础模型。它继承自预训练的视觉-语言模型(如 CLIP、VLM),通过添加动作接口将静态图像理解扩展到动态控制。典型架构包括视觉编码器、语言编码器和动作解码器三部分,通过监督学习从遥操作数据中学习观测-动作映射。关键挑战是如何保留预训练的视觉语义先验,同时适应机器人控制所需的精细运动和接触敏感性。
WALL-WM 是对 VLA 范式的扩展,引入了未来观察建模和事件级推理。理解 VLA 的基本架构和局限性(如缺乏显式世界模型、长泛化困难)是理解 WALL-WM 创新点的基础。
Flow Matching / Diffusion 模型
Flow Matching 是一类生成式建模方法,通过学习从噪声分布到目标分布的向量场来生成样本。相比传统扩散模型,Flow Matching 通常有更简单的训练目标和更好的收敛性。在机器人动作生成中,常用 v-prediction 形式,即预测噪声和干净样本 x0 的速度向量。给定噪声时间步,模型学习预测,通过迭代去噪生成样本。优势是可以建模复杂的动作分布,保持多模态特性。
WALL-WM 的视频和动作塔都基于 Wan 系列的 DiT 架构,使用 v-prediction flow matching。理解这一技术对于掌握模型如何进行去噪预测、视频生成和动作轨迹生成至关重要。
DiT (Diffusion Transformer)
DiT 是将扩散模型中的 U-Net 骨干网络替换为 Transformer 的架构设计。通过在像素或潜在空间使用自注意力机制,DiT 能够更好地捕捉长距离依赖和空间关系。核心组件包括:层归一化用于时间条件注入、多头自注意力、前馈网络。每个 DiT 块通常遵循 Attention 到 FFN 的结构,并通过可学习的调节参数控制去噪过程。DiT 的优势在于可扩展性强、易于与预训练视觉模型结合。
WALL-WM 的视频塔和动作塔都基于 DiT 架构。理解 DiT 的工作原理、层间耦合机制、以及如何在预训练基础上添加跨视图注意力等扩展,对于把握 WALL-WM 的架构设计至关重要。
事件-中心学习 (Event-Centric Learning)
事件-中心学习将语义事件(如 reach, grasp, lift, move, place)而非固定时间窗口作为学习的基本单元。每个事件在语言中有自然描述,在视频中可观察到完整的空间-时间演化,在动作上可执行为控制序列。这种方法的关键优势是保持了三种模态在语义和时间上的一致性:语言描述事件的意图,视频展示事件的执行过程,动作实现事件的具体控制。事件边界由底层可执行行为的变化决定,而非外部的时钟。
这是 WALL-WM 的核心创新点。理解事件-中心学习的理念,以及它如何解决传统固定时间块方法中的粒度失配问题,是理解整篇论文的关键。
多视图几何与相机校准
多视图几何研究从不同相机视角观测同一场景时图像之间的关系。核心概念包括对极几何、基础矩阵、本质矩阵,它们描述了点在不同视图间的对应约束。在机器人系统中,通过相机内参(焦距、主点)和外参(位置、姿态)可以将像素坐标映射到三维空间。跨视图一致性要求同一物理点在不同相机中的投影应满足几何约束。WALL-WM 使用 Camera RoPE(可学习的旋转位置编码)来标识相机身份,无需运行时校准。
WALL-WM 支持多视图视频输入,并通过跨视图注意力模块进行信息融合。理解多视图几何的原理,以及如何在不泄露校准信息到推理阶段的情况下训练跨视图一致性,对于理解模型的设计很重要。
研究动机
现有的 World Action Models (WAMs) 和 Vision-Language-Action (VLA) 模型普遍采用固定长度动作块的预测范式。具体来说,模型给定当前观测和语言指令,预测未来 H 步的动作序列。这种块中心的表述虽然实现方便,但存在根本性的粒度失配问题。语言描述的是语义目标和事件(如将杯子放到桌子上),其时间尺度是事件级的;视觉信息通过连续的场景动态演化(帧级时间尺度);动作则在控制级时间尺度上操作,对接触、时序和微小扰动高度敏感。将这三者强行塞入同一个固定长度的预测窗口,会导致 VLA 训练退化为短视的相关性拟合。这不仅未能充分利用预训练的视觉-语义先验,还可能用块特定的动作捷径覆盖这些先验,削弱了组合性和长视距泛化能力。具体数据上,如果固定块长度设为 50 步,但一个完整的抓取事件可能需要 30-80 步不等,模型要么切断事件中途(语义不完整),要么在一个块中包含多个事件(语义混杂)。
本文的目标是WALL-WM 的具体目标是将视频-动作学习从块中心优化转移到以事件为锚点的视觉-语言-动作预训练。核心思路是使用语义连贯的动作事件作为学习的原子单元,这些事件在语言中有自然描述,在视频中可观察到完整演化,在动作上可执行为控制序列。目标包括:将语言、视频和动作在同一个语义事件单位上对齐;从视频基础模型继承语义和时间结构,同时获得可执行性、接触敏感性和因果接地;支持两种互补的推理模式:事件模式(在事件空间中展开,支持变长执行)和统一模式(使用 Staircase 解码在固定长度块上进行推理,同时保持梯度连续的 VLA 路径);通过大规模数据生态系统和 Muon 优化器基础设施实现可扩展的预训练。
与已有工作不同的是,本文的独特切入角度是重新思考视频-动作学习的原子单元。现有工作大多将固定长度块视为理所当然,关注的是如何在块内进行更好的动作预测(如改进动作 token 化、添加扩散专家等)。WALL-WM 则从第一性原理出发,提出根本问题:预训练的视觉-语义先验应该在哪个粒度上变成可执行的?论文指出,固定块不是语言、视频和动作共享的自然对象——它可能切断语义行为的中途,或合并多个行为到一个目标中,或需要历史上下文才能确定块的含义。WALL-WM 的创新在于用动作接地的语义事件替代固定块,这是一个时间连贯的可执行行为片段(如 reaching, grasping, lifting, moving, placing),在语言中可表达,在视频中可观察,在动作上可实现。这种以事件为中心的训练目标是一个先验保留的提升路径:从视频基础模型到可执行的世界-动作模型。
核心方法
WALL-WM 整体上实现了一个以事件为中心的世界-动作建模框架。核心思想是构建一个层耦合的视频-动作去噪器,给定当前多视图观测和下一事件指令,联合去噪未来视频潜变量和相应的末端执行器轨迹。训练采用两个阶段:第一阶段仅训练视频 DiT,在事件潜变量上使用 Wan 风格的 v-prediction flow matching,将继承的 Wan 先验调整为具身、多视图的事件-未来预测器;第二阶段冻结视频 DiT,仅训练动作塔,在相同的以事件为中心的视频-动作对上使用动作 flow matching 来预测对齐的末端执行器轨迹。推理时支持两种模式:事件模式在事件空间中展开,由 VLM、人类或代理提出下一事件描述,WALL-WM 执行对应的变长视频-动作片段;统一模式使用带有 Staircase 解码的 VLM 来生成事件结构的潜在推理,指导局部块预测,同时保持梯度连续的 VLA 路径。
WALL-WM 的核心创新点是将固定长度动作块替换为动作接地的语义事件作为训练和推理的基本单元。与已有方法的本质区别在于:粒度对齐:语言描述事件、视频展示事件、动作实现事件,三者保持语义和时间一致性;先验保留:事件级训练保持从视频基础模型继承的 caption-to-video 结构,而不是用短视动作相关性覆盖它;可执行因果性:预测目标具有清晰的时间支撑,但持续时间跟随任务而非固定时钟;双重推理模式:同一事件预训练主干支持事件模式(变长事件执行)和统一模式(固定长度块推理),保持梯度连续路径。关键在于将训练问题本身建立在事件级,而不仅仅将事件作为辅助条件。
方法步骤详情
方法包含四个主要步骤:多视图视频事件建模:视频塔继承 Wan 单视图 DiT 并扩展到多视图、多具身视频生成。在每个 DiT 块中添加跨视图分支,输出投影器初始化为零。引入 Camera RoPE 给每个相机一个可学习的、无需校准的身份。训练时使用跨视图几何遮挡来强化跨视图几何一致性。使用 Wan 风格的 v-prediction flow matching 目标。事件中心动作动力学建模:动作塔是动作 DiT,与视频塔深度相同。每层动作 token 通过跨注意力读取匹配的 DiT 层的视频特征。使用相对姿态动作流:所有非锚点 token 编码相对于观测姿态的末端执行器位移。在非对称 1-to-Nd 映射下,冻结视频流在锚点 s=45(50 步去噪调度),每个动作步骤都从这个单一锚定视频前向中读取交叉注意力键/值。语言引导推理:使用 Qwen3.5-9B 主干和 Staircase 潜在解码。Staircase 解码在继电器深度 Nr 处划分 Transformer,下层编码共享的视觉-语言接地特征,上层渐进专业化为不同的推理步骤。通过冻结的潜在到文本重构目标监督。训练配方:视频预训练、动作预训练、VLM 文本条件适配、Staircase 蒸馏、可选的下一块适应。
技术新颖性
WALL-WM 的技术新颖性体现在多个层面:架构层面:首次系统性地提出事件-中心的世界-动作建模范式,将语言、视频、动作在语义事件粒度上对齐。层耦合的视频-动作去噪器设计,动作塔单向交叉注意力读取视频特征,保持视频先验不被破坏。训练层面:事件级配对的训练目标,而非固定长度块。非对称 1-to-Nd 时间步映射,在训练时用单一锚定视频前向支持所有动作去噪步骤,提高吞吐量。推理层面:Staircase 潜在解码,通过继电器深度划分实现推理步骤的并行生成,避免传统自回归解码的串行瓶颈。支持事件模式和统一模式两种互补的推理路径,共享同一事件预训练主干。数据层面:构建包含网络视频、自中心视频、机器人自由 UMI 风格录制、异构遥操作数据的数据生态系统,四层层次化标注,视觉-语言聚类和动作聚类的平衡采样,接触富集随机初始化的恢复数据。基础设施层面:DMuon(分布式 Muon 优化器)解决牛顿-舒尔茨迭代的分布式开销;定制核库将可融合模式合并到复合核中;多事件序列打包将多个事件打包到单个长序列中训练,避免每步加载完整片段;分布匹配蒸馏减少去噪步数,FP8 量化降低每步计算成本,将推理延迟降低到 10Hz 满足实时控制需求。
实验结果
实验分为三个主要部分。第一部分是具身视频生成评估,使用 200 个分布内任务和 50 个分布外任务的评估集。WALL-WM 在运动质量、语义一致性和物理合理性等具身相关维度上持续优于 Wan 系列基线。具体数据:Motion Quality 指标,WALL-WM 达到 0.484,显著高于 Wan2.1-1.3B 的 0.199 和 Wan2.2-5B 的 0.418;Physical Plausibility 指标中,Interaction Quality 达到 0.434,远高于基线的 0.219-0.226。这些增益表明大规模具身训练将继承的视频先验转变为更强的物理先验,在生成过程中保持连贯的运动和接触动力学。第二部分是真实机器人评估,在内部开发的高性能桌面双臂机器人平台上进行,包含四个互补的评估套件:直接操作、指令级推理、精细接触控制和杂乱场景泛化。使用 Task Progress(0-100 分)作为主要指标,能够捕捉部分完成和中间进展。在 Diverse Manipulation 上,事件模式 WALL-WM 达到平均 75.86 分,对比 WALL-WM-U-Scratch 的 63.00 分、π0.5 的 55.64 分、DreamZero 的 39.97 分、LingBot-VA 的 29.71 分。在 Reasoning Manipulation 上,WALL-WM 达到 71.60 分,显著优于基线。在 Dexterous Manipulation 上,WALL-WM 达到 32.00 分,虽然绝对分数较低(反映精细接触控制的挑战),但仍优于大多数基线。在 Generalization 上,WALL-WM 达到 53.75 分,远超 WALL-WM-U-Scratch 的 18.50 分,表明事件预训练能够在杂乱场景和随机化指令中重用预训练的事件知识。第三部分是消融实验,评估事件条件执行和跨视图建模的贡献。在 Reasoning Manipulation 上,预训练事件模式 WALL-WM 相比预训练统一基线(无 VI-SA)从 32.6 分提升到 71.6 分;在 Generalization 上从 22.0 分提升到 53.75 分。这些结果支持预训练在与跨视图交互和变长事件执行结合时最有效的观点。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| Embodied Video Generation | Motion Quality (Dynamic Flow Degree) | 0.484 | Wan2.2-5B: 0.418 | +15.8% |
| Embodied Video Generation | Semantic Consistency (Interaction Quality) | 0.434 | Wan2.2-5B: 0.226 | +92.0% |
| Diverse Manipulation (Robot) | Average Task Progress | 75.86 | π0.5: 55.64 | +36.3% |
| Reasoning Manipulation (Robot) | Average Task Progress | 71.60 | π0.5: 56.40 | +27.0% |
| Generalization (Robot) | Average Task Progress | 53.75 | WALL-WM-U-Scratch: 18.50 | +190.5% |
局限与改进
作者承认的局限性包括:数据构建配方仍然依赖大规模时间接地和细粒度标注来暴露事件结构。当前方法需要 Task/Subtask/Action/Segment 四层层次化标注,这带来了标注负担。推理延迟:虽然通过蒸馏和 FP8 量化将推理延迟降低到 10Hz,但大型模型仍然面临实时部署的挑战。Distillation 和 quantization 虽然有效,但需要仔细平衡以避免性能下降。平台对齐优势:真实机器人评估在内部开发的具身套件上进行,WALL-WM 模型在从或与该平台对齐的数据上接受大规模预训练。这带来了评估环境优势,无法完全消除。调优资源不平等:与 LingBot-VA 和 DreamZero 等资源密集系统的比较不可避免地涉及不等量的方法特定调优。Dexterous Manipulation 套件上的绝对分数仍然较低(平均 32.00 分),表明精细接触控制、精确末端执行器对齐、谨慎接触时序和窄容差插入仍然是高度挑战性的任务。补充观察的局限性:事件模式推理需要可靠的下一事件描述来源。如果 VLM 或人类无法准确预测下一事件,性能会下降。跨视图几何遮挡仅在训练时使用,推理时依赖 Camera RoPE。如果测试时的相机配置与训练时的分布相差太大,跨视图一致性可能受到影响。模型大小与泛化的权衡:虽然较大模型在长视距状态跟踪和组合指令接地方面表现更好,但计算和存储成本也更高。
独立分析的弱点
独立分析的弱点包括:标注依赖性:当前方法需要密集的四层层次化标注,这在大规模数据收集时成为瓶颈。虽然论文提到正在探索自监督预训练以减少标注负担,但当前实现仍依赖手动标注。改进方向:开发自监督事件发现算法,通过视觉-动作流或语言-视觉对齐自动检测事件边界,减少对人工标注的依赖。事件模式推理的可靠性:事件模式需要一个可靠的下一事件描述来源(VLM、人类或代理)。在复杂的、未见过的场景中,VLM 可能生成不准确或不合理的下一事件描述,导致执行失败。改进方向:增强事件描述的验证机制,允许在执行过程中基于实时观测调整事件描述;开发多层次事件规划,既有高层事件目标,也有低层执行监控。跨视图泛化:虽然训练时使用了多视图数据和跨视图注意力,但如果测试时的相机配置(视角数量、相机类型、位置关系)与训练分布差异较大,性能可能下降。改进方向:增加训练数据的相机配置多样性,包括更多异构多视图设置;开发领域自适应技术,在少量目标平台数据上快速适应新的相机配置。接触敏感任务的性能:Dexterous Manipulation 套件上的绝对分数仍然较低(32.00 分),表明模型在精细接触控制方面仍有不足。改进方向:增加接触富集的恢复数据,特别是在接触事件附近的局部覆盖;开发专门的接触预测头,显式建模接触状态和力信息;使用触觉或力传感器数据(如果可用)来增强接触感知。长视距任务的累积误差:在需要长时间、多阶段执行的任务中,小误差可能累积导致最终失败。改进方向:引入中间检查点和错误恢复机制;在训练中加入更多的失败恢复和非标称轨迹数据,增强鲁棒性。
未来方向
作者提出的未来方向包括:自监督预训练:当前数据构建配方仍然依赖大规模时间接地和细粒度标注来暴露事件结构。中心方向是视觉、语言和动作上的自监督预训练,其中事件边界不作为密集标注提供,而是由训练目标本身捕获。这将部分减少学习显式标注的难度,将注释负担转移到可扩展的训练计算中。蒸馏和压缩:通过分布匹配蒸馏减少去噪步数,通过 FP8 量化降低每步计算成本。进一步优化包括更高效的蒸馏算法、更激进的量化策略(如 4-bit 量化)、以及推理时的投机执行或流式执行。下一代 WAM 预训练范式:将自回归单元从固定 token 或块重新定义为事件级潜在原语,可能支持下一代 WAM 预训练范式。结合更深层次的基础设施优化,目标是使预训练更高效,减少对手工时间标签的依赖,同时保持物理接地。更高效的评估协议:探索更高效的预训练规模迭代评估协议,包括更细粒度的每事件评分、密集进展度量、评估期间的人类修正和重新标注,以及从生成的视频、动作轨迹和中间状态预测机器人性能的通用评估模型。这种评估器不需要完美替代真实硬件,只需要与下游机器人性能保持可靠的正相关,以便在最有希望的候选者上在真实机器人上验证之前筛选模型和数据决策。事件级指令泛化作为代理接口:将每事件文本泛化视为具身控制与更广泛代理智能之间的桥梁。高层任务推理由通用代理完成,而具身模型专注于全域动作-文本指令泛化,在语义事件上进行。这解决了是否高频操作模型应该在高层任务推理上与前沿多模态模型竞争的边界问题。系统级优化:继续优化基础设施,包括更高效的 DMuon 实现、更广泛的核融合模式、动态批处理策略等,以支持更大规模的预训练。
复现评估
复现评估:开源情况:论文提供了代码链接,但截至评估时不确定代码的完整性和可用性。数据:论文描述了大规模数据生态系统,包括网络视频(1.2M-clip OpenVID 切片)、自中心人类动作视频(如 Ego4D、EPIC-KITCHENS)、机器人自由 UMI 风格录制(XRZero-GO)、异构机器人数据集和自收集遥操作数据。然而,这些数据的公开访问可能受限,特别是自收集的数据和内部开发的具身平台数据。算力:模型大小从低于 10B 参数到数十亿参数级别。训练使用 DMuon 优化器、多事件序列打包、定制核库等基础设施优化,需要大规模 GPU 集群。具体算力需求论文未详细说明,但基于模型规模和数据量,推测需要数百到数千 GPU 小时。难度:中等偏难。论文提供了详细的训练配方、架构设计、训练超参数(如去噪调度、锚点选择、损失权重等),但大规模数据和算力需求是主要障碍。基础设施优化(DMuon、定制核库等)的实现细节可能需要额外工程努力。评估协议:论文提供了详细的任务定义、多视图观察流、语言指令、场景随机化协议和 Task Progress 评分标准,这有助于公平比较和复现。然而,真实机器人评估需要特定的硬件平台(内部开发的高性能桌面双臂机器人),这在其他实验室中可能不可用,限制了直接复现。总的来说,虽然论文提供了相当多的技术细节,但大规模数据和算力需求,以及专用硬件平台的要求,使得完全复现实验具有挑战性。部分组件(如事件中心训练范式、架构设计)可以在较小规模数据上验证概念,但完整复现可能需要大量资源。
论文图表
该图左侧展示了模态对齐的示意景观,横轴是语义抽象度,纵轴是空间-时间精度。文本提供粗粒度的语义对齐,视觉提供密集的空间-时间接地,动作需要最精细的局部精度。触觉-力输入(如果可用)被视为可选的接触富集信号而非必需模态。右侧展示了 WALL-WM 在操作任务性能和视频生成指标上的明显优势,表明事件中心预训练改善了可执行控制和未来观察建模。底部展示了代表性真实机器人任务快照,提供了物理任务的示例。
这张图对理解论文至关重要,因为它直观地展示了 WALL-WM 的核心设计哲学——语言、视觉、动作在不同粒度上对齐,以及这种事件中心方法如何同时改善控制性能和世界建模。它为整篇论文的动机和方法提供了高层概念框架。