1Step-WAM: Your World Action Model but Faster using 1-Step Action Generation
cs.CV
Submitted: 2026-10-11
Updated: 2026-10-11
Code: https://github.com/changdong1687/1Step-WAM
License: http://creativecommons.org/licenses/by/4.0/
The gist: World Action Models (WAMs) unify visual world modeling and action generation, yet their real-world deployment is heavily bottlenecked by iterative action denoising and the sequential execution of
Abstract
World Action Models (WAMs) unify visual world modeling and action generation, yet their real-world deployment is heavily bottlenecked by iterative action denoising and the sequential execution of visual prefilling and action prediction. We demonstrate that the action-denoising trajectories in WAMs are approximately straight, rendering the first-step velocity a strong estimate of the final action direction. Building on this observation, we introduce 1Step-WAM, a framework that transforms multi-step action denoising into one single forward. Specifically, we employ a lightweight calibrator to directly correct the first step velocity, and use bandwidth-constrained action decoding to further reduce errors in the calibrated one-step actions. Compressing action generation into a single step enables it to execute in parallel with visual prefilling, completely breaking the sequential bottleneck of the traditional imagination-then-action paradigm. Evaluations on RoboTwin and LIBERO show that 1Step-WAM largely preserves multi-step task success with only about 13.4M trainable parameters, while achieving inference speedups of 7.44 times, 5.98 times, and 24.84 times for FastWAM, ImageWAM and Kairos, respectively. Furthermore, real-world robotic deployments using the FastWAM backbone confirm a consistent 7.44 times speedup while maintaining close success rate to the multi-step baseline. Code is avaliable at https://github.com/changdong1687/1Step-WAM.
Sources
- Flash-WAM: Modality-Aware Distillation for World Action Models
- World Simulation with Video Foundation Models for Physical AI
- GR00T N1: An Open Foundation Model for Generalist Humanoid Robots
- $\pi_0$: A Vision-Language-Action Flow Model for General Robot Control
- RoboTwin 2.0: A Scalable Data Generator and Benchmark with Strong Domain Randomization for Robust Bimanual Robotic Manipulation
- One Step Diffusion via Shortcut Models
- CogVideo: Large-scale Pretraining for Text-to-Video Generation via Transformers
- OpenVLA: An Open-Source Vision-Language-Action Model
- Causal World Modeling for Robot Control
- One-Step Flow Policy: Self-Distillation for Fast Visuomotor Policies
- Flow Matching for Generative Modeling
- LIBERO: Benchmarking Knowledge Transfer for Lifelong Robot Learning
- SnapFlow: One-Step Action Generation for Flow-Matching VLAs via Progressive Self-Distillation
- $\pi_{0.5}$: a Vision-Language-Action Model with Open-World Generalization
- Consistency Policy: Accelerated Visuomotor Policies via Consistency Distillation
- Progressive Distillation for Fast Sampling of Diffusion Models
- SmolVLA: A Vision-Language-Action Model for Affordable and Efficient Robotics
- Consistency Models
- Galaxea Open-World Dataset and G0 Dual-System VLA Model
- Gemini Robotics: Bringing AI into the Physical World
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models