Why Do Conventional World Models Fail to Learn Cellular Automata?
On the map: World models → Latent predictive world models also fits: Video-generation world models and simulators
also in 1 more
Abstract (arXiv)
Although conventional world models - auto-regressive or diffusion models based on transformers or convolutional networks - may learn surface statistics of world dynamics, can they learn the exact world dynamics from its observed history? Leveraging cellular automata as a simple testbed, we find the answer to be no in many cases. Conventional architectures predict most pixels correctly yet rarely complete a rollout: a CNN predicts 96.3% of cells but completes 18.9% of rollouts; a joint diffusion model completes none. We trace the gap to three failure modes of these world models - namely, they fail to exactly capture spatial locality, temporal locality or temporal stability. Simple changes repair each: (1) for spatial locality, two-dimensional rotary positions lift a transformer from 39.1% to 100% on the Game of Life; (2) for temporal locality, handing each token its cell's previous-frame neighbourhood lifts the same transformer from 25.8% to 99.9% on unseen rules; (3) for temporal stability, causal freezing lifts the same diffusion weights from 42.2% to 99.9%. None of the three changes touches the architectural backbone; each only modifies the information flow within it. We also compare joint and ordered sampling on billiards and, in an exploratory study, on a simulated Burgers equation.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Latent predictive world models, one of 11 directions in World models. Builds on 45 of its 73 references that are in the map. First posted on arXiv 30 Sep 2026.
Builds on
45 of 73 references.
Closest ancestors · 2
Same direction, newest first
- 2026From Kepler to Newton: Inductive Biases Guide Learned World Models in Transformers6 Feb
- 2025Force Prompting: Video Generation Models Can Learn and Generalize Physics-based Control Signals26 May
Other references · 43
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2018Neural Ordinary Differential Equationscited by 3,779
- 2018World Modelscited by 1,513
- 2018Relational inductive biases, deep learning, and graph networkscited by 1,360
- 2023Mastering Diverse Domains through World Modelscited by 1,253
- 2018Towards Accurate Generative Models of Video: A New Metric & Challengescited by 1,234
- 2022MaskGIT: Masked Generative Image Transformercited by 1,099
- 2020Learning to Simulate Complex Physics with Graph Networkscited by 880
- 2018TSM: Temporal Shift Module for Efficient Video Understandingcited by 856
- 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
- 2024Genie: Generative Interactive Environmentscited by 747
- 2019Hamiltonian Neural Networkscited by 712
31 more
- 2025Self Forcing: Bridging the Train-Test Gap in Autoregressive Video Diffusioncited by 625
- 2023Are Emergent Abilities of Large Language Models a Mirage?cited by 523
- 2024From Slow Bidirectional to Fast Autoregressive Video Diffusion Modelscited by 433
- 2022Emergent World Representations: Exploring a Sequence Model Trained on a Synthetic Taskcited by 431
- 2020Lagrangian Neural Networkscited by 398
- 2022Transformers are Sample-Efficient World Modelscited by 328
- 2024Diffusion Models Are Real-Time Game Enginescited by 296
- 2025MAGI-1: Autoregressive Video Generation at Scalecited by 276
- 2024Rotary Position Embedding for Vision Transformercited by 236
- 2024VideoPhy: Evaluating Physical Commonsense for Video Generationcited by 228
- 2023Zoology: Measuring and Improving Recall in Efficient Language Modelscited by 204
- 2025History-Guided Video Diffusioncited by 171
- 2025Do generative video models understand physical principles?cited by 161
- 2021Physion: Evaluating Physical Prediction from Vision in Humans and Machinescited by 153
- 2024FIFO-Diffusion: Generating Infinite Videos from Text without Trainingcited by 137
- 2023The mechanistic basis of data dependence and abrupt learning in an in-context classification taskcited by 118
- 2024The Evolution of Statistical Induction Heads: In-Context Learning Markov Chainscited by 118
- 2024How Transformers Learn Causal Structure with Gradient Descentcited by 118
- 2024Rolling Diffusion Modelscited by 104
- 2024What needs to go right for an induction head? A mechanistic study of in-context learning circuits and their formationcited by 89
- 2024Transformers Can Do Arithmetic with the Right Embeddingscited by 78
- 2023The Expressive Power of Transformers with Chain of Thoughtcited by 50
- 2024Softmax is not Enough (for Sharp Size Generalisation)cited by 46
- 2025Physics of Language Models: Part 4.1, Architecture Design and the Magic of Canon Layerscited by 33
- 2024One-layer transformers fail to solve the induction heads taskcited by 32
- 2024Frequency-Time Diffusion with Neural Cellular Automatacited by 9
- 2024LifeGPT: Topology-Agnostic Generative Pretrained Transformer Model for Cellular Automatacited by 8
- 2024KV Shifting Attention Enhances Language Modelingcited by 8
- 2025What One Cannot, Two Can: Two-Layer Transformers Provably Represent Induction Heads on Any-Order Markov Chainscited by 5
- 2024Learning Elementary Cellular Automata with Transformerscited by 4
- 2025AutomataGPT: Forecasting and Ruleset Inference for Two-Dimensional Cellular Automatacited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025SmallWorlds: Assessing Dynamics Understanding of World Models in Isolated Environments
- 2025From Masks to Worlds: A Hitchhiker's Guide to World Models
- 2025Can World Models Benefit VLMs for World Dynamics?
- 2024One-shot World Models Using a Transformer Trained on a Synthetic Prior
- 2026Platonic Representation Hypothesis on World Models
- 2026MiniWorld: Democratizing the Training of Video World Models from Scratch
- 2025On Memory: A comparison of memory mechanisms in world models
- 2026FluidWorld: Reaction-Diffusion Dynamics as a Predictive Substrate for World Models