SteerQuant: Steering Quantization Error with Action-Guided Scaling in World-Action Models
cs.RO
Submitted: 2026-09-30
Updated: 2026-09-30
Comments: Yunhan Wang and Haodong Wang contributed equally to this work
License: http://creativecommons.org/licenses/by/4.0/
The gist: World-action models (WAMs) jointly generate future world states and actions through iterative denoising, using shared weights to process heterogeneous semantic streams of video, proprioceptive, and
Terminology
Abstract
World-action models (WAMs) jointly generate future world states and actions through iterative denoising, using shared weights to process heterogeneous semantic streams of video, proprioceptive, and action tokens. Quantization reduces inference cost, but comparable numerical errors in different streams can have markedly different effects on final actions, making numerical accuracy alone insufficient for reliable control. We introduce SteerQuant, a 4-bit quantization framework for WAMs that steers errors toward computations with less influence on final actions. It maps how each stream's quantization errors affect final actions and uses this map to guide shared channel scaling. Activation scaling is further calibrated for each stream and denoising step to accommodate changes in activation ranges and action impact. This adapts quantization to different stream requirements without duplicating weights or increasing bit-widths for selected streams. To reduce the extra kernel launches and memory traffic introduced by scaling, we develop Rudder, a 4-bit inference engine for WAMs that fuses scaling and output compensation into low-bit kernels. Under W4A8 and W4A4, SteerQuant maintains mean LIBERO success within 0.8 percentage points of full precision, while delivering up to 2.23 times denoising speedup over BF16 across three WAMs with reduced peak GPU memory usage. On a real dual-arm robot, W4A8 deployment achieves a 1.35 times end-to-end inference speedup while maintaining average task success relative to BF16.
Sources
- ActQuant: Sub-4-bit Action-Guided Quantization for Vision-Language-Action Models
- MosaicQuant: Inlier-Outlier Disaggregation for Unified 4-Bit LLM Quantization
- Cosmos Policy: Fine-Tuning Video Models for Visuomotor Control and Planning
- OnlineWM: Causality-Aware Active Online Learning for Effective World Modeling
- Cosmos 3: Omnimodal World Models for Physical AI
- HALO: A Unified Vision-Language-Action Model for Embodied Multimodal Chain-of-Thought Reasoning
- TwinQuant: Learnable Subspace Decomposition for 4-Bit LLM Quantization
- World Action Models are Zero-shot Policies
- Fast-WAM: Do World Action Models Need Test-time Future Imagination?
- QuantWAMs: Calibrating at the Right Granularity for World Action 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