arXiv · 2603.16666 · Mar 2026 · IIIS Tsinghua University · Galaxea AI · Zotero NZPAHBVG · PDF RUBWIY2Q
Fast-WAM: Do World Action Models Need Test-time Future Imagination?
Tianyuan Yuan, Zibin Dong, Yicheng Liu, Hang Zhao
受控消融实验 Video Co-training 190ms 实时推理 Flow Matching RoboTwin · LIBERO 6B params · Wan2.2
§ 01

核心问题:WAM 的收益从哪里来?

这一节解决什么

世界动作模型(WAM)比普通 VLA 更强,普遍认为是因为它能"想象未来"再执行。但这个假设从来没被严格验证过——也许 WAM 的收益只是来自训练时视频联合学习塑造了更好的特征,而不是推理时真正"展开未来"。Fast-WAM 把这两个因素精确解耦,得出一个出人意料的结论:推理时的未来想象几乎可以去掉,训练时的视频共同训练才是关键。结果:速度提升 4× 以上,性能几乎不变。

绝大多数现有 WAM 遵循 imagine-then-execute 范式:先迭代去噪生成未来视频帧,再以此为条件预测动作。这带来两个麻烦:① 推理延迟高(需要完整跑视频去噪);② 让我们搞不清楚"到底是哪一步在起作用"。Fast-WAM 提出一个干净的问题:

核心研究问题

WAMs 需要在测试时显式生成未来观测,还是它们的收益主要来自训练过程中对未来的建模?

为了回答这个问题,论文设计了一套受控实验:用同一个骨干(Wan2.2-5B),在同一个训练框架下,实例化多种"推理时是否生成视频"的变体,控制住所有其他变量,只改变这一个因素。

核心结论(先看结果)

① 训练时的视频共同训练是主要驱动力:去掉 video co-training → 性能下降 8pp(RoboTwin)到 4pp(LIBERO),实机任务从 80% 跌到 10%。
② 推理时的视频生成贡献有限:Fast-WAM(无推理视频)vs imagine-then-execute 变体 → 差距 ≤ 1.2pp,几乎可忽略。
③ Fast-WAM 速度极快:190ms vs IDM 的 810ms,快 4× 以上,同时无需大规模 embodied pretraining。

展开原文 · 核心假设

"The effectiveness of WAMs may stem from two distinct sources: (i) the video prediction objective during training, which can encourage the model to learn physically meaningful latent representations, and (ii) explicit future generation during inference, which may provide additional foresight for action prediction."

— §1 Introduction, p.1
🧠 理解检查
Fast-WAM 的核心发现是什么?
§ 02

三种 WAM 范式:受控实验设计

这一节解决什么

上一节提出了问题,这一节是实验设计的核心:怎么才能"公平"地比较训练时视频目标 vs 推理时视频生成的贡献?答案是:在同一个骨干、同一个训练框架下,实例化三种不同推理行为的变体,加上一个"去掉视频 co-training"的对照组,总共四个变体,两两对比。

Figure 1: Three WAM paradigms
Fig 1 · p.2 Three representative WAM paradigms: (A) Joint future video & action prediction, (B) Future video prediction followed by action prediction (IDM), (C) Fast-WAM: action prediction without attending future video at inference. 三种 WAM 范式的训练(上)和推理(下)行为对比。蓝色=视频 latent token,黄色=动作 token。Fast-WAM 推理时完全不生成未来视频,直接单次前向预测动作。

四个实验变体

FAST-WAM-JOINT · 变体 A
联合视频+动作去噪
视频和动作 token 在同一 Transformer 中联合去噪,双向 attention 共享。训练和推理都需要视频生成。代表"最强的 imagine-then-execute"设计。
FAST-WAM-IDM · 变体 B
先生成视频,再预测动作
两阶段:先用视频 DiT 生成完整未来视频,再以视频为条件预测动作。最接近原始 WAM 理念,但推理最慢(810ms)。
FAST-WAM (OURS) · 变体 C
只保留训练时视频目标
训练时保留视频 co-training,推理时完全跳过视频生成,当前帧表征只提取一次,再进行 10 步动作去噪。190ms,4× 更快,是本文的核心贡献。
FAST-WAM W.O. VIDEO CO-TRAIN · 对照组
去掉视频 co-training
保持 Fast-WAM 推理架构不变,去掉训练时的视频 co-training 损失。这是纯净的基准:揭示视频 co-training 本身的贡献有多大。结果:性能大幅下降。
实验设计的精妙之处

通过比较"Fast-WAM vs (Joint/IDM)"可以测量推理时视频生成的贡献(其他一切相同)。
通过比较"Fast-WAM vs (w.o. video co-train)"可以测量训练时 video co-training 的贡献(推理行为相同)。
两个对比都在同一个骨干、相同超参下,控制变量非常干净。

🪜 四个变体的核心区别
1 / 4

变体 A:Fast-WAM-Joint(联合去噪)

训练和推理都做联合去噪。视频和动作 token 共享 attention,动作生成全程"看到"视频去噪过程。推理需要多步联合去噪,延迟约 580ms(联合去噪,比 IDM 的 810ms 快)。

给第一次接触这个领域的读者

输入是什么:随机噪声,以及当前画面、语言指令或机器人状态等条件信息。噪声只是生成过程的起点,不是机器人真的要执行的动作。

这一步究竟做什么:训练和推理都做联合去噪。视频和动作 token 共享 attention,动作生成全程"看到"视频去噪过程。推理需要多步联合去噪,延迟约 580ms(联合去噪,比 IDM 的 810ms 快)。

输出到哪里:一组比原始输入更紧凑的内部特征;它不是最终动作或画面。该变体直接输出动作;IDM 是另一套对照方法,不是它的下游。

为什么不能省略:它把未经处理的原始问题变成后续模块可以计算的输入,是整条链路的起点。

术语先翻译:token:模型处理信息的基本小单元;在这里通常是一小段图像、语言、动作或其压缩向量;diffusion(扩散生成):从随机噪声出发,经过多次去噪得到结构化动作或图像;WAM(世界—动作模型):既预测动作,也显式或隐式预测动作会把未来世界变成什么样。

变体 B:Fast-WAM-IDM(推理最慢)

两段串行:先跑 5B 视频 DiT 完整去噪生成未来帧(多步,最慢部分),再把生成帧送入动作 DiT。串行导致延迟叠加,810ms。但这也是最接近"视频作为显式推理媒介"这一理念的变体。

给第一次接触这个领域的读者

输入是什么:来自上一步“变体 A:Fast-WAM-Joint(联合去噪)”的产物:一组比原始输入更紧凑的内部特征;它不是最终动作或画面。本步骤还会按论文设置读取当前条件、时间步或训练信号。

这一步究竟做什么:两段串行:先跑 5B 视频 DiT 完整去噪生成未来帧(多步,最慢部分),再把生成帧送入动作 DiT。串行导致延迟叠加,810ms。但这也是最接近"视频作为显式推理媒介"这一理念的变体。

输出到哪里:一个或多个候选未来、动作或轨迹,以及必要的中间状态。这个结果会交给下一步“变体 C:Fast-WAM(推理最快)”继续处理。

为什么不能省略:它承担上下两步之间的接口;缺少它,前一步产生的信息无法以正确格式或正确含义传到下一步。

术语先翻译:diffusion(扩散生成):从随机噪声出发,经过多次去噪得到结构化动作或图像;DiT:用 Transformer 充当扩散/流模型的去噪或速度预测网络;WAM(世界—动作模型):既预测动作,也显式或隐式预测动作会把未来世界变成什么样。

变体 C:Fast-WAM(推理最快)

推理时只用第一帧干净 latent 做一次单向前向,提取世界表征 z(o,l),再跑 10 步动作去噪。没有视频生成,没有多步视频去噪,190ms。训练时保留视频 co-training 损失,让 video DiT 学到好表征但推理时不用它生成未来。

给第一次接触这个领域的读者

输入是什么:当前观测和语言条件;Fast-WAM 不接收 IDM 的输出,也不读取未来视频 token。

这一步究竟做什么:推理时只用第一帧干净 latent 做一次单向前向,提取世界表征 z(o,l),再跑 10 步动作去噪。没有视频生成,没有多步视频去噪,190ms。训练时保留视频 co-training 损失,让 video DiT 学到好表征但推理时不用它生成未来。

输出到哪里:一组比原始输入更紧凑的内部特征;它不是最终动作或画面。该变体输出动作;w.o. video co-train 是训练消融,不是下游步骤。

为什么不能省略:它承担上下两步之间的接口;缺少它,前一步产生的信息无法以正确格式或正确含义传到下一步。

术语先翻译:latent(隐表示):模型内部压缩后的数字特征,人看不见,但比原始图像或动作更便于计算;diffusion(扩散生成):从随机噪声出发,经过多次去噪得到结构化动作或图像;DiT:用 Transformer 充当扩散/流模型的去噪或速度预测网络。

对照组:w.o. video co-train(训练缺失)

推理行为和 Fast-WAM 完全一样(也是单次前向),但训练时去掉了视频 co-training 损失 L_vid。纯粹的动作监督。这个变体用来回答:如果没有视频联合训练,Fast-WAM 的架构本身值多少分?答案:大幅下降,说明 video co-training 是核心。

给第一次接触这个领域的读者

输入是什么:来自上一步“变体 C:Fast-WAM(推理最快)”的产物:一组比原始输入更紧凑的内部特征;它不是最终动作或画面。本步骤还会按论文设置读取当前条件、时间步或训练信号。

这一步究竟做什么:推理行为和 Fast-WAM 完全一样(也是单次前向),但训练时去掉了视频 co-training 损失 L_vid。纯粹的动作监督。这个变体用来回答:如果没有视频联合训练,Fast-WAM 的架构本身值多少分?答案:大幅下降,说明 video co-training 是核心。

输出到哪里:参数得到更新的模型,或能供下一轮训练使用的新监督信号。这是图中这条链路的最终产物,随后会进入真实执行、模型更新或指标评测。

为什么不能省略:前面的数据或分数本身不会自动改变模型;必须通过损失函数把误差信号变成参数更新。

术语先翻译:WAM(世界—动作模型):既预测动作,也显式或隐式预测动作会把未来世界变成什么样。
§ 03

Fast-WAM 架构:MoT + 结构化注意力掩码

这一节解决什么

上一节定义了三种范式,这一节讲 Fast-WAM 本身的具体架构。关键设计目标:训练时让视频 DiT 和动作 DiT 共同学习(共享表征),但阻止动作 token 在训练时看到未来视频 token——这样推理时才能安全地移除未来视频,不影响动作预测的信息依赖关系。类比:厨师在训练时看过很多优质料理视频(学了感觉),但做菜时不需要边看视频边操作。

Figure 2: Fast-WAM architecture
Fig 2 · p.5 Fast-WAM model architecture (left) and structured attention mask used to disentangle video co-training from action generation (right). 左:Fast-WAM 架构(Wan2.2-5B 视频 DiT + 1B 动作 Expert DiT,MoT 共享 attention);右:训练时(上)和推理时(下)的结构化注意力掩码。推理时完全去掉未来帧 token。

关键设计:为什么动作 token 不能看未来视频 token?

这是 Fast-WAM 能在推理时安全跳过视频生成的根本原因。如果训练时动作 token 可以 attend 到未来视频 token,那么推理时去掉未来视频 token 就会造成训练-推理分布偏移——模型在推理时缺少了它依赖的信息,性能会崩溃。

