How Should Reasoning Be Organized in a Transformer's Latent Space?
cs.AI
Submitted: 2026-09-12
Updated: 2026-09-29
Comments: 20 pages,4 figures
License: http://creativecommons.org/licenses/by-nc-sa/4.0/
The gist: Large language models solve hard problems through intermediate computations across multi-step reasoning.
Abstract
Large language models solve hard problems through intermediate computations across multi-step reasoning. Traditional chain-of-thought encodes these computations as tokens. Recent continuous and recurrent methods instead move partial computations into fixed-dimensional latent states, where a single thought can superpose multiple alternatives. This raises a fundamental design question:what should continuous thoughts preserve as reasoning proceeds? An intuitive approach discards past computations and keeps only the current reasoning frontier. Storing more items seems to dilute states and waste limited representational capacity. We show this intuition can be incorrect. Under identical downstream computations, cumulative superposition retaining full reasoning history can require lower representational dimensions than frontier-only superposition holding only current alternatives. At fixed hidden width, this advantage allows latent reasoners to retain more valid evidence, distinguish more plausible downstream outcomes, and delay the point where compressed states turn unreliable. This counter-intuitive effect emerges because informative historical components coherently reinforce each other, while unrelated alternatives bring random interference. This perspective also answers a practical design question: how should models weight memories accumulated inside latent states when their future use is unknown? Across reusable weighted superpositions, prioritizing a small set of recent or salient items produces weakly-represented memories that bottleneck subsequent attention. Uniform cumulative weighting avoids this flaw, and we prove it is minimax-optimal for robust future reasoning. Our results turn superposition from an observed latent-space effect into a design principle: balanced cumulative memory lets a fixed representational budget support more reliable, reusable computations.
Sources
- Superposition as Lossy Compression: Measure with Sparse Autoencoders and Connect to Adversarial Vulnerability
- Quantifying the Necessity of Chain of Thought through Opaque Serial Depth
- Tackling Polysemanticity with Neuron Embeddings
- Simply Stabilizing the Loop via Fully Looped Transformer
- Continuous Chain of Thought Enables Parallel Exploration and Reasoning
- Disentangling Polysemantic Neurons with a Null-Calibrated Polysemanticity Index and Causal Patch Interventions
- Training Large Language Models to Reason in a Continuous Latent Space
- Understanding Reasoning in Chain-of-Thought from the Hopfieldian View
- Making Long-Context Language Models Better Multi-Hop Reasoners
- Tyler: Typed Latent Reasoning for Language Models -- When to Think, What to Compute, and How Much to Allocate
- On Expressive Power of Looped Transformers: Theoretical Analysis and Enhancement via Timestep Encoding
- Buffer of Thoughts: Thought-Augmented Reasoning with Large Language Models
- Language models can learn implicit multi-hop reasoning, but only if they have lots of training data
- Emergence of Superposition: Unveiling the Training Dynamics of Chain of Continuous Thought
- Reasoning by Superposition: A Theoretical Perspective on Chain of Continuous Thought
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems