1Step-WAM: Your World Action Model but Faster using 1-Step Action Generation
Organizations: Fudan University · University of Science and Technology of China · Peking University · Shanghai Innovation Institute
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 , , and for FastWAM, ImageWAM and Kairos, respectively. Furthermore, real-world robotic deployments using the FastWAM backbone confirm a consistent 7.44 speedup while maintaining close success rate to the multi-step baseline. Code is avaliable at https://github.com/changdong1687/1Step-WAM.
Figures & tables
| Model | Method | Train params. | Clean | Random | Avg. | Latency (ms) | Speedup |
| FastWAM ( Yuan et al., 2026 ) | Original | – | 91.88 | 91.78 | 91.83 | 669.54 | |
| Naive 1-Step | – | 69.92 | 67.52 | 68.72 | 147.62 | ||
| DMD2 ( Yin et al., 2024a ) | 2.04B | 82.48 | 83.40 | 82.94 | 147.62 | ||
| SnapFlow Luan et al. (2026) | 1.02B | 90.52 | 86.96 | 88.74 | 147.69 | ||
| Flash-WAM Akbari et al. (2026) | 1.02B | 90.44 | 89.80 | 90.12 | 147.62 | ||
| Ours (1Step-WAM) | 13.42M | 91.10 | 90.68 | 90.89 | 89.95 | 7.44 |
| Model | Method | Spatial | Object | Goal | Long | Mean |
| FastWAM ( Yuan et al., 2026 ) | Original | 97.8 | 99.8 | 96.0 | 96.0 | 97.4 |
| Naive 1-Step | 97.0 | 99.2 | 95.0 | 91.6 | 95.7 | |
| Ours (1Step-WAM) | 97.4 | 99.2 | 95.4 | 94.2 | 96.6 | |
| ImageWAM ( Zhang et al., 2026 ) | Original | 97.2 | 99.2 | 98.8 | 98.4 | 98.4 |
| Naive 1-Step | 95.8 | 99.6 | 97.2 | 94.6 | 96.8 | |
| Ours (1Step-WAM) | 97.6 | 100.0 | 98.2 | 96.8 | 98.2 |
| Method | Pick Place | Stack Cups | Hang Mug | Avg. |
| Original FastWAM | 100 | 90 | 50 | 80 |
| Naive 1-Step | 100 | 40 | 10 | 50 |
| 1Step-WAM | 100 | 80 | 45 | 75 |
| Model | Method | Clean | Randomized | |
| FastWAM | Ours | 8 | ||
| DMD2 ( Yin et al., 2024a ) | 8 | |||
| SnapFlow ( Luan et al., 2026 ) | 8 | |||
| Flash-WAM ( Akbari et al., 2026 ) | 8 | |||
| ImageWAM | Ours | 4 | ||
| Kairos | Ours | 8 |
| Model | w/o Joint (ms) | w/ Joint (ms) | Speedup | Reduction (%) |
| FastWAM ( Yuan et al., 2026 ) | 149.05 | 89.95 | 39.65 | |
| ImageWAM ( Zhang et al., 2026 ) | 77.64 | 51.28 | 33.95 | |
| Kairos ( Team et al., 2026 ) | 134.78 | 73.73 | 45.30 |
| Task | Fast-WAM | ImageWAM | Kairos | |||||||||
| Clean | Rand. | Clean | Rand. | Clean | Rand. | |||||||
| Original | Ours | Original | Ours | 10-steps | Ours | 10-steps | Ours | Original | Ours | Original | Ours | |
| adjust_bottle | 1.00 | 1.0 | 1.00 | 0.99 | 1.00 | 1.0 | 0.99 | 0.99 | 1.0 | 1.0 | 1.0 | 1.0 |
| beat_block_hammer | 0.99 | 1.0 | 0.97 | 0.98 | 1.00 | 0.98 | 0.99 | 0.9 | 1.0 | 1.0 | 0.98 | 1.0 |
| blocks_ranking_rgb | 1.00 | 1.0 | 1.00 | 0.98 | 0.99 | 0.9 | 0.99 | 0.98 | 1.0 | 1.0 | 1.0 | 1.0 |
| blocks_ranking_size | 0.94 | 0.91 | 0.98 | 0.92 | 0.95 | 0.9 | 0.98 | 0.96 | 1.0 | 1.0 | 0.92 | 0.96 |
Efficient Inference for World Action Models
Efficient inference for world action models cuts the cost of predicting action-conditioned futures for responsive control and planning.
See all 80 papers