On Memory: A comparison of memory mechanisms in world models
On the map: World models → Video-generation world models and simulators
Abstract (arXiv)
World models enable agents to plan within imagined environments by predicting future states conditioned on past observations and actions. However, their ability to plan over long horizons is limited by the effective memory span of the backbone architecture. This limitation leads to perceptual drift in long rollouts, degrading the model's capacity to recall recently observed scenes. In this work, we investigate the effective memory span of transformer-based world models through an analysis of memory augmentation mechanisms. We introduce a taxonomy that distinguishes between memory encoding and memory injection mechanisms, motivating their roles in extending the world model's memory through the lens of residual stream dynamics. We evaluate twenty combinations of four encoding methods and five injection methods in the MemoryMaze environment. Using a state recall evaluation task across multiple imagination horizons, we measure the memory recall capacity of each mechanism and analyze their respective trade-offs in reconstruction quality, latent prediction error, and computational cost. We further ablate the effect of injection depth and compare the best memory-augmented vision transformer against a pure state-space model backbone. Our central finding is that the mLSTM memory encoder outperforms all alternatives in both reconstruction and latent fidelity metrics. Paired with additive injection, it exhibits the strongest recall capabilities at a moderate computational cost while matching or slightly exceeding a pure Mamba backbone. This evaluation is limited to a single environment and does not explore the effect of each mechanism on downstream task performance. We believe these research directions merit an in-depth study of their own to clearly isolate the effects, and therefore are left for future work.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Video-generation world models and simulators, one of 11 directions in World models. Builds on 34 of its 47 references that are in the map. First posted on arXiv 7 Dec 2025.
Builds on
34 of 47 references.
Closest ancestors · 4
Same direction, newest first
- 2025StateSpaceDiffuser: Bringing Long Context to Diffusion World Models28 May
- 2025Long-Context State-Space Video World Models26 May
- 2025WorldMem: Long-term Consistent World Simulation with Memory16 Apr
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale22 Oct
Other references · 30
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
- 2017Neural Discrete Representation Learningcited by 4,650
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,424
- 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,830
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2018World Modelscited by 1,444
18 more
- 2018Self-Attention with Relative Position Representationscited by 1,112
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2023Vision Transformers Need Registerscited by 844
- 2016A Learned Representation For Artistic Stylecited by 340
- 2024xLSTM: Extended Long Short-Term Memorycited by 320
- 2024Titans: Learning to Memorize at Test Timecited by 303
- 2024Diffusion for World Modeling: Visual Details Matter in Ataricited by 285
- 2024Navigation World Modelscited by 262
- 2015Highway Networkscited by 257
- 2025ATLAS: Learning to Optimally Memorize the Context at Test Timecited by 57
- 2022Evaluating Long-Term Memory in 3D Mazescited by 41
- 2024Memory Layers at Scalecited by 39
- 2024Mastering Memory Tasks with World Modelscited by 38
- 2023Facing Off World Model Backbones: RNNs, Transformers, and S4cited by 31
- 2024Adaptive Memory Replay for Continual Learningcited by 21
- 2025Pretraining with hierarchical memories: separating long-tail and common knowledgecited by 13
- 2022BioSLAM: A Bio-inspired Lifelong Memory System for General Place Recognitioncited by 12
- 2025EDELINE: Enhancing Memory in Diffusion-based World Models via Linear-Time Sequence Modelingcited by 3
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025From Masks to Worlds: A Hitchhiker's Guide to World Models
- 2026MemoWM: How World Models Change What Agents Need to Remember
- 2025Video World Models with Long-term Spatial Memory
- 2026Composition of Memory Experts for Diffusion World Models
- 2026ReWorld: An Interactive World Model with Long-Horizon Memory
- 2026LOCI: Spatial Linear Memory for Streaming World Models
- 2026Echo-Memory: A Controlled Study of Memory in Action World Models
- 2026MemWM: Memory-Augmented Text-Based World Model