How Representational Priors Shape Generalization in World Action Models
Organizations: Department of Electrical and Computer Engineering, Seoul National University · Graduate School of Data Science, Seoul National University
Abstract
World action models (WAMs) are shown to substantially diverge under generalization performance based on their generative backbones. We investigate this gap through the representational geometry that WAMs inherit from their backbones and how policy training reorganizes it. Our central hypothesis is that a useful representational prior would structure geometry that separates nuisance information from the in-distribution manifold in advance of training as a policy. We quantify this with nuisance orthogonality; orthogonality of the feature under visual variation outside the empirical in-distribution manifold. Across various image and video backbones, we show that a higher orthogonality from the representational prior is associated with stronger robustness to visual variations after WAM training, enabling policy generalization measurable prospectively for world models. In addition, analyzing the resulting robust WAMs reveals nuisance suppression; relative suppression of feature displacement caused by such variations and robust action generation of the policy. Together, these findings provide a geometric perspective on how the representational prior of the world model shapes the generalization performance of WAMs.
Figures & tables
| Model | Camera | Background | Light | Sensor Noise | Average | |||||
| Orth. | SR (%) | Orth. | SR (%) | Orth. | SR (%) | Orth. | SR (%) | Orth. | SR (%) | |
| Baselines | ||||||||||
| Wan2.2 | 0.673 | 16.4 | 0.420 | 53.7 | 0.631 | 78.2 | 0.653 | 37.7 | 0.594 | 46.5 |
| FLUX.2 | 0.803 | 77.6 | 0.774 | 84.9 | 0.845 | 98.5 | 0.773 | 94.9 | 0.799 | 89.0 |
| Scratch FLUX.2 | 0.754 | 65.3 | 0.584 | 56.8 | 0.778 | 71.4 | 0.778 | 62.0 | 0.724 | 63.9 |
| OmniGen2 | 0.802 | 80.0 | 0.700 | 69.4 | 0.712 | 82.6 | 0.768 | 77.1 | 0.746 | 77.3 |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Type | Backbone & Size | Action Expert Size | |
|---|---|---|---|---|
| FastWAM | Video | Wan2.2-TI2V ( Wan et al., 2025 ) | 5.00B | 1.02B |
| ( Yuan et al., 2026 ) | ||||
| ImageWAM (FLUX.2) | Image | FLUX.2 ( Labs, 2025 ) | 3.88B | 0.64B |
| ( Zhang et al., 2026b ) | ||||
| ImageWAM (Scratch FLUX.2) | Image | Scratch FLUX.2 | 3.88B | 0.64B |
| ImageWAM (OmniGen2) | Image | OmniGen2 ( Wu et al., 2025 ) | 3.97B | 0.76B |
| Backbone | FastWAM design | ImageWAM design | ||
|---|---|---|---|---|
| FlashMotion | Kiwi-Edit | SANA-Video 2.0 | MAGI-1 | |
| Action Expert Configurations | ||||
| Layers | 30 | 30 | 30 | 30 |
| Hidden Dimension | 1024 | 1024 | 1024 | 1024 |
| FFN Dimension | 4096 | 4096 | 4096 | 4096 |
| Activation | GELU | GELU | SwiGLU | SwiGLU |
World Action Models
World Action Models connect predictions of how scenes change under action to executable policies for robots, vehicles, and other embodied agents.
See all 223 papers