MATH · IN · MODELS
structures / Hypotheses / Belief State Geometry Hypothesis (Mixed-State Presentation)

Belief State Geometry Hypothesis (Mixed-State Presentation)

CLAIMhypothesisadvancedhow it's classified →

For a sequence-generating process with a known hidden state (an HMM / epsilon-machine), a predictive model's residual stream linearly embeds the current Bayesian belief b_t = p(hidden state | x_{1:t}) — a point in a probability simplex over the process's causal states — via the process's Mixed-State Presentation (MSP), with next-token predictions recoverable as a barycentric combination of per-vertex predictive distributions.

Replicationcomputed from the corpus — never hand-assigned
4 papers · no shared authors2 architecture classes · across papers3 domains · across papers4 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.

Relative to Attention–MLP Sufficiency Staging Hypothesis: both hypotheses descend from the same computational-mechanics tradition (Shai et al. 2024) and both concern a network’s internal representation of a process’s state, but this hypothesis claims a curved Mixed-State-Presentation simplex geometry for belief over a stationary, ergodic HMM’s hidden states, while Attention–MLP Sufficiency Staging Hypothesis claims a two-stage linear-subspace separation (a full world state vs. a compressed task-specific state, tied to a specific attention→MLP sublayer boundary) for a non-stationary, non-ergodic process — a genuinely different shape claim about a different architectural location, not a restatement.

Statement

Let a data-generating process have a finite set of hidden causal states with known transition and emission dynamics (a hidden Markov model or epsilon-machine). At each position tt, an ideal Bayesian observer maintains a belief bt=p(statex1:t)b_t = p(\text{state}\mid x_{1:t}) — a point in the probability simplex Δk1\Delta^{k-1} over the kk hidden states. The hypothesis claims a sequence-predicting transformer trained on this process’s outputs linearly embeds btb_t in its residual stream, via a specific map called the Mixed-State Presentation (MSP): the image of Δk1\Delta^{k-1} under repeated Bayesian updating, which is generically a curved, self-similar (sometimes fractal) point set rather than a flat simplex. The model’s next-token distribution should then decompose as a barycentric combination: p(xt+1x1:t)=vbt(v)p(xt+1v)p(x_{t+1}\mid x_{1:t}) = \sum_{v} b_t^{(v)}\, p(x_{t+1}\mid v), a convex combination of each vertex’s own predictive distribution, weighted by belief coordinates.

Intuition

Most representation-geometry hypotheses on this map (Linear Representation Hypothesis, Conceptual Belief Space Hypothesis) posit structure for informally-defined, human-labeled concepts, where there is no independently derivable “correct” geometry to check against — only self-consistency and causal tests. This hypothesis is unusual in being checkable against a known ground truth: because the generating process’s causal states and their transition dynamics are specified in advance, the exact target geometry (the MSP, however curved or fractal) can be computed analytically and compared directly to what a trained network’s activations reveal, rather than being invented after the fact to fit the data.

Properties

  • Grounded in a known generative process, unlike Conceptual Belief Space Hypothesis. Both hypotheses describe “belief that updates as a sequence is read,” but this one concerns a probability distribution over the states of a specified hidden Markov process (with a provably-derivable target geometry), while Conceptual Belief Space Hypothesis concerns belief over informally-defined semantic concepts (emotions, genres) whose target geometry has no independent derivation — a genuinely different grounding for a superficially similar idea, not a restatement of it.
  • The target geometry is generically curved, not a flat simplex. Repeated Bayesian updating on processes like Mess3 (a 3-state HMM) or Tom Quantum/Bloch-Walk processes produces an MSP that is a curved, often fractal, self-similar subset of the simplex’s affine hull — the flat (k1)(k-1)-simplex of Polytope (Simplex) is the convex hull containing the MSP, not the MSP itself.
  • Barycentric decomposability is the operational signature. The distinguishing, falsifiable prediction is that a convex combination of per-vertex predictive distributions, weighted by recovered belief coordinates, predicts the model’s actual next-token distribution better than the single best individual feature or latent direction can — a genuine mixture claim, not just “some feature correlates with state.”
  • Must be distinguished from a tiling artifact. A set of latents that individually specialize to disjoint sub-regions of state space (see Manifold capture theory (subspace capture / tiling / dilution)‘s tiling/shattering regime) can mimic an apparent simplex-mixture signal near the vertices without any latent actually encoding a graded mixture; the discriminating test is whether the barycentric combination’s advantage holds specifically in the interior of the simplex, where no single specialized latent can substitute for a true mixture coordinate.
  • Predicts vertex-directed causal steering. Adding an intervention vector that shifts the recovered belief coordinate toward a target vertex should shift the model’s generated continuations toward that vertex’s characteristic predictive distribution, in proportion to the shift — a causal, not just correlational, consequence of the geometry actually being used.

