Summary
WLA 提出了一种新的 embodied foundation model 范式,用 AR Transformer 统一 world modeling、language reasoning 和 action synthesis,将 WAM 的世界建模能力与 VLA 的语言推理能力结合,实现了高效的长程任务控制。
Problem & Motivation
现有 WAM(World-Action Model)用 bidirectional DiT 预测纯视觉状态,被低级细节拖累,缺乏语义推理和泛化能力;而 VLA 缺乏对未来状态的预测引导。作者的核心 insight:next state 应包含高层 textual intention + 低层 physical dynamics,前者提供 compact、generalizable 的抽象表示,后者连接 intention 与 motion control。
Method
WLA 采用三个组件:
- AR Transformer backbone:基于 RynnBrain-2B(VLM),预测 textual subtask + physical dynamics(latent action),与 WAM 的 bidirectional DiT 形成鲜明对比
- World Expert:轻量级 diffusion Transformer(SANA-600M),以 physical dynamics 为条件预测未来视觉帧,world modeling loss 监督 backbone 学到有效的 latent action
- Action Expert:flow-matching head(390M),以 meta-queries 的输出为条件生成 action chunk
关键设计:
- Meta-queries(64个):使 world prediction 通过 implicit parameter update 影响 action generation,推理时可禁用 World Expert 实现实时控制
- Textual intention learning:从 instruction 分解的 subtask 序列,配合 memory buffer 支持长程任务
- Test-time scaling:激活 World Expert,采样多个 action candidates,选择预测 value 最高的执行
Key Results
Simulation Benchmarks:
- RoboTwin 2.0 Clean: 92.94% success rate(bimanual manipulation,50 tasks)
- LIBERO: 98.6% average success(四套 suite),test-time scaling 后达 98.9%
- RMBench: 56.5% success rate,接近翻倍最佳 baseline(Mem-0 28.5%,pi_0.5 17.3%)
Real-World:
- Stack Cup 任务完成时间比 baseline WAM 减半
- 在动态 OOD 设置下表现稳健,Dispose Trash 任务成功率最高
Cross-Embodiment Transfer:
- 从 human egocentric videos 学习 unseen tasks,无需 action annotation,成功率接近 same-embodiment video
Efficiency:
- 2B active parameters(总 3.4B),RTX 5090 上 40ms inference latency
- 无 embodied pretraining,30k steps 即达到强性能
Strengths & Weaknesses
Strengths:
- 方法简洁优雅:用 AR Transformer 统一文本和物理动态预测,避免了 WAM 的 bidirectional DiT 复杂性
- 实验全面:覆盖 simulation(RoboTwin/LIBERO/RMBench)+ real-world + cross-embodiment transfer
- RMBench 结果亮眼:56.5% vs 28.5%,证明 language-level subtask tracking 的价值
- 推理高效:World Expert 可禁用,40ms latency 支持实时控制
- 无需 embodied pretraining,训练效率高(30k steps)
Weaknesses:
- World Expert 用静态帧而非视频片段,可能丢失时序动态信息(作者自称简化,但可能是 tradeoff)
- ablation 显示去掉 subtask loss 从 56.5% 掉到 17.25%,说明依赖 subtask supervision——如果 task decomposition 不准确怎么办?
- 2B 参数虽小,但 comparison 中 pi_0.5(3B)有 embodied pretraining,公平性存疑
- cross-embodiment 结果虽 promising,但样本量有限(100 videos per task),泛化能力待更严格验证
Mind Map
mindmap root((WLA)) Problem WAM: 纯视觉预测,缺乏语义推理 VLA: 缺乏未来状态引导 ::icon(fa fa-question) Method AR Transformer backbone World Expert (SANA-600M) Action Expert (flow-matching) Meta-queries implicit conditioning Test-time scaling Results RoboTwin2.0: 92.94% LIBERO: 98.6% RMBench: 56.5% 40ms inference Cross-emb transfer
Notes
- 与 Ye et al. 的 WAM(World Action Models are Zero-shot Policies)直接对比,但 WLA 用 AR 而非 bidirectional DiT——这是核心 architectural difference
- subtask decomposition 的自动化程度如何?是否依赖 LLM 或人工标注?
- meta-queries 机制类似 Perceiver IO / DINOv2 的 register tokens,但用于 cross-modal conditioning
- RMBench 的 memory-dependent setting 很有意思,考验 agent 的 trial-and-error recovery 和 long-term memory 能力