mlsanityData to 14 Oct 2026

Transformers Linearly Represent Highly Structured World Models

arXiv 2605.18847 ↗First posted 13 May 2026cs.LG cs.AIcited by 0

On the map: World models → World models for language agents

also in 1 more
Abstract (arXiv)

Do transformers, when trained on sequential reasoning traces, build internal models of the underlying task? And if so, does the structure of those internal representations mirror the structure of the domain? We train an 8-layer transformer on Sudoku solving traces and perform a mechanistic analysis of its internal computation. We establish two results. First, the model builds a substructure world model: it does not represent the board state cell by cell, as a human analyst would expect, but organizes information around the rows, columns, and boxes that Sudoku's constraints act on. Second, we identify a naked-single circuit: a small set of dedicated neurons in the final MLP layer, each individually detecting when exactly one digit remains possible for a specific cell, and reliably promoting that digit. These findings show that the geometry of an emergent world model is shaped by the constraint algebra of the domain, not its surface presentation, and that the resulting decision circuit is sparse, monosemantic, and fully interpretable. More broadly, they demonstrate that mechanistic interpretability tools can recover an end-to-end algorithmic account of how a transformer solves a combinatorial reasoning task.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in World models for language agents, one of 11 directions in World models. Builds on 2 of its 17 references that are in the map. First posted on arXiv 13 May 2026.

Builds on

2 of 17 references.

References · 2

  1. 2016Understanding intermediate layers using linear classifier probescited by 1,353
  2. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2024Causal Language Modeling Can Elicit Search and Reasoning Capabilities on Logic Puzzles
  2. 2025Teaching Transformers to Solve Combinatorial Problems through Efficient Trial & Error
  3. 2025Liquid Reasoning Transformers: A Sudoku-Based Prototype for Chess-Scale Algorithmic Tasks
  4. 2026Transformers Struggle to Use Their Emergent World Models: Revisiting the Tower of Hanoi, and the Illusion of Thinking
  5. 2024Transformers Use Causal World Models in Maze-Solving Tasks
  6. 2026Knowledge transfer between simple discrete worlds
  7. 2022Transformers Learn Shortcuts to Automata
  8. 2026Invariant Learning Dynamics of Transformers in Inductive Reasoning Tasks