Key evidence

Shai, Marzen, Teixeira, Gietelink Oldenziel & Riechers (2024, NeurIPS spotlight) originate this hypothesis: they derive the Mixed-State Presentation analytically for known epsilon-machines (Mess3, RRXOR, and related processes) and then train real GPT-2-style transformers from scratch, via ordinary gradient descent, on sequences generated by these processes. Linear regression from the theoretically-derived belief coordinates to the trained network’s own residual-stream activations recovers the predicted (often fractal, Sierpinski-triangle-like for Mess3) MSP geometry with high fit quality at specific layers/token positions, and next-token predictions decompose as the predicted barycentric combination of per-vertex distributions — the founding empirical confirmation for this node, on a real trained network even though the training data is synthetically generated by a known HMM. See shai-etal-2024-real-gpt-2-style-transformers-trained-from-scratch-on-hmm-generated-sequences-linearly-embed-the-theoretically-predicted-fractal-mixed-state-presentation-belief-geometry-in-their-residual-stream.

Kamel, Rastogi, Ma, Ranganathan & Zhu (2025) extend this hypothesis beyond a hand-derivable epsilon-machine to a genuinely learned stochastic game with no closed-form target geometry: a GPT-2-style transformer (87M parameters, 12 heads) trained from scratch on 2M+ synthetically-generated poker hand histories (Poker Hand History / PHH format). PCA of the model’s own activations reveals a recurring triangular geometry — resembling a belief-state simplex — among hand-strength clusters, alongside a strong linear probe for deterministic hand-rank (~80% accuracy) and t-SNE/UMAP clustering by conceptual similarity. Unlike Shai et al.’s epsilon-machine setting, the ground-truth belief geometry here (equity under uncertain opponent hands) cannot be derived in advance and checked against, and the equity/belief-state sub-claim itself rests only on a nonlinear (2-layer MLP, r=0.59) probe with no causal steering performed — flagged accordingly as suggestive, not confirmatory, evidence for the hypothesis on a real learned game rather than a fully specified generative process. See kamel-etal-2025-a-trained-poker-transformer-shows-a-triangular-pca-belief-state-geometry-and-linear-hand-rank-decoding.

Exercises

Base

  1. Why can the exact target geometry (the MSP) be computed in advance for this hypothesis, when it cannot for Conceptual Belief Space Hypothesis?
Solution

Because the generating process here is a fully specified hidden Markov model / epsilon-machine — its states, transition probabilities, and emission distributions are all known by construction, so the Bayes-optimal belief-updating map and its image (the MSP) can be derived analytically without observing the trained network at all. Conceptual Belief Space Hypothesis‘s concepts (emotions, genres) have no such formal generative specification; their “correct” geometry can only be estimated empirically, from data, with no independently computable ground truth to check the estimate against.

  1. A flat 2-simplex (triangle) and a curved, fractal MSP inscribed within it are both described in the Statement. Which one is the object Polytope (Simplex) already covers, and which is new here?
