logo Mr.Robota 具身智能学习手册 EN
第十七章:具身世界模型

τ0-WM:统一视频-动作世界模型导读

本文导读论文 τ0-WM: A Unified Video-Action World Model for Robotic Manipulation。它来自 Shanghai Innovation Institute 与 AGIBOT Finch,项目页发布时间为 2026-05-31。官方把它定位为一个面向机器人操作的开源 video-action world model:既能预测动作,也能预测动作导致的未来视频,还能在测试时用 imagined future 评估和修正动作。

1. 先说结论

τ0-WM 不是单纯的 VLA,也不是只会生成视频的世界模型。它试图把三件事放到同一个框架里:

  1. Action Generation:像策略模型一样,根据多视角观测、语言指令和机器人状态输出可执行 action chunk。
  2. Video Prediction:像视频世界模型一样,预测未来多视角视觉 latent / 视频。
  3. Action Evaluation:像 learned simulator / reward model 一样,给候选动作 rollout 出未来,并估计任务进度分数。

一句话概括:

τ0-WM 的核心不是“机器人会生成视频”,而是让机器人在执行前先提出动作、想象动作后果、评估后果,再必要时修正动作。

这篇适合放在 17-具身世界模型,而不是 VLA 章节。原因是它虽然包含策略接口,但核心创新是把 policy、video prediction、action-conditioned simulator 和 test-time computation 统一成一个世界模型式决策机制。

2. 论文、项目和开源状态

条目 链接 当前状态
项目页 AGIBOT Finch τ0-WM 官方项目页,含图、视频、论文、代码和 HF 入口
论文 arXiv:2606.01027 / HTML 2026-05-31 提交
PDF 官方 PDF 项目页直接链接
GitHub sii-research/tau-0-wm 官方实现,Apache-2.0;已发布 post-training 与 inference 代码
Hugging Face sii-research/tau-0-wm VAM 预训练权重入口,模型卡显示 Apache-2.0

开源状态要读得细一点。GitHub README 写明:

  • 2026-06-26 发布了 post-training training 和 inference code。
  • VAM pretrained weights 已放在 Hugging Face。
  • Simulator pretrained weights 仍写着 will be released soon。
  • Test-Time Computation 代码会随 Simulator 权重进一步发布。

所以当前可以学习和二次开发的重点是 VAM 权重、post-training pipeline、policy server / client inference 链路;完整复现论文里的 ACVS simulator-based rectification,还需要等官方继续放出 Simulator 权重和对应 TTC 代码。

3. 它为什么值得放进世界模型线

本仓库前面已经有几类世界模型导读。τ0-WM 可以放在下面这个谱系里理解:

方法 世界模型预测什么 怎样服务机器人
LeWM latent future / dynamics 建立世界模型基础概念
RISE action-conditioned future + value 在想象中做 policy improvement
RAW-Dream task-agnostic video rollout + VLM reward 用 world model 做 VLA 后训练
WoG action-relevant future condition 不生成完整视频,把未来压成动作条件
WALL-WM event-level world-action video modeling 把动作和视频事件对齐
RoboDream robot-only trajectory + scene/object priors 合成机器人示教视频数据
Gamma-World 多智能体同步未来视频 多 agent 共享世界 rollout
BWM action-conditioned robot video rollout 低成本视觉世界模拟
τ0-WM 未来视频 + executable action chunk + progress score 执行时 propose / evaluate / revise 动作

τ0-WM 和 BWM、WALL-WM 最接近,都在“动作条件视频世界模型 / world-action model”这条线上。但 τ0-WM 更进一步:它不只是生成动作条件未来视频,还把 动作生成器 VAM 和 动作条件视频模拟器 ACVS 接到一个测试时决策闭环里。

4. 总览图:从多源数据到真实部署

τ0-WM 总览图

图 1 τ0-WM 总览。 左边是多源异构数据,中间是 Video Action Model 与 Action-Conditioned Video Simulator,右边是真实机器人任务部署。图中最重要的不是“5B 很大”,而是它把不同来源的数据按各自能提供的监督信号接进统一模型。
来源:τ0-WM 官方项目页。

这张图可以按四块读。

第一块是 Diverse Data Source。官方项目页写 τ0-WM 使用约 27,300 小时异构数据:

数据来源 时长 占比 提供什么监督
Real-robot teleoperation 17,800 小时 65% 多视角真实机器人观测、机器人状态、可执行 action
UMI-style data 6,500 小时 24% 更丰富的操作行为和场景,带动作相关信号
Egocentric human interaction data 3,000 小时 11% 第一人称人类交互视频,提供物体运动、接触、任务时序

