MATH · IN · MODELS
structures / Manifolds / Nonlinear World-Model Decodability

Nonlinear World-Model Decodability

PROPERTYlabeled-data-propertyintermediatehow it's classified →

A discrete world-state (e.g. a board configuration) is reliably decodable from a network's internal activations only by a nonlinear probe (a small MLP), not a linear one — the representation exists and is causally load-bearing, but the labeled classes it induces are not linearly separable in the model's own basis.

Replicationcomputed from the corpus — never hand-assigned
2 papers · no shared authors2 architecture classes · across papers2 domains · across papers3 model families · across papers
Filled = two or more values reported by papers that share no author — replication. Outlined = two or more values, but all from a single study — breadth, not replication. Grey = a single value. Derived from paper authorship and each model's architecture class, domain and family; it updates itself when a paper is added.

Definition

Fix a network’s activation space Rd\mathbb{R}^d and a discrete labeling of inputs by some ground-truth latent state yYy \in \mathcal{Y} (e.g. the full configuration of a game board). The labeled classes {x:y(x)=c}cY\{x : y(x) = c\}_{c\in\mathcal{Y}} exhibit nonlinear world-model decodability if a nonlinear probe fθf_\theta (e.g. a 1-2 hidden-layer MLP) trained on activations reliably predicts yy, while the same classes fail the Linear Separability test — no single hyperplane per class (or small set of hyperplanes) achieves comparable accuracy. Reliability is typically established two ways: decoding accuracy of fθf_\theta against the true y(x)y(x), and causal validity — intervening on the activations to flip fθf_\theta‘s decoded state and checking that the network’s own downstream behavior (e.g. its next move) changes consistently with the intervention, ruling out the probe merely fitting an epiphenomenal correlate.

Relative to linear-separability

This is the negative/nonlinear counterpart of Linear Separability, not an unrelated finding: the same labeled classes are being tested, but the question is whether the representation itself only becomes causally identifiable class-by-class once probed with a nonlinear function, versus already being linearly readable in the model’s own coordinate basis. A representation can migrate from one regime to the other under a change of basis (not of the network) — the canonical case in this map is Othello-GPT, whose board-state representation was first reported nonlinear in this basis, then shown to be linear once probed in the network’s own relative (mine/yours) frame rather than the human-preconceived absolute (black/white) one — see othello-relative-frame-direction. Nonlinear decodability by itself is therefore evidence of some systematic, causally load-bearing structure, but does not by itself imply either linear or nonlinear geometry is the “true” shape — only that no linear readout was found in the basis tested.

Key evidence

The original Othello-GPT world-model finding: small-MLP probes reliably decode the full board state from a real trained transformer, and activation patching that flips the decoded state causally changes subsequent move predictions. Li, Hopkins, Bau, Viégas, Pfister & Wattenberg (2023) train an 8-layer, 8-head, 512-dim GPT-style transformer on Othello move sequences (synthetic self-play and championship human games) and show nonlinear (MLP) probes decode the full board state reliably at every move — a nonlinear emergent world-model representation. Activation-patching interventions that flip the probe-decoded board state causally change the model’s subsequent move predictions (validated via move-legality checks, ~0.01% error synthetic / ~5.17% championship, and latent-saliency-map analysis). This is the foundational paper behind Nanda, Lee & Wattenberg (2023)‘s later linear reframing of the same phenomenon in the model’s relative frame (already in this map, see othello-relative-frame-direction) and behind Yuan & Søgaard (2025)‘s cross-architecture Procrustes-alignment follow-up. See li-etal-2023-nonlinear-probes-decode-a-full-othello-board-state-from-othello-gpts-activations-and-activation-patching-that-flips-the-decoded-state-causally-changes-subsequent-move-predictions.

A real architecture-dependent split between the linear and nonlinear regimes, without any change of probing basis. Zhang (2026) probes two real RL world models trained on real Atari Breakout/Pong replay data — IRIS (discrete VQ-VAE-token transformer) and DIAMOND (continuous diffusion UNet) — for game-state variables (ball/paddle position, score) using both linear and nonlinear (MLP) probes at every layer. IRIS is approximately linearly decodable throughout (selectivity gap between MLP and linear probe accuracy, Delta <= 0.06 at every layer), while DIAMOND shows an “inverted-V”: linear decodability peaks only at the UNet’s compression bottleneck and requires a nonlinear probe everywhere else in the network — the same discrete-vs-continuous architectural distinction determines which regime a given real world model falls into, rather than a change of basis. Causal validation: activation patching along the probe-derived direction shifts predicted next-state token distributions consistent with the intervention (r=0.97), and multi-baseline token-ablation on IRIS’s spatial token grid identifies score/brick-region tokens as most causally important (rank consistency rho > 0.92). See zhang-2026-real-atari-world-models-iris-is-linearly-decodable-throughout-while-diamond-needs-nonlinear-probes-except-at-its-unet-bottleneck-both-causally-validated-via-activation-patching.

