cs.AISep 30, 2026

Why Do Conventional World Models Fail to Learn Cellular Automata?

Authors: Shaoyang Guo, Ziming Liu

Organizations: Meta Circle(元环智能) · Peking University · Tsinghua University

Abstract

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.

Figures & tables

Explore similar work

CardsList
  1. Storing Is Not Remembering: LSTM-UT and Bounded Gated Memory for Looped Transformers

    Sep 17, 2026Aras Kavuncu, Muhammad Burhan HafezTransformer ArchitecturesNeural Cellular Automata

  2. World Machine: Towards Generative World Modeling for Time-Series

    May 21, 2026Elton Cardoso do Nascimento, Alexandre da Silva Simões, Esther Luna Colombini +2Many WorldsEnergy-Based Transformer

  3. Looped World Models

    Jun 16, 2026Hongyuan Adam Lu, Z. L. Victor Wei, Qun Zhang +28World ModelsLong-Horizon Inference