这里的关键是 不要强行把所有数据当成同一种数据。人类第一人称视频有丰富视觉动态,但没有机器人控制空间里的动作标签;真实机器人数据有动作标签,但采集贵、覆盖窄。τ0-WM 用 modality-specific supervision masks,让每个样本只监督它真实拥有的信号。

第二块是 Video Action Model, VAM。它是 policy-facing interface:输入多视角图像、语言指令和机器人状态,输出未来视觉 latent 与 continuous action chunk。

第三块是 Action-Conditioned Video Simulator, ACVS。它不是给一个文本 prompt 生成视频,而是给定当前观测、指令和候选 action chunk,预测这串动作会导致什么未来,并输出 dense task-progress / reward score。

第四块是 Test-Time Computation。部署时不是只采样一个动作就执行,而是采样多个候选动作,用 re-denoising consistency 和 simulator-based scoring 排序;如果候选质量低,就用模拟出来的未来反过来修正动作。

5. 方法图:VAM 和 ACVS 是两个互补接口

τ0-WM 方法图

图 2 τ0-WM 方法架构。 左边是 VAM,右边是 ACVS。VAM 负责动作生成,ACVS 负责动作后果模拟和奖励评估,两者共享视频生成 backbone。
来源:τ0-WM 官方项目页。

5.1 左半边:Video Action Model

VAM 的输入包括:

  • multi-view observations:多视角当前观测。
  • language instruction:自然语言任务指令。
  • robot state:机器人 proprioception / end-effector state。
  • noised action:扩散式动作分支中的带噪动作 token。

VAM 的输出包括:

  • predicted video:未来视觉 latent / 视频预测。
  • predicted action:连续 action chunk。

图里的结构很关键:

mermaidflowchart LR
  A["多视角观测"] --> B["Video DiT Block"]
  C["语言指令"] --> B
  D["带噪 action token"] --> E["Action DiT Block"]
  B --> F["未来视频 latent"]
  B --> E
  E --> G["连续 action chunk"]

Video DiT 分支不是一个旁路装饰。Action DiT 通过 layer-wise cross-attention 读取视频中间表征,所以动作分支被迫使用“未来会怎么变化”的表示来预测动作。这一点和普通 action-only policy 不一样。

普通 VLA 往往是:

textimage + language + state -> action

τ0-WM 的 VAM 更像:

textimage + language + state -> future visual latent + action

这就把未来预测变成了控制相关训练目标,而不是额外可视化结果。

5.2 右半边:Action-Conditioned Video Simulator

ACVS 的输入是:

textcurrent observation + instruction + candidate action chunk

输出是:

textpredicted multi-view future + dense task-progress score

这和 VAM 的方向正好互补:

模块 问的问题 输入 输出
VAM 我现在应该做什么 观测、指令、状态 action chunk + future latent
ACVS 如果做这串动作会怎样 观测、指令、候选动作 未来视频 + progress / reward

ACVS 不是传统物理仿真器。它没有显式接触力、刚体状态、碰撞求解器,也不能保证像 MuJoCo / Isaac Sim 那样严格可验证。它更像一个 learned visual consequence evaluator:用视频世界模型预测动作后果,再用 task-progress score 给候选动作排序。

6. 5B 和 5.5B 口径怎么理解

外部传播常说 τ0-WM 是 5B 级具身世界模型。论文架构细节里写得更具体:VAM builds on Wan2.2-TI2V-5B,并加入约 0.5B 参数的 Action DiT branch,所以 VAM 总参数量约 5.5B。

两个说法并不矛盾:

口径 含义
5B 项目传播中强调的视频 diffusion backbone / world model 规模
5.5B VAM 论文架构里把 5B video DiT backbone 和 0.5B Action DiT 分支合在一起算

教程里建议统一称为 5B 级开源具身世界模型,在架构细节处补充 VAM 的 5.5B 口径。

7. 异构数据训练:每种数据只监督它真的有的信号

τ0-WM 的数据设计很值得学习。它没有只依赖昂贵的真实机器人数据,也没有幻想人类视频可以直接提供机器人动作标签,而是把数据分层使用。

mermaidflowchart TB
  A["Real Robot Teleoperation<br/>有机器人动作"] --> D["视频预测 loss"]
  A --> E["动作生成 loss"]
  A --> F["progress / failure loss"]
  B["UMI-style Data<br/>有手持操作轨迹"] --> D
  B --> E
  C["Egocentric Human Videos<br/>无机器人动作"] --> D
  G["Rollout / Failure Trajectories"] --> F
  D --> H["共享视频-动作表征"]
  E --> H
  F --> H

这个设计背后的判断很现实:

  • 真实机器人数据最值钱,因为它把视觉和 robot control space 绑定起来。
  • UMI-style 数据能扩大操作场景和行为多样性,但和目标机器人 embodiment 仍有差异。
  • 人类第一人称视频最便宜、覆盖最广,但缺少可执行机器人动作。
  • failure / rollout 数据对 reward / progress 学习很重要,因为成功示教很少告诉模型“哪里会失败”。

modality-specific supervision masks 的作用就是:如果某条数据没有动作标签,就不要对它计算 action loss;如果某条数据没有某个相机视角,就不要让缺失视角产生错误监督;如果有 failure/progress 信号,就用它训练 ACVS 的评价能力。

这点和很多多模态预训练很像:真正难的不是把数据堆起来,而是让不同数据在同一个模型里各司其职。

8. Test-Time Computation:先提议,再评估,再修正

τ0-WM 最有意思的地方是部署时的计算逻辑。它不是传统 feed-forward policy:

textobservation -> action -> execute

而是:

mermaidflowchart LR
  A["当前多视角观测 + 指令 + 状态"] --> B["VAM 采样多个候选 action chunks"]
  B --> C["Re-denoising Consistency Score 排序"]
  C --> D{"候选动作质量足够?"}
  D -- "是" --> E["执行最高分 action chunk"]
  D -- "否" --> F["ACVS 模拟候选动作未来"]
  F --> G["根据 progress / reward 选择较好未来"]
  G --> H["用 imagined future 条件化第二次 VAM"]
  H --> E

这里有两个关键技巧。

Re-denoising Consistency Score 用来判断候选动作是否和模型学到的条件动作分布一致。直觉上,如果一个动作 chunk 重新加噪、再去噪后仍然能回到相近动作,说明它处在模型认为可靠的区域;如果一致性差,说明这个候选可能不稳定。

Simulator-based Rectification 则更像 world model planning:当候选动作质量低时,让 ACVS 预测不同候选动作的未来视频和任务进度,挑出更有希望的 imagined future,再把这个未来作为条件去生成修正后的 action chunk。

这也是 τ0-WM 和很多 VLA 的本质差异:VLA 通常把模型算力花在一次前向推理上,τ0-WM 把额外算力花在执行前的动作评估和修正上。

9. 实验结果怎么读

τ0-WM 主实验结果

图 3 真实机器人任务成功率和任务进度对比。 论文比较 Fast-WAM、π0.5 和 τ0-WM,在四个未出现在预训练数据中的精细操作任务上统计 success rate 与 task accomplishment progress。
来源:τ0-WM 官方项目页。

实验覆盖三个机器人 embodiment:

  • AGIBOT-G01。
  • ARX manipulators。
  • 双臂 Franka。

任务包括 Toolbox、School Bag、Badminton、Faucet,都是长时程或精细几何对齐任务。论文强调这些任务不在预训练语料中,用来测试泛化和 post-training 后的执行能力。

从图 3 可以读出几个重点:

对比项 观察
Fast-WAM 在这些任务上整体较弱,尤其长流程和精细对齐场景吃亏
π0.5 是强 VLA baseline,在 Toolbox 等任务上有竞争力
τ0-WM 平均成功率和平均任务进度最高,在 School Bag、Badminton、Faucet 上优势明显

注意这里的 π0.5 是 baseline,不是 τ0-WM 的基座。τ0-WM 的核心基座是 Wan2.2-TI2V-5B 视频 diffusion backbone,再加 action branch 和 simulator/evaluation 机制。

论文还做了数据消融。比较 Robot-only 和 Robot+UMI+Ego 可以看到:

设置 任务 Clean Cluttered Avg
Robot-only Zero-shot Pen-to-holder 0.22 0.06 0.14
Robot+UMI+Ego Zero-shot Pen-to-holder 0.56 0.53 0.55
Robot-only SFT Object-wipe-place 0.85 0.55 0.70
Robot+UMI+Ego SFT Object-wipe-place 0.90 0.75 0.83

这张表说明异构数据不是宣传点,而是确实影响 zero-shot 和 SFT 表现。尤其 cluttered 场景里,Ego / UMI 带来的视觉交互多样性会让模型更稳。

10. 代码仓库能复现到什么程度

官方 GitHub 当前给出的可操作入口包括:

bashgit clone https://github.com/sii-research/tau-0-wm.git
cd tau-0-wm
pip install -r requirements.txt

仓库 README 的主线可以分成三块。

10.1 预训练 VAM 推理