Key papers

  • Li, Hopkins, Bau, Viégas, Pfister & Wattenberg (2023). Emergent World Representations: Exploring a Sequence Model Trained on a Synthetic Task. arXiv:2210.13382 — origin of the finding above.
  • Nanda, Lee & Wattenberg (2023). Emergent Linear Representations in World Models of Self-Supervised Sequence Models. arXiv:2309.00941 — shows the same OthelloGPT board state becomes linear under a change of probing basis; see othello-relative-frame-direction and Linear Representation Hypothesis.

Found in (2 observations · 3 families)

OthelloGPT

Emergent World Representations: Exploring a Sequence Model Trained on a Synthetic Task (2023)measured

Nonlinear (small-MLP) probes reliably decode the full Othello board state from an 8-layer, 512-dim GPT-style transformer's internal activations at every move, and activation-patching interventions that flip the decoded board-state representation causally change the model's subsequent move predictions

Details

Li, Hopkins, Bau, Viegas, Pfister & Wattenberg (2023, the original Othello-GPT paper) train an 8-layer, 8-head, 512-dim GPT-style transformer on sequences of legal Othello moves (synthetic self-play and championship human games), and show nonlinear (MLP) probes reliably decode the full board state from internal activations at each move -- a nonlinear emergent world-model representation, clearing scope on criterion (a). Causally, activation-patching interventions that flip the probe-decoded board-state representation change the model's subsequent move predictions accordingly (validated via move-legality and latent-saliency-map analyses; move-legality error rates ~0.01% synthetic, ~5.17% championship), clearing scope on criterion (b). This is the foundational world-model paper behind later linear reframings of the same phenomenon (Nanda et al. 2023, already in this map) and cross-architecture alignment follow-ups (Yuan & Sogaard 2025, already in this map). See [[nonlinear-world-model-decodability]].

models: OthelloGPT · method: Mlp probing, Causal interventions (steering)

IRIS (VQ-VAE + Transformer world model)

What Do World Models Learn in RL? Probing Latent Representations in Learned Environment Simulators (2026)measured

Real Atari world models IRIS and DIAMOND fall into architecture-dependent linear vs. nonlinear decodability regimes

Details

Zhang (2026) probes two real RL world models trained on real Atari Breakout/Pong replay data -- IRIS (discrete VQ-VAE-token transformer) and DIAMOND (continuous diffusion UNet) -- for game-state variables using both linear and nonlinear (MLP) probes at every layer [zhang-2026-what-do-world-models-learn-in-rl-probing-latent-representations] IRIS is approximately linearly decodable throughout (selectivity gap Delta <= 0.06 at every layer), while DIAMOND shows an inverted-V: linear decodability peaks only at the UNet's compression bottleneck and requires a nonlinear probe elsewhere -- an architecture-dependent split, not a change of probing basis [zhang-2026-what-do-world-models-learn-in-rl-probing-latent-representations] Activation patching along the probe-derived direction shifts predicted next-state token distributions consistent with the intervention (r=0.97), and multi-baseline token ablation on IRIS's spatial token grid identifies score/brick-region tokens as most causally important (rank consistency rho > 0.92) [zhang-2026-what-do-world-models-learn-in-rl-probing-latent-representations]

models: IRIS (Atari 100k, Breakout/Pong) · method:

DIAMOND (diffusion UNet world model)

What Do World Models Learn in RL? Probing Latent Representations in Learned Environment Simulators (2026)measured

Real Atari world models IRIS and DIAMOND fall into architecture-dependent linear vs. nonlinear decodability regimes

Details

Zhang (2026) probes two real RL world models trained on real Atari Breakout/Pong replay data -- IRIS (discrete VQ-VAE-token transformer) and DIAMOND (continuous diffusion UNet) -- for game-state variables using both linear and nonlinear (MLP) probes at every layer [zhang-2026-what-do-world-models-learn-in-rl-probing-latent-representations] IRIS is approximately linearly decodable throughout (selectivity gap Delta <= 0.06 at every layer), while DIAMOND shows an inverted-V: linear decodability peaks only at the UNet's compression bottleneck and requires a nonlinear probe elsewhere -- an architecture-dependent split, not a change of probing basis [zhang-2026-what-do-world-models-learn-in-rl-probing-latent-representations] Activation patching along the probe-derived direction shifts predicted next-state token distributions consistent with the intervention (r=0.97), and multi-baseline token ablation on IRIS's spatial token grid identifies score/brick-region tokens as most causally important (rank consistency rho > 0.92) [zhang-2026-what-do-world-models-learn-in-rl-probing-latent-representations]

models: DIAMOND (Atari 100k, Breakout/Pong) · method: