返回研究文献库

世界模型 · 2026 · arXiv · CCF 未收录

Fast-WAM: Do World Action Models Need Test-time Future Imagination?

Fast-WAM 保留训练时的未来视频预测,通过注意力掩码让动作只读取当前观测;部署时将 Wan2.2 视频骨干作为单次编码器,再以动作专家进行 10 步去噪,无需生成未来视频。论文在 RoboTwin 2.0 / LIBERO 上取得 91.8% / 97.6% 平均成功率;RTX 5090D V2 上延迟 190 ms,相比 IDM 的 810 ms 加速约 4.26 倍。它是需要机器人示教的世界动作模型,不输出物体 6D pose;真实任务仍有速度与成功率的取舍。

研究方向

世界模型

任务层级

不适用(非物体位姿任务)

物体设定

不适用(策略评测)

输入模态

多相机 RGB 观测任务语言指令

关键依赖

Wan2.2-5B 视频骨干预训练视频 VAE / T51B 动作专家机器人视频与动作示教联合 flow-matching 训练
完整论文元数据 作者、单位、研究方向、数据集、指标、方法标签和阅读状态 展开
作者
Tianyuan Yuan, Zibin Dong, Yicheng Liu, Hang Zhao
作者单位
IIIS, Tsinghua University / Galaxea AI
会议 / 期刊
预印本 · arXiv · CCF 未收录
研究方向
世界模型
开源状态
已开源
阅读状态
粗读:已生成 精读:已生成
任务设定
不适用(非物体位姿任务) · 不适用(策略评测)
关键依赖
Wan2.2-5B 视频骨干预训练视频 VAE / T51B 动作专家机器人视频与动作示教联合 flow-matching 训练

Reading Notes

粗读 / 精读

已生成 src/paper-notes/fast-wam/coarse.md

论文粗读笔记

用于快速判断论文定位、问题设定、核心方法、实验结论和是否值得精读。

论文粗读笔记:Fast-WAM

1. 基本信息

  • 标题:Fast-WAM: Do World Action Models Need Test-time Future Imagination?
  • 作者:Tianyuan Yuan、Zibin Dong、Yicheng Liu、Hang Zhao
  • 机构:清华大学交叉信息研究院(IIIS)、Galaxea AI
  • 年份 / 版本:2026;本文按 arXiv v2(2026-03-23)解读,首版提交于 2026-03-17;已核对来源仍以预印本标注
  • 论文:arXiv:2603.16666 / v2 全文
  • 项目 / 代码:官方项目页 / yuantianyuan01/FastWAM
  • 研究方向:世界动作模型(World Action Model, WAM)、语言条件机器人操作
  • 阅读模式:粗读;资源状态核对于 2026-09-12

2. 一句话定位

Fast-WAM 将“训练时预测未来视频”和“执行动作前生成未来视频”拆开:保留前者作为表征学习监督,推理时只对当前观测做一次视频骨干编码,再生成动作块,在所测任务上保留大部分策略收益并显著降低延迟。

它不是物体 6D pose estimator,不输出物体的旋转和平移,也不属于实例级或类别级位姿估计。本站将其作为位姿感知下游的世界模型 / 具身控制研究收录。

方法输入与输出

项目内容(本站整理)
推理输入当前多相机 RGB 观测与任务语言指令;图像拼接、编码后为动作专家提供上下文。
最终输出连续机器人 action chunk,论文默认包含 32 个动作步;不输出对象 6D pose,也不在默认推理中生成未来视频。
必要条件训练仍需要机器人示教动作和对应未来视频;未来帧是训练监督,不是测试输入。方法未列 CAD、深度、物体 mask 或相机内参为必要模型输入。

2.1 论文图表导读

下表为本站整理;插图采用作者官方项目页提供、与论文对应的原图。

