Chess-World-Model: A 10M-Game Benchmark for Exact State Tracking from Chess Move Sequences
On the map: World models → World models for language agents
Abstract (arXiv)
World models require state tracking, which is the ability to maintain a correct latent state across action sequences. Existing benchmarks are often synthetic or language-based, limiting their value as tests of structured state updates in realistic domains. We introduce Chess-World-Model, a large-scale state-tracking benchmark built from 10 million real chess games, where models predict the exact board state reached after a sequence of legal moves. Alongside a held-out real-game split, we include an out-of-distribution split from uniformly random legal play, which tests whether models learn the transition rules rather than shortcuts from common human positions. Prior theoretical and empirical work has shown that Transformers struggle to state-track, while input-dependent linear RNNs require expressive state-transition matrices to do so. We therefore benchmark a causal Transformer, block-diagonal SLiCE, Mamba-3, and Gated DeltaNet with negative eigenvalues under a matched interface and training protocol. The recurrent models strongly outperform the Transformer at 3 and 8 million parameters. Real-game performance saturates above 18 million parameters, but the random-uniform split remains discriminative up to 40 million, exposing failures otherwise hidden by scale. Additionally, ablations show that less expressive state-transition mechanisms reduce performance on the out-of-distribution split for all three recurrent models. Together, these results establish Chess-World-Model as a practical large-scale benchmark for state tracking that exposes failures model scale would otherwise conceal.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in World models for language agents, one of 11 directions in World models. Builds on 31 of its 44 references that are in the map. First posted on arXiv 28 May 2026.
Builds on
31 of 44 references.
Closest ancestors · 4
Same direction, newest first
- 2025Tracking World States with Language Models: State-Based Evaluation Using Chess27 Aug
- 2024Evaluating the World Model Implicit in a Generative Model6 Jun
- 2024Emergent World Models and Latent Variable Estimation in Chess-Playing Language Models21 Mar
- 2023Emergent Linear Representations in World Models of Self-Supervised Sequence Models2 Sep
Other references · 27
- 2017Attention Is All You Needcited by 60,361
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2016Gaussian Error Linear Units (GELUs)cited by 3,502
- 2020GLU Variants Improve Transformercited by 1,804
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 1,039
15 more
- 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planningcited by 809
- 2018Recurrent World Models Facilitate Policy Evolutioncited by 785
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 532
- 2022Emergent World Representations: Exploring a Sequence Model Trained on a Synthetic Taskcited by 431
- 2022Neural Networks and the Chomsky Hierarchycited by 217
- 2024The Illusion of State in State-Space Modelscited by 187
- 2021Saturated Transformers are Constant-Depth Threshold Circuitscited by 139
- 2024Unlocking State-Tracking in Linear RNNs Through Negative Eigenvaluescited by 101
- 2026Mamba-3: Improved Sequence Modeling using State Space Principlescited by 89
- 2025DeltaProduct: Improving State-Tracking in Linear RNNs via Householder Productscited by 75
- 2024Theoretical Foundations of Deep Selective State-Space Modelscited by 74
- 2021Chess as a Testbed for Language Model State Trackingcited by 72
- 2025(How) Do Language Models Track State?cited by 34
- 2025Structured Linear CDEs: Maximally Expressive and Parallel-in-Time Sequence Modelscited by 18
- 2023Advancing Regular Language Reasoning in Linear Recurrent Neural Networkscited by 13
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2024Amortized Planning with Large-Scale Transformers: A Case Study on Chess
- 2026Finding the Move Is Not Winning the Game: XiangqiBench for Closed-Loop Evaluation of LLM Agents
- 2025LLM CHESS: Benchmarking Reasoning and Instruction-Following in LLMs through Chess
- 2025Out-of-distribution Tests Reveal Compositionality in Chess Transformers
- 2026Engineering Efficient Self-Play Chess: Search, Replay, and Throughput Under Limited Compute
- 2025Implicit Search via Discrete Diffusion: A Study on Chess
- 2025ChessArena: A Chess Testbed for Evaluating Strategic Reasoning Capabilities of Large Language Models
- 2024Superior Computer Chess with Model Predictive Control, Reinforcement Learning, and Rollout