Solution

Polytope (Simplex) covers the flat convex hull (the bounding triangle itself, generically a (k1)(k-1)-simplex with kk vertices). The MSP — the specific curved, self-similar point set that Bayesian updating actually traces out inside that hull — is the new object this hypothesis introduces; the flat simplex is a prerequisite container, not the claim itself.

Middle

  1. Suppose an SAE cluster’s latents perfectly tile three disjoint regions near the vertices of a belief simplex, with no latent responding anywhere in the interior. Explain concretely why a barycentric-combination prediction could still numerically outperform the single best latent near the vertices, yet this would not count as confirming evidence for the hypothesis.
Solution

Near a vertex, the belief coordinate is close to one-hot (dominated by a single component), so a weighted sum that is nearly all one vertex-latent’s prediction can trivially match or slightly beat that latent alone (e.g. by softly blending in a small correction from neighboring latents), producing a numerically positive advantage without any latent ever encoding a genuine graded mixture. This is exactly the tiling artifact: the win is concentrated at the extremes, where a specialized detector already suffices, and disappears in the interior — which is why the hypothesis’s operational test specifically requires the advantage to hold on interior (non-vertex) samples, where no single tiling latent can substitute for an actual mixture coordinate.

  1. A study finds a positive causal-steering effect toward a target vertex, but no significant barycentric predictive advantage for that same cluster. What does this combination suggest about which part of the hypothesis is and is not supported?
Solution

A positive steering effect without a predictive advantage suggests the recovered direction is causally efficacious (pushing along it does shift behavior toward the target vertex) but that this alone doesn’t establish it’s a genuine mixture coordinate — the same causal push could work via a single dominant, non-mixture feature that happens to correlate with the target state, rather than via belief geometry that continuously interpolates between vertices. The hypothesis’s full claim (a true barycentric belief representation, not just a steerable direction) is best supported only when both signals — predictive advantage and causal steering — converge on the same axis, as the stronger joint criterion.

Pro

  1. Formalize the “tiling artifact” test using held-out R2R^2: let Rbary2R^2_{\text{bary}} be the barycentric combination’s held-out coefficient of determination and Rbest2R^2_{\text{best}} the best single latent’s, both computed separately on near-vertex and simplex-interior sample subsets. State a precise pair of inequalities across both subsets that would constitute strong, artifact-resistant evidence for the hypothesis, and explain why requiring the inequality on only the near-vertex subset is insufficient.
Solution

Strong evidence requires Rbary2>Rbest2R^2_{\text{bary}} > R^2_{\text{best}} on both the near-vertex subset and the simplex-interior subset (or, at minimum, a significant advantage on the interior subset alone, since that is the harder-to-fake regime). Requiring it only near vertices is insufficient because — as in Exercise 3 — a one-hot-dominated belief coordinate near a vertex can produce a numerically positive barycentric advantage purely from tiling latents with no true mixture encoding; only the interior subset, where the belief coordinate is a genuine non-trivial mixture no single tiling latent can approximate, discriminates a real mixture representation from an artifact of vertex-localized specialization.

Found in (4 observations · 4 families)

Gemma

Finding Belief Geometries with Sparse Autoencoders (2026)measured

SAE latents in Gemma-2-9B show preliminary simplex-structured belief geometry

Details

Levinson extends the belief-state-simplex framework from toy transformers to a real LLM's SAE latent space, fitting 13 priority latent clusters in a Gemma-2-9B layer-20 JumpReLU SAE with archetypal simplex fitting against 3 null-cluster controls [levinson-2026] On the barycentric predictive test, 5 of 13 real clusters show a significant advantage over the best single latent versus 0 of 3 null clusters, indicating recovered structure rather than a tiling artifact [levinson-2026] Causal steering along vertex directions gives positive scores for all 8 qualifying real clusters, but null-cluster scores overlap substantially, limiting discriminative power [levinson-2026] Only cluster 768_596 shows joint predictive and causal convergence, and the author frames the evidence as preliminary (single model and layer, modest effect sizes, a phantom vertex) [levinson-2026]

