Factorized Representations of Concurrent Temporal Variables in Pretrained Language Models
Abstract
Language models often need to maintain multiple contextual variables that evolve independently within a document. Prior work has identified structured representations of individual temporal or contextual quantities, but their joint organization remains less understood. We study this question using controlled documents with independently phased language-switch and topic-change timers. Across ten pretrained checkpoints, both timers are linearly decodable from the same hidden states. Removing a timer-associated centroid plane substantially reduces access to its associated variable while having little effect on the concurrent timer, although the planes are nonorthogonal and contain cross-timer information. Separate timer contributions also explain most reproducible variation among joint-state centroids. These signatures persist from 3 to 51 timer values and across Qwen3 checkpoints from 0.6B to 14B. Schedule position affects continuation probabilities, but the tested centroid displacement produces no consistent timer-specific output effect. Together, these results support approximate factorization based on selective accessibility and approximately additive joint-state geometry in overlapping subspaces.
Figures & tables
| Values | Rank | Self | Cross | Random | Cross/self | ||
|---|---|---|---|---|---|---|---|
| 2 | 3 | 0.91 | 1.75 | 84.9% | 0.15% | 0.18% | 0.18% |
| 5 | 6 | 0.89 | 2.50 | 75.3% | 0.44% | 0.31% | 0.56% |
| 10 | 11 | 0.87 | 3.25 | 79.9% | 2.05% | 0.53% | 2.58% |
| 20 | 21 | 0.87 | 3.48 | 79.8% | 2.96% | 0.69% | 3.71% |
| 50 | 51 | 0.86 | 3.98 | 84.6% | 4.86% | 0.76% | 5.71% |
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
| Sent. | Text | Lang. | Topic | ||
|---|---|---|---|---|---|
| 0 | The lamp is bright. | EN | house | 1 | 3 † |
| 1 | The mountain is high. | EN | field | 2 | 0 |
| 2 | Der Fluss ist breit. | DE | field | 0 | 1 |
| 3 | Der Stein ist grau. | DE | field | 1 | 2 |
| 4 | Der Wald ist dunkel. | DE | field | 2 | 3 † |
| 5 | The chair is heavy. | EN | house | 0 | 0 |
| Paper name | Checkpoint | Layers | Analyzed |
|---|---|---|---|
| Llama-3.2-1B | meta-llama/Llama-3.2-1B | 16 | 8 |
| Mistral-7B-v0.3 | mistralai/Mistral-7B-v0.3 | 32 | 16 |
| Qwen3-1.7B | Qwen/Qwen3-1.7B | 28 | 14 |
| Qwen3-1.7B-Base | Qwen/Qwen3-1.7B-Base | 28 | 14 |
| Qwen3-4B | Qwen/Qwen3-4B | 36 | 18 |
| pythia-1.4b | EleutherAI/pythia-1.4b | 24 | 12 |
| Language timer | Topic timer | |||||
|---|---|---|---|---|---|---|
| Model | Intact | Own | Other | Intact | Own | Other |
| Llama-3.2-1B | 0.891 | 0.423 | 0.886 | 0.460 | 0.390 | 0.459 |
| Mistral-7B | 0.917 | 0.433 | 0.909 | 0.513 | 0.403 | 0.506 |
| Qwen3-1.7B | 0.878 | 0.421 | 0.867 | 0.502 | 0.404 | 0.498 |
| Qwen3-1.7B-Base | 0.941 | 0.431 | 0.938 | 0.531 | 0.412 | 0.528 |
| Qwen3-4B | 0.864 | 0.428 | 0.851 | 0.522 | 0.414 | 0.512 |
| Own, random | Own, phase-matched | Own, within-document | Other, random | |
|---|---|---|---|---|
| Language | 84.5% | 84.0% | 81.4% | 1.6% |
| Topic | 58.4% | 63.0% | 53.6% | 2.3% |
| Timer | Intact | Own, cross-fitted | Other, cross-fitted | Own removed | Other removed |
|---|---|---|---|---|---|
| Language | 0.893 | 0.576 | 0.886 | 81% | 1.7% |
| Topic | 0.598 | 0.553 | 0.591 | 45% | 6.3% |
| Transferred fraction | Cross-ablation | Self-ablation | Above-chance accuracy |
|---|---|---|---|
| 0.00 | 0.6 | 16.5 | +0.164 |
| 0.25 | 1.1 | 11.2 | +0.109 |
| 0.50 | 4.4 | 8.1 | +0.081 |
| 0.75 | 10.4 | 8.5 | +0.084 |
| 1.00 | 11.0 | 10.8 | +0.110 |
| Model | Own plane | Concurrent plane | Placebo | Angle |
|---|---|---|---|---|
| SmolLM2-1.7B | +0.596 | +0.136 | +0.164 | |
| Qwen3-1.7B-Base | +0.537 | +0.259 | +0.149 | |
| OLMo-2-1B | +0.528 | +0.188 | +0.146 | |
| Mistral-7B-v0.3 | +0.515 | +0.206 | +0.166 | |
| Llama-3.2-1B | +0.451 | +0.153 | +0.117 | |
| gemma-2-2b | +0.438 | +0.217 | +0.105 |
| Interaction residual | Held-out variance explained | |||
|---|---|---|---|---|
| Model | Observed | Additive floor | Additive | Saturated |
| Llama-3.2-1B | 9.1% | 4.3% | 0.839 | 0.833 |
| Mistral-7B | 6.8% | 3.4% | 0.876 | 0.852 |
| Qwen3-1.7B | 12.8% | 6.0% | 0.758 | 0.761 |
| Qwen3-1.7B-Base | 7.5% | 3.1% | 0.868 | 0.874 |
| Qwen3-4B | 12.8% | 6.1% | 0.767 | 0.753 |
| Timer | Across | Split-half | Across / split-half | Shuffled |
|---|---|---|---|---|
| Language | 0.855 | 0.889 | 96% | +0.005 |
| Topic | 0.331 | 0.587 | 54% | +0.005 |
| Model | Interaction SEM | Shuffled | Split-half angle (lang./topic) |
|---|---|---|---|
| Qwen3-0.6B | 50.4% | ||
| Qwen3-1.7B | 50.6% | ||
| Qwen3-4B | 49.9% | ||
| Qwen3-8B | 50.1% | ||
| Qwen3-14B | 50.3% |
| Model | SEM | ||
|---|---|---|---|
| pythia-1.4b | 3.301 | -0.837 | |
| Qwen3-1.7B | 2.147 | -0.741 | |
| Phi-3-mini | 2.094 | -0.750 | |
| OLMo-2-1B | 1.997 | -0.767 | |
| Llama-3.2-1B | 2.262 | -0.280 | |
| SmolLM2-1.7B | 1.962 | -0.449 |
| Model | Target | Concurrent | Random | Target concurrent |
|---|---|---|---|---|
| Phi-3-mini | 0.0249 | 0.0231 | 0.0010 | 0.0018 |
| gemma-2-2b | 0.0243 | 0.0066 | -0.0070 | 0.0177 |
| OLMo-2-1B | 0.0173 | 0.0499 | -0.0015 | -0.0326 |
| pythia-1.4b | 0.0126 | -0.0164 | 0.0037 | 0.0290 |
| Mistral-7B-v0.3 | 0.0046 | 0.0215 | -0.0036 | -0.0169 |
| SmolLM2-1.7B | 0.0021 | 0.0324 | -0.1014 | -0.0303 |
Representation Geometry in Language Models
Representation geometry in language models studies how embeddings and hidden states organize concepts, computation, and behavior.
See all 421 papers