需要准备:

  • τ0-WM VAM pretrained weight。
  • Wan2.2-TI2V-5B 权重。
  • VAE 权重。
  • text encoder / tokenizer 权重。

然后修改:

textconfigs/deployment/tau_pretrain_rela_eef6d.yaml

主要替换这些字段:

textdiffusion_model.model_path
vae_path
text_encoder.checkpoint_path
text_encoder.tokenizer_path

仓库提供 policy server 方式运行:

bashbash run_infer_server.sh $HOST $PORT
python web_infer_utils/simple_client.py

这里的 simple client 只是随机观测示例,能验证服务链路,不代表真实机器人任务已经跑通。真正接入机器人时,需要把多视角图像、双臂末端位姿、夹爪状态和 action layout 对齐。

10.2 Post-training

仓库提供下游任务 post-training 入口:

bashbash scripts/train.sh main.py \
    configs/tau_model/posttrain_taco_play_abs.yaml \
    runner/posttrain.py

下游任务需要三类文件:

文件 作用
LeRobot-format dataset 训练数据,官方建议 LeRobot >= 0.4.0
configs/data/<task>/*.yaml 定义 dataset class、路径、统计文件、action/state layout
configs/tau_model/*.yaml 定义模型、训练参数、权重路径和输出目录

如果大家未来要复现,建议先用官方示例 taco_play,不要一开始接自己的机器人数据。原因是 action/state layout 很容易错:末端位姿、四元数顺序、夹爪开合范围、绝对/相对动作表示都必须和模型配置一致。

10.3 Action Space

仓库 README 写得比较具体:

  • server 输入的 state 是双臂末端执行器绝对位姿,共 14 维:左右臂各自 xyz + quaternion,四元数顺序为 xyzw。
  • gripper state 是 2 维,范围 0 到 120,0 表示 open,120 表示 close。
  • server 输出 action shape 为 {T, 16}。
  • 输出顺序是左臂末端位姿、左夹爪 openness、右臂末端位姿、右夹爪 openness。
  • 预训练阶段内部预测的是相对末端位姿,包含 20 维:每只手 xyz + 6d-rotation,代码会自动做 quaternion 与 6D rotation 转换。

这部分对复现非常关键。很多 VLA / WAM 复现失败不是模型跑不起来,而是 action space 对不齐:坐标系、四元数顺序、夹爪开合方向、绝对/相对动作混了,结果策略看起来“能推理”,但动作完全不可执行。

11. 当前不建议怎么复现

这篇暂时不建议大家把完整论文实验重跑一遍,原因有三个。

第一,模型和依赖很重。Wan2.2-TI2V-5B + VAM 权重不是普通笔记本任务,真实部署还涉及多视角相机、机器人状态接口和 server-client 延迟。

第二,论文里的完整 test-time computation 依赖 Simulator 权重和更完整 TTC 代码。GitHub README 目前明确说 Simulator pretrained weights 和 TTC 代码后续继续发布。

第三,真实机器人任务不是公开标准 benchmark。论文展示的是 AGIBOT-G01、ARX、双臂 Franka 上的真实操作任务;没有对应硬件和数据,很难复刻同一成功率。

因此本教程当前建议把复现目标分成三档:

档位 目标 适合人群
读懂方法 看论文、项目页和代码结构,理解 VAM / ACVS / TTC 组队学习同学
跑通代码链路 下载 VAM 权重和 Wan2.2,启动 policy server,用 simple client smoke test 有大显存 GPU 的同学
下游微调 准备 LeRobot-format 数据,改 data YAML 和 training YAML 有机器人数据或仿真数据的同学
完整复现 TTC 等 Simulator 权重和 TTC 代码继续发布 做机器人世界模型研究的同学

12. 和 π0.5、Fast-WAM、BWM 的区别

方法 类型 核心问题 τ0-WM 与它的关系
π0.5 VLA / policy baseline 从视觉语言输入预测动作 τ0-WM 把它作为强 baseline,对比说明 world-model-style prediction 和 TTC 的价值
Fast-WAM video-action model baseline 强调更快的 WAM 推理 τ0-WM 对比其在长时程精细任务上的表现
BWM action-conditioned video world model 根据初始图像和动作生成未来视频 τ0-WM 更强调动作生成、动作评估和测试时修正
RAW-Dream world model + VLM reward + RL 在 imagined world model 中强化 VLA τ0-WM 更偏部署时 proposal-evaluation-revision,不是主要做 RL 后训练
WALL-WM event-level world-action modeling 把视频事件和动作块对齐 τ0-WM 更强调统一 VAM + ACVS 接口和异构数据预训练

如果只从“是否能输出 action”看,τ0-WM 像 VLA;如果只从“是否能预测未来视频”看,它像视频世界模型。但它真正的定位是 world-action model:动作和未来不是两个任务,而是同一套预测表征的两个接口。

13. 这篇论文真正的创新点

可以把创新点拆成四个层次。

13.1 统一接口

过去很多方法把 policy learning 和 world modeling 分开:

textpolicy: observation -> action
world model: observation + action -> future
reward model: future -> score

τ0-WM 把它们统一到共享视频 diffusion backbone 上,让 action branch 和 simulator branch 使用同一类未来预测表征。这样世界模型不是训练时的辅助 loss,而是部署时可以调用的决策模块。

13.2 异构监督

真实机器人数据、人类视频、UMI-style 数据和 failure rollout 很难直接混训。τ0-WM 的处理方式是统一模型、分开监督:有动作就训动作,没有动作就只训视频,有 progress/failure 就训评价,没有某个视角就 mask 掉。

这对机器人很现实,因为未来很长时间里,机器人数据都不可能像互联网图文那样干净统一。

13.3 测试时计算

τ0-WM 把“多采样、打分、必要时模拟修正”做成部署机制。这点和大语言模型里的 test-time compute 思路类似:不是只依赖一次前向,而是在关键状态花更多计算,换取更稳的动作。

机器人里这件事尤其重要,因为一次错误执行可能会把物体碰飞、卡住、掉落,后续很难恢复。

13.4 真实机器人多 embodiment 验证

论文没有只在一个固定机器人上测,而是在 AGIBOT-G01、ARX、双臂 Franka 上做实验。这说明它的接口设计至少考虑了 embodiment diversity。不过这还不等于“任意机器人即插即用”,因为 action space、状态定义和控制器仍然需要工程适配。

14. 局限和需要冷静看的地方

这篇很强,但不要被“最大规模”“开源”“世界模型”几个词带跑。

第一,它不是传统仿真器。ACVS 可以预测视频和 progress score,但不能提供严格物理状态、接触力、碰撞约束和可验证动力学。

第二,完整 TTC 目前还没完全开源。VAM 权重和部分代码可用,但 Simulator 权重和完整 test-time computation 还在官方计划中。

第三,真实机器人评测规模有限。四个任务很有挑战,但还不能代表所有操作任务和所有机器人 embodiment。

第四,模型很重。论文附录写真实机器人推理在单张 RTX 5090 上部署,标准配置约 220 ms per query,缓存文本表示后约 180 ms,进一步优化可到约 140 ms。这个速度对闭环 action chunk 执行可以接受,但不是低成本边缘部署模型。

第五,未来视频预测不等于可靠控制。视频看起来对,不代表动作在真实接触中一定稳定。尤其插入、柔性物体、细小零件、遮挡后的接触状态,都可能需要触觉或力反馈。

15. 组队学习里怎么使用这篇

建议把 τ0-WM 放在世界模型方向 Task04,和 RAW-Dream、WALL-WM、BWM 一起对比。

推荐作业题:

对比 RAW-Dream、BWM、τ0-WM:三者都使用视频世界模型,但它们分别是在训练时改进策略、生成动作条件视频,还是在部署时评估和修正动作?

交付建议:

  1. 画出 VAM 和 ACVS 的输入输出。
  2. 解释为什么 τ0-WM 不是 π0.5 基座。
  3. 说明 27.3K 小时异构数据各自提供什么监督。
  4. 解释 re-denoising consistency 和 simulator-based rectification 的直觉。
  5. 写清楚当前 GitHub / HF 能复现什么,暂时不能复现什么。

16. 推荐阅读顺序

如果时间有限,建议这样读:

  1. 先看项目页总览图,理解 diverse data、VAM、ACVS、TTC 四块。
  2. 再看论文 Figure 2,弄清 VAM 和 ACVS 的共享 backbone。
  3. 读论文 Section III,理解异构数据和 supervision masks。
  4. 读 Section IV / V,分别看 VAM 和 ACVS。
  5. 读 Section VI,理解 test-time computation。
  6. 最后看 GitHub README,确认当前代码、权重和复现边界。

这篇最值得带走的观点是:

机器人世界模型不应该只停留在“生成未来视频给人看”,而应该成为执行时可调用的动作评估器和修正器。

17. 引用与图片来源

本文图片来自 τ0-WM 官方项目页,为了教程稳定访问已压缩保存到本地 assets/。论文内容参考 arXiv:2606.01027、官方 PDF、GitHub 仓库 和 Hugging Face 权重页。