WAM: Distilling Action Tangent Fields into World Action Models
cs.RO
Submitted: 2026-10-07
Updated: 2026-10-07
Comments: 9 pages, 4 figures
License: http://creativecommons.org/licenses/by/4.0/
The gist: World Action Models (WAM) improve robot policies by augmenting sparse action supervision with dense future prediction.
Abstract
World Action Models (WAM) improve robot policies by augmenting sparse action supervision with dense future prediction. However, much of the predictable future is dominated by appearance and scene persistence rather than action-dependent dynamics. We observe that several recent WAM designs, including optical flow, motion-centric representations, and latent actions, can be understood from a common perspective in which world supervision becomes more efficient as it contains a higher proportion of action-relevant variation. Based on this insight, we introduce Action Tangent Fields, which reformulate world supervision through a local Taylor expansion of how actions induce changes in future dynamics. We represent future dynamics in Residual-VAE space, where the future latent remains recoverable from the current latent and its residual, and use a strong action-conditioned world model (ACWM) to probe the local correspondence between action variations and residual-world variations. This local first-order structure is distilled into the WAM to guide its denoising supervision toward dynamics that are more tightly coupled to action, rather than merely predictable from appearance. Across LIBERO-Plus, RoboTwin, and RoboTwin2.0-Plus, our method consistently improves robustness to lighting, background, camera, layout, and other environmental perturbations. Despite using no large-scale embodied pretraining, it achieves stronger robustness under several distribution shifts than pretrained policies. We further distill multi-step VideoDiT denoising into a single step for efficient inference. Our results suggest that effective WAM supervision should remain information-rich while concentrating its predictive capacity on the directions along which actions change the future.
Sources
- Motus: A Unified Latent Action World Model
- $\pi_{0.5}$: a Vision-Language-Action Model with Open-World Generalization
- $\pi_0$: A Vision-Language-Action Flow Model for General Robot Control
- UniVLA: Learning to Act Anywhere with Task-centric Latent Actions
- RoboTwin 2.0: A Scalable Data Generator and Benchmark with Strong Domain Randomization for Robust Bimanual Robotic Manipulation
- ABot-PhysWorld: Interactive World Foundation Model for Robotic Manipulation with Physics Alignment
- LIBERO-Plus: In-depth Robustness Analysis of Vision-Language-Action Models
- DROID: A Large-Scale In-The-Wild Robot Manipulation Dataset
- Cosmos Policy: Fine-Tuning Video Models for Visuomotor Control and Planning
- Efficient-WAM: A 1B-Parameter World-Action Model with Low-Cost Future Imagination
- Causal World Modeling for Robot Control
- Light-WAM: Efficient World Action Models with State-Fusion Action Decoding
- Genie Envisioner: A Unified World Foundation Platform for Robotic Manipulation
- HoloBrain-0 Technical Report
- FAST: Efficient Action Tokenization for Vision-Language-Action Models
- VLA-JEPA: Enhancing Vision-Language-Action Model with Latent World Model
- Interactive Post-Training for Vision-Language-Action Models
- Wan: Open and Advanced Large-Scale Video Generative Models
Related papers
- FMT x: An Efficient and Asymptotically Optimal Extension of the Fast Marching Tree for Dynamic Replanning
- MPCFormer: A physics-informed data-driven approach for explainable socially-aware autonomous driving
- RoboLab: A High-Fidelity Simulation Benchmark for Analysis of Task Generalist Policies
- ARC: A Reasoning Recipe for Robot Foundation Models
- HRDexDB: A 4D Dexterous Grasping Dataset Across Human and Multiple Robot Embodiments
- APT: Action Expert Pretraining Improves Instruction Generalization of Vision-Language-Action Policies