Search papers, labs, and topics across Lattice.
To overcome the severe denoising latency that bottlenecks diffusion- and flow-based World Action Models (WAMs) during real-time robotic control, the authors develop $\mathrm{WAM}{\scriptstyle\mathrm{ACHINE}}$, a training-free framework that preserves and dynamically updates intermediate inference states. The approach counteracts state staleness via a three-tiered hierarchy: trajectory remapping across closed-loop replanning cycles, anticipatory observation rebinding across denoising steps, and probe-verified residual rescaling across Transformer layers. Benchmarked across three WAM architectures on LIBERO and RoboTwin 2.0, it yields 1.47–3.05$\times$ observation-to-action latency speedups and 2.23–3.27$\times$ GPU inference speedups while maintaining 96.7–99.5% of baseline task success rates.
Generative world action models can run up to 3× faster in closed-loop control without fine-tuning or task degradation by dynamically repairing and reusing intermediate computational states instead of recomputing them from scratch.
World Action Models (WAMs) enable future-aware control by jointly modeling actions and environment dynamics. However, iterative diffusion or flow inference incurs substantial denoising latency. Prior inference state offers a natural opportunity for acceleration, yet changing planning contexts, observations, and intermediate representations can quickly render retained state stale. Preserving useful computation therefore requires adapting inference state rather than reusing it as-is. To this end, we present $\mathrm{WAM}{\scriptstyle\mathrm{ACHINE}}$, a training-free framework that accelerates WAM inference by preserving and adapting inference state for efficient and accurate continuation as the control loop evolves. Across closed-loop replans, Trajectory Remapping remaps replan state from the preceding replan to initialize the next replan, reducing redundant trajectory generation. Across denoising steps, Observation Rebinding performs anticipatory inference during action execution and rebinds retained denoising state to the real observation for continuation when consistency checks pass, reducing latency exposed to the control loop. Across Transformer layers, Residual Rescaling selectively rescales retained layer state and refreshes it through full computation of the middle layers when probe checks fail, reducing repeated Transformer computation. Evaluations of three representative WAM architectures on LIBERO and RoboTwin 2.0 show that $\mathrm{WAM}{\scriptstyle\mathrm{ACHINE}}$ achieves 1.47-3.05$\times$ speedups in observation-to-action latency and 2.23-3.27$\times$ speedups in GPU inference time per replan, while preserving 96.69-99.54% of native WAM task success.