图表回答的问题阅读重点
Figure 1WAM 一定要先生成未来才能执行吗Joint、IDM 与 Fast-WAM 的训练 / 推理路径不同
Figure 2(a)视觉、语言与动作怎样连接Wan 视频 DiT 与动作专家共享注意力交互
Figure 2(b),见精读训练见过未来会不会造成信息泄漏动作始终只能读取当前帧和动作 tokens
Tables 1–2,本站整理收益主要来自哪一部分去掉视频联合训练的损失大于省略测试时未来生成
Figures 3–4真实机器人能否兼顾成功率和效率毛巾折叠成功率、完成时间、推理延迟需分开看

3. 问题设定

维度设定(本站整理)
输入当前多相机 RGB 观测、任务语言指令
输出连续 action chunk,论文设置为 32 个动作步
训练监督机器人示教中的动作及对应未来视频
推理条件当前视觉上下文和语言,不需要真实或生成的未来帧
基础模型预训练 Wan2.2-5B 视频 DiT、视频 VAE、T5 文本编码器,加 1B 动作专家
几何依赖方法未要求 CAD、深度图、物体 mask 或显式相机内参作为模型输入
不应混淆不是 training-free,不是零样本执行,也没有验证未见物体 6D pose 泛化

“不使用具身预训练”是指不依赖额外的大规模机器人预训练阶段。通用视频预训练仍然存在,下游机器人示教仍然需要。

4. 核心动机

WAM 常先想象未来,再据此预测动作;也有方法联合去噪未来视频与动作。两种路线都让视频生成占据推理预算,但其性能增益可能有两个来源:

  1. 训练时学习预测未来,促使骨干保留运动、交互与时序信息。
  2. 推理时实际生成未来,为当前动作提供额外条件。

论文提出的关键问题不是“世界模型有没有用”,而是:为了利用视频训练获得的表征,执行时是否还需要支付完整未来视频去噪的成本?

Fast-WAM Figure 1:联合视频动作去噪、先视频后动作,以及不生成未来视频的三种训练推理范式
原论文 Figure 1:上排比较训练路径,下排比较推理路径。Fast-WAM 仍训练视频预测,但部署时先编码当前帧并缓存视觉条件,再去噪动作。来源:作者官方项目页。图片版权归论文作者及出版方所有,本站仅作论文解读展示。

5. 方法概览

5.1 训练:同一视觉上下文,两个预测目标

多相机图像先拼接,再经视频 VAE 转成 latent tokens。当前帧保持干净,未来视频与动作分别加噪;视频 DiT 和动作专家组成 Mixture-of-Transformer(MoT),所有分支通过 cross-attention 读取 T5 语言特征。

注意力掩码是关键:未来视频与动作都能读取当前帧,但动作不能读取未来视频,当前帧也不能读取未来视频或动作。这样动作分支训练时不会依赖部署时缺失的信息。

两个分支使用 flow matching,联合优化:

L=Lact+λLvid.\mathcal L=\mathcal L_{\mathrm{act}}+\lambda\mathcal L_{\mathrm{vid}}.
Fast-WAM Figure 2a:T5、视频 VAE、Video DiT 和 Action DiT 的联合训练结构
原论文 Figure 2(a):语言提供共同条件,视频预测与动作预测共享当前视觉锚点;未来帧是训练目标,不是默认动作分支的输入条件。来源:作者官方项目页。图片版权归论文作者及出版方所有,本站仅作论文解读展示。

5.2 推理:一次视觉编码,仍然多步动作去噪

部署时不再创建未来视频 tokens。视频 DiT 只处理当前观测一次,缓存其特征,动作专家据此生成 action chunk。

Single-pass 指视频骨干,不代表整个策略单步生成动作。 论文推理仍使用 10 步去噪,CFG scale 为 1.0。延迟下降主要来自省略未来视频生成,并复用不随动作去噪变化的当前帧条件。

6. 实验概览

6.1 最重要的不是排行榜,而是受控变体

以下为根据原文 Tables 1–2 的本站整理,数值为成功率(%),越高越好;RoboTwin 为 Clean / Rand. 的报告均值,LIBERO 为四个 suite 的报告均值。

方法训练预测视频测试生成未来RoboTwin 2.0LIBERO
Fast-WAM是否91.897.6
Fast-WAM-Joint是与动作联合生成90.698.5
Fast-WAM-IDM是先视频、后动作91.398.0
去掉视频联合训练否否83.893.5

去掉视频联合训练,两个基准分别下降 8.0 和 4.1 个百分点;相比之下,Fast-WAM 与两个显式生成未来的变体差距小得多。注意 Fast-WAM 在 LIBERO 略低于 Joint / IDM,不能概括成“去掉未来生成总会提升准确率”。

外部对照中,预训练 LingBot-VA 的 RoboTwin / LIBERO 均值为 92.2 / 98.5;Fast-WAM 与强基线有竞争力,但并非所有项目最优。不同方法的预训练数据与评测细节存在差异,内部消融比跨方法排名更能回答论文的核心问题。

6.2 真实毛巾折叠:效率不是单一数字

真实实验使用 Galaxea R1 Lite,收集 60 小时遥操作示教。下图同时报告任务成功率、平均完成时间和单次推理延迟。

Fast-WAM Figure 4:真实毛巾折叠的成功率与完成时间散点图,以及 190、580、810 毫秒推理延迟比较
原论文 Figure 4:左图越靠左上越好,右图为单张 RTX 5090D V2 32GB 上的延迟。Fast-WAM / Joint / IDM 分别为 190 / 580 / 810 ms。来源:作者官方项目页。图片版权归论文作者及出版方所有,本站仅作论文解读展示。
  • 速度:相对 IDM 的 810 ms,Fast-WAM 的 190 ms 对应约 4.26 倍加速;相对 Joint 约 3.05 倍,不是对所有基线都超过 4 倍。
  • 质量:从散点图读数,Fast-WAM 成功率约 75%,IDM 约 90%;预训练 π0.5\pi_{0.5} 约 100%,也是完成时间最短的方法。
  • 消融:去掉视频联合训练后成功率仅 10%,且完成时间最长。
  • 边界:190 ms 是 action chunk 推理延迟,不是机器人控制频率;约 5.3 次 chunk 推理/秒也不能直接等同于闭环控制 Hz。

作者声称:视频联合训练是主要收益来源,测试时未来想象相对次要。

我的分析:两套仿真和一个真实任务支持这个方向性判断,但真实任务中 IDM 的成功率优势仍值得重视;论文没有证明所有需要长程推理的任务都能省去未来生成。

7. 优缺点

优点

  • 将训练目标和部署计算拆开讨论,给出了能够直接检验设计假设的消融。
  • 注意力结构让训练和推理的动作条件一致,减少了“训练看未来、测试看不到”的错配。
  • 保留强视频先验,同时将推理压缩到当前观测编码和动作去噪,具有明确工程价值。

缺点 / 风险

  • 主体约 6B 参数,省去视频生成不等于轻量模型;32GB GPU 测速也不是最低显存需求证明。
  • 真实机器人只有一个任务,缺少系统的跨场景、遮挡和分布外长程决策分析。
  • 表征是否真正获得物理推理能力没有被直接测量;成功率提升本身不能证明因果世界模型已经学成。
  • “无具身预训练”的模型仍使用较多下游示教,不应与零样本 WAM 或无机器人数据方法混比。

8. 对我的启发

  • 可借鉴:把未来状态、几何或接触预测当作训练辅助目标,同时设计不依赖这些输出的低延迟动作路径。
  • 作为 baseline:适合视频骨干驱动的机器人策略研究;不适合直接与 SAM-6D、FoundationPose 比 ADD(-S) 或 BOP AR。
  • 值得精读:优先看 Figure 2(b)、四个变体和真实实验的速度 / 成功率权衡。
  • 后续问题:哪些状态需要显式未来生成?能否在默认快速执行的基础上,用不确定性触发慢速想象?这属于进一步研究,不是本文已验证结论。

开源版本提示

截至资源核对时,官方仓库已提供训练 / 仿真评测代码、权重和数据链接,README 还增加了推理加速与 Optional IDM 模式。这些是论文后的工程更新,不替换上面的 v2 结果。复现原始权重需特别核对 noise scheduler、归一化统计和模拟器版本,详见精读的复现部分及官方 README。

Related Papers

相关论文