结构化掩码的三条规则

① 当前帧 f₀(干净 latent):所有 token 都可以看到它——它是整个模型的"视觉锚点"
② 未来帧 f₁..fₙ(加噪 latent):只有视频 token 互相看,动作 token 禁止访问
③ 动作 token a₁..aₙ:动作 token 互相可见(双向 attention)+ 可看 f₀,通过 video DiT 共享表征间接获得视频动态信息

BackboneWan2.2-5B(视频 DiT + T5 文本编码器 + 视频 VAE)
Action Expert基于 Wan2.2 架构,hidden dim 缩减至 d_a=1024,参数量 1B,动作 horizon H=32
总参数量5B + 1B = 6B
视频帧率时间维度 4× 下采样 → 每 action chunk 对应 9 个视频帧
推理去噪步数10 步(CFG scale=1.0,logit-normal 噪声调度)
推理延迟190ms(vs Fast-WAM-IDM 810ms,vs π₀.₅ ~180ms)
训练设备NVIDIA RTX 5090D V2 32GB GPU
优化器AdamW,lr=1×10⁻⁴,weight decay=0.01,cosine annealing
§ 04

训练目标:联合 Flow Matching

这一节解决什么

架构确定了,这一节讲怎么训练。Fast-WAM 用的是 flow matching——把视频 latent 预测和动作块预测统一进同一个速度场框架。直觉:flow matching 是"在噪声中学习走直线",训练时同时学走向视频目标和走向动作目标,推理时只走动作方向。

Flow Matching 基础

给定目标变量 $y$(可以是动作块或未来视频 latent),构造从纯高斯噪声 $\epsilon$ 到 $y$ 的插值轨迹:

$$y_t = (1 - t)\,y + t\,\epsilon, \qquad t \in (0, 1),\quad \epsilon \sim \mathcal{N}(0, I)$$ Eq. 5 · 插值样本构造 · §3.2, p.5
逐项拆解
$y_t$
时间步 $t$ 处的插值样本:$t=0$ 时等于干净目标 $y$,$t=1$ 时等于纯噪声 $\epsilon$。训练时从 $t \in (0,1)$ 均匀(或 logit-normal)采样
$(1-t)y + t\epsilon$
从 $y$ 到 $\epsilon$ 的线性插值——flow matching 的路径是直线(对比 DDPM 的曲线路径),所以理论上更高效

模型学习预测速度场(velocity field),使得顺着速度场走能从噪声还原到目标:

$$\mathcal{L}_\text{FM}(y) = \mathbb{E}_{y,\,\epsilon,\,t}\!\left[\left\|f_\theta(y_t, t, o, l) - (\epsilon - y)\right\|_2^2\right]$$ Eq. 6 · Flow Matching 损失 · §3.2, p.5
逐项拆解
$f_\theta(y_t, t, o, l)$
模型(视频 DiT 或动作 DiT)在时间步 $t$、输入噪声样本 $y_t$、条件观测 $o$、语言指令 $l$ 下预测的速度向量
$\epsilon - y$
目标速度:从干净目标 $y$ 到噪声 $\epsilon$ 的方向(即"往噪声方向走")。去噪时反方向走,即 $y - \epsilon$
$\|\cdot\|_2^2$
MSE loss,让预测速度尽量匹配真实速度

联合目标

$$\mathcal{L}_\text{act} = \mathcal{L}_\text{FM}(a_{1:H}), \qquad \mathcal{L}_\text{vid} = \mathcal{L}_\text{FM}(z_{1:T})$$ Eq. 7 & 8 · 动作和视频分支的损失 · §3.2, p.5
两条 flow 损失为何必须分开报告
$a_{1:H}$
未来动作 chunk;动作 flow 直接决定策略输出和控制成功率。
$z_{1:T}$
未来视频 latent;视频 flow 提供动作后果和时空动力学监督。
$\mathcal L_{act}$
要求 action expert 恢复动作分布。
$\mathcal L_{vid}$
要求 video expert 恢复视觉未来;两项同时优化不代表权重和梯度天然平衡,需通过消融确认视频分支确实帮助动作。
$$\mathcal{L} = \mathcal{L}_\text{act} + \lambda\,\mathcal{L}_\text{vid}$$ Eq. 9 · 总训练损失 · §3.2, p.5
逐项拆解
$a_{1:H}$
动作块(ground-truth),长度 $H=32$。$\mathcal{L}_\text{act}$ 是动作 Expert DiT 的训练损失
$z_{1:T}$
未来视频帧的 VAE latent token($T=9$ 帧)。$\mathcal{L}_\text{vid}$ 是视频 DiT 的训练损失,即 video co-training
$\lambda$
视频 co-training 的权重,平衡两个损失的量级。论文未指定具体值(通过消融确定)
w.o. video co-train 对照组
把 $\lambda = 0$:等同于去掉 $\mathcal{L}_\text{vid}$,只留 $\mathcal{L}_\text{act}$。这就是对照组实验的操作,完全相同的架构和推理流程,只是训练时没有视频监督。
Video Co-training 的作用机制(直觉)

$\mathcal{L}_\text{vid}$ 强迫视频 DiT 学会"物理世界怎么演变"——物体运动轨迹、接触力学、形变规律。这些知识通过 MoT 共享 attention 渗入到动作 Expert 的表征里(通过 f₀ 的 latent)。即使推理时不显式生成未来,这些视觉物理先验已经编码在 z(o,l) 里,帮助动作预测更稳健。

🧠 理解检查
「Fast-WAM w.o. video co-train」对照组和 Fast-WAM 的区别在哪里?
§ 05

实验结果:数据支持核心结论

这一节解决什么

前几节都是设计和理论,这一节让数据说话。三个评估平台:RoboTwin 2.0(50任务双臂仿真)、LIBERO(4套件仿真)、Galaxea R1 Lite 实机毛巾折叠。关键发现:Fast-WAM 不需要 embodied pretraining,在三个平台上表现都强,核心消融验证了"训练时 video co-training 远比推理时视频生成更重要"。

实机任务:Galaxea R1 Lite 毛巾折叠

Figure 3: Towel folding task
Fig 3 · p.7 Real-world towel-folding task on the Galaxea R1 Lite platform. Folding a deformable object requires long-horizon planning and precise closed-loop manipulation. Galaxea R1 Lite 平台上的实机毛巾折叠任务。可形变物体 + 长时序规划,是评估 policy 执行效率和成功率的高难度真实环境基准。

速度 vs 性能:核心权衡可视化

Figure 4: Success rate vs completion time and latency
Fig 4 · p.9 Real-world results: success rate vs. average completion time (left, upper-left is better) and inference latency comparison (right). 左:实机任务成功率 vs 完成时间散点图(左上 = 更好);右:各变体推理延迟比较。Fast-WAM 用 190ms 实现了接近最优的实机性能,是 IDM 延迟的 1/4。

RoboTwin 2.0 结果(Table 1)

Fast-WAM(Ours)190ms
91.8%
Fast-WAM-Joint580ms
90.6%
Fast-WAM-IDM810ms
91.3%
w.o. video co-train190ms
83.8%
LingBot-VA(有预训练)
92.2%
π₀.₅(有预训练)
79.8%

数值为 clean + randomized 双设置平均成功率。Fast-WAM 无需 embodied pretraining。

关键对比数字

Fast-WAM vs Fast-WAM-Joint:91.8 vs 90.6(差距 1.2pp)← 推理时视频生成的贡献
Fast-WAM vs Fast-WAM-IDM:91.8 vs 91.3(差距 0.5pp)← 推理时更贵的视频生成不值得
Fast-WAM vs w.o. video co-train:91.8 vs 83.8(差距 8pp)← 训练时视频 co-training 的贡献
结论:训练时视频 co-training 的贡献(8pp)≫ 推理时视频生成的贡献(0.5-1.2pp)

LIBERO 结果(Table 2)

Fast-WAM(Ours) Spatial 98.2 / Object 100.0 / Goal 97.0 / Long 95.2 → 平均 97.6%
Fast-WAM-Joint 99.6 / 99.4 / 98.2 / 96.8 → 平均 98.5%
Fast-WAM-IDM 98.8 / 97.8 / 97.8 / 97.6 → 平均 98.0%
w.o. video co-train 89.2 / 99.2 / 95.4 / 90.0 → 平均 93.5%(-4.1pp)
LingBot-VA(有预训练) 98.5 / 99.6 / 97.2 / 98.5 → 平均 98.5%
π₀.₅(有预训练) 98.8 / 98.8 / 98.0 / 92.4 → 平均 97.0%
LIBERO 上的一个有趣发现

w.o. video co-train 在 LIBERO-Object(99.2%)上接近满分,但在 LIBERO-Spatial(89.2%)和 LIBERO-Long(90.0%)上明显更差。这与 video co-training 的直觉一致:空间理解和长时序推理更依赖视觉物理表征,而目标识别(Object)更多依赖语义匹配。§4.3.2, p.8

🧠 理解检查
在 RoboTwin 消融中,以下哪个对比的性能差距最大?
2026-10-23 · 固定 v2 的补充核读

与 VPP2 对照:到底删掉了哪条信息通路?

Fast-WAM 主变体的 action tokens 在训练与推理时都看不到 future-video tokens。当前 clean frame 也不能从未来组读取信息,避免绕路泄漏。推理无需实例化未来视频,但动作仍要做 10 步去噪,不能写成“整个策略一步直出动作”。

paper · 2603.16666v2,§3–4
设置视频和动作如何交互同文延迟
Joint未来视频与动作共同去噪580 ms
IDM先生成未来视频,再条件化动作810 ms
Fast-WAM当前帧表征一次提取 + 10 步动作去噪190 ms
VPP2(另一篇)保留一步未来预测,动作读取未来 KV约 220 ms;硬件信息不足,不与此表同机排名
仿真协议
Benchmark训练与评测结果与边界
RoboTwin 2.0 Full2,500 Clean + 25,000 Random 训练;每任务每条件 100 trialsClean 91.88%,Random 91.78%;Random 已在训练分布内。
LIBERO4 suites、40 tasks、共 2,000 评测试验平均 97.6%;Joint 98.5、IDM 98.0,Fast-WAM 不是每项最高。
分维度 OOD:不要把数据增强评测当零样本迁移
维度原文支持未支持
环境随机化Full 训练同时包含 Clean 与 Random不能代替 Clean-only → Random 的泛化成绩。
未见任务 / 本体评测适配后的基准策略与 R1-Lite 毛巾任务没有仅凭这些结果证明任意未见任务或全身人形零样本迁移。
真机与部署
设置证据限制
Galaxea R1-Lite 叠毛巾60 h 遥操作;Figure 4 中预训练 π₀.₅ 最强,Fast-WAM 比无 video co-train 的 10% 好得多除正文明确数字外,其余点为原图估读;未提供完整 trial 数,不推算成功计数。
延迟RTX 5090;190 ms vs IDM 810 ms,约 4.26 倍推理延迟不是关节控制周期,也不是端到端闭环安全性证明。

结论范围:视频辅助训练在本配方中很有价值,不等于所有 WAM 都不需要未来预测。VPP2 改变了视频预训练、数据、模型规模与蒸馏配方,不能把它和这里的消融当成只有一个变量不同。

VPP2 信息通路对照 · 返回六篇串读 · 固定论文版本