RM-WAM: Reliable Memory-Augmented World Action Models for Long-Horizon Robot Manipulation
On the map: World models → Embodied and robot world models
Abstract (arXiv)
World Action Models (WAMs) improve robotic manipulation by jointly predicting future visual states and actions, yet most existing methods make decisions primarily from the current observation. This limits their reliability in long-horizon tasks, where visually similar states may correspond to different interaction histories and task stages. Moreover, directly incorporating historical observations can introduce stale, stage-mismatched, or error-contaminated information, leading to inaccurate world prediction and accumulated closed-loop errors. We propose RM-WAM, a reliable memory-augmented World Action Model that treats memory as a context-dependent predictive resource rather than universally useful historical context. RM-WAM represents history using complementary short-term visual state memory and latent event memory, which respectively capture recent interaction dynamics and longer-term state transitions. A counterfactual memory learning scheme varies the accessible history while keeping the current inputs and prediction targets fixed, enabling the model to learn a context-dependent utility estimate for historical memory. Based on this estimate, a utility-aware gating mechanism adaptively controls memory injection into the world-action backbone. Experiments demonstrate that RM-WAM highlights the importance of reliable and selective memory utilization in long-horizon tasks.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Embodied and robot world models, one of 11 directions in World models. First posted on arXiv 11 Oct 2026.
Builds on
No reference list yet.
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026MemoryWAM: Efficient World Action Modeling with Persistent Memory
- 2026DIM-WAM: World-Action Modeling with Diverse Historical Event Memory
- 2026HiMem-WAM: Hierarchical Memory-Gated World Action Models for Robotic Manipulation
- 2026Mem-World: Memory-Augmented Action-Conditioned World Models for Persistent Robot Manipulation
- 2026WorldScape Policy 2.0: Empowering Steerable World Action Modeling with Reasoning-Augmented Memory and In-Context Learning
- 2026Memory as Plans: World-Action Modeling with Memory-Grounded Planning
- 2026PACT-WAM: Predicting Actions and Visual Foresight with Compact Temporal Encoding for Robot Manipulation
- 2026Long-WAM: Scaling the Context of World-Action Models