One Mechanism for Many Mental Spaces: A Shared Router over a Value Slot in Language Models

A subspace trained to control which mental space a value belongs to, only ever shown belief, or only ever shown counterfactuals, also controls the others, on three unrelated model families. The mechanism generalizes across space types; only its rank has to be large enough to matter.

Oliver Steele, Jiangtao Wen, Yuxing Han
Tsinghua SIGS · NYU Shanghai · 2026
arXiv:2607.10248 Qwen2.5-3B · Pythia-2.8b · Falcon3-3B

The setup

A model has to hold a base-world value and an alternative-space value for the same entity apart: a flower that's red in reality and purple in a painting, mentioned in the same sentence, queried later without the words “painting” or “reality” appearing again. Nine different builders open the alternative space: a painting, a belief, a dream, a movie, the past, yesterday, a hypothetical, a modal possibility, reported speech.1

Both values stay decodable from the model's residual stream at a later mention, across every builder type, generalizing across which entity, which attribute, and which kind of entity is involved.2 That's a necessary starting point, but decodability alone doesn't say whether one mechanism handles all nine builders or nine separate ones happen to leave a similar trace.

Coexisting values share one code

The binding is re-read from the source context each time it's queried, decaying like a retrieval effect, not a persistent register set once and forgotten.3 And it's low-rank: the per-builder-type directions live in a shared subspace (a handful of directions in the model's hidden state) of effective rank ~4–5 out of 10 possible dimensions, with one component carrying about 40% of the variance across builder types.

The space code also composes additively with entity identity (R² ≈ 0.92–0.95; the two factors nearly orthogonal, i.e. independent) and reaches the model's actual output: both the base and alternative values rise above chance through the late layers in all three families, written by a handful of late-layer heads. The heads that write the alternative-space value and the heads that write the base-world value don't overlap at all.4

Show writers of:
L31L32L338 attention heads per layer →alt: greenbase: blue

Direct logit attribution localizes writing to 3 late layers (Qwen2.5-3B, L31–L33). The image (alt) and base readouts are written by disjoint heads: 0 of 8 top movers overlap, in every model.

Real: the writing layers (Qwen2.5-3B L31–33) and the 0-of-8 mover-set overlap are the paper's fig-dla direct-logit-attribution result. Schematic: which specific head index is a top mover is a drawing of “8 heads per layer, two disjoint sets,” not the exact per-head attribution.

Rank is what makes it real

Hand-built directions (a simple difference between alt-space and reality activations, or steering along it) sit right at, or inside, the band a matched random direction reaches by chance. A learned subspace is what actually controls the readout, and how much rank it needs is itself a measured, swept quantity.

Alternative space:
Patch the space index at rank:
queries
In reality the cup is bluebase world
In the painting the cup is greenalternative space
value slotbasebluealtgreen
Consider the cup.neutral probe
space indexselects the space
blueanswer
flip toward the alternative value: 0.000

Base run: the neutral probe reads the base-world value. Patch the space index to flip it toward the alternative.

One subspace, many spaces: the same learned subspace, trained on one builder, applied to another (rank-8 transfer index; bar 0.70):
Qwen2.5-3Bwithin 1.00cross 0.730
Pythia-2.8bwithin 1.00cross 0.885
Falcon3-3Bwithin 1.00cross 0.711

Real: the flip rates (random / hand-built rank-1 stuck at 0.583; learned rank-1 0.883, rank-2 0.967, rank-4+ 1.000) and cross-type transfer indices (0.71–0.885, bar 0.70) are the paper's tab-das-rank and tab-transfer-index, at layer L34 on Qwen2.5-3B. Schematic: the circuit layout, the direction-arrows in the space index, and the traveling slice are a drawing of the interchange, not a readout of activations.

A random or hand-built rank-1 direction gets stuck at 0.583, the leakage floor. A learned rank-1 subspace already reaches 0.883; rank-2 reaches 0.967; rank-4 and up saturate at 1.000.5 Full control needs a subspace, not just a direction, and the subspace it needs is small.

That same rank result is also what makes the cross-type generality claim from the overview page meaningful: a rank-8 subspace trained on one builder type transfers to control an entirely different one at a transfer index of 0.71–0.89 across the three families, and rank-4 already gets within 0.06 of rank-8's transfer, so the shared mechanism itself is compact, not merely present somewhere in a big subspace.

What's said vs. what's used

If a model reports one color but derives a rule-based conclusion from it, are those the same internal value read out twice, or two things a single subspace happens to move together?

They're separable. In a single-space color-to-recipient sorting task, a subspace trained on the reported color fully flips the report (1.00) while leaving the rule-derived consequent almost untouched (0.00); a subspace trained on the consequent does the reverse.6 The two subspaces are nearly orthogonal. A model's stated answer and what it actually reasoned from can be moved independently, a mechanistic foothold for why a model's explanation of its own reasoning needn't match what drove the answer.

A taxonomy, but no special belief

Formal semantics predicts belief should be a distinct case: a different logic from counterfactual or fictional reference. The shared subspace does recover a coarse taxonomy: builder types that formally group together (attitude reports, intensional modals, temporal reference) transfer a little better within their group than across groups.7

But belief doesn't get its own boundary beyond that generic within-class effect. The mechanism does recover a semantic line; the line it uses simply groups belief with every other non-actual context rather than setting it apart. That reading is the paper's answer to the introduction's linguistic question: Fauconnier's unification, not the semanticist's separation, describes what the causal machinery is doing.

The belief companion

Belief earns its own paper because it carries independent philosophical and psychological weight (theory of mind, propositional attitudes) that this paper's cross-type sweep doesn't have room to develop, not because the mechanism needs a special case. The companion goes deep on exactly how belief fills its slot: directly when asserted, by a visibility-gated lookback when it has to be derived from what a character could see.

Belief in depth
Belief–Reality Separation Lives in Routing
The value slot carries no belief/reality tag on its own; two different routing subspaces, dissociated from each other, decide which frame a query reads.

← Mental spaces overview