models: Gemma-2-9B · method: Sparse Autoencoders (SAE), k-Subspace Clustering, Archetypal Analysis / Simplex Fitting (AANet), Barycentric Predictive Advantage Test, Causal interventions (steering)

Poker Hand History Transformer (Kamel, Rastogi, Ma, Ranganathan & Zhu)

Emergent World Beliefs: Exploring Transformers in Stochastic Games (2025)measured

A poker-trained transformer's own activations show a triangular PCA belief-state geometry

Details

Kamel, Rastogi, Ma, Ranganathan & Zhu (2025) train a real GPT-2-style transformer (87M params, 12 heads) from scratch on 2M+ synthetically-generated poker hand histories (PHH format) [kamel-etal-2025-emergent-world-beliefs-transformers-stochastic-games] A linear probe decodes deterministic hand-rank at ~80% held-out accuracy, and PCA/t-SNE/UMAP of the model's own activations reveals genuine clustering by hand-rank and conceptual similarity, including a recurring triangular geometry resembling a belief-state simplex [kamel-etal-2025-emergent-world-beliefs-transformers-stochastic-games] The equity/belief-state sub-claim rests only on a nonlinear (2-layer MLP) probe (r=0.59); no causal intervention (steering along probe directions) is performed, flagged as suggestive rather than confirmatory evidence [kamel-etal-2025-emergent-world-beliefs-transformers-stochastic-games]

models: Custom GPT-2-style transformer (87M params, trained on synthetic Poker Hand History trajectories) · method: Linear probing, PCA

Bayesian Wind Tunnel Mamba

The Bayesian Geometry of Transformer Attention (2026)measured

A Mamba model discovers the corner geometry of the belief simplex

Details

Agarwal, Dalal & Misra train a from-scratch Mamba/SSM on HMM filtering and find its final-layer representations organize into five discrete clusters, one per hidden state [agarwal-etal-2026] The paper describes this as discovering the corner geometry of the belief simplex, concentrating at the vertices rather than the transformer's continuous interior-populated Mixed-State Presentation [agarwal-etal-2026] A permutation/relabeling test confirms the clustering tracks genuine posterior structure, not arbitrary state IDs [agarwal-etal-2026] Mamba realizes belief accumulation and belief transport but not random-access binding, distinguishing it from the transformer's richer continuous geometry [agarwal-etal-2026]

models: Bayesian Wind Tunnel Mamba (HMM filtering task, 9 layers, d_model=256, state dim 16) · method: PCA, Geometric analysis

Shai et al. (2024) Custom Epsilon-Machine Transformers

Transformers Represent Belief State Geometry in Their Residual Stream (2024)measured

Real trained transformers linearly embed the theoretically predicted fractal belief-state geometry

Details

Shai, Marzen, Teixeira, Gietelink Oldenziel & Riechers derive the Mixed-State Presentation (MSP) analytically for known epsilon-machines such as Mess3 and RRXOR, a curved, often fractal, self-similar image of the belief simplex under repeated Bayesian updating [shai-etal-2024-transformers-represent-belief-state-geometry-residual-stream] Real GPT-2-style transformers trained from scratch by gradient descent on sequences generated by these processes linearly embed the predicted MSP geometry in their residual stream, recovered via linear regression from theoretically-derived belief coordinates to the network's own activations [shai-etal-2024-transformers-represent-belief-state-geometry-residual-stream] Next-token predictions decompose as the predicted barycentric combination of per-vertex predictive distributions, weighted by the recovered belief coordinates, matching the operational signature of the Belief State Geometry Hypothesis [shai-etal-2024-transformers-represent-belief-state-geometry-residual-stream]

models: Custom GPT-2-style transformer (trained on Mess3 epsilon-machine sequences) · method: