The Gradient Does Not See Rank: Rank-Indifference in Matrix-CODI on ProsQA
On the map: LLM reasoning → Prompting and chain-of-thought reasoning
Abstract (arXiv)
Continuous chain-of-thought models compress reasoning into latent tokens. Matrix-valued variants, which route each latent token through a d x d matrix bottleneck, introduce rank as a single-sample structural observable on the latent matrix Z. If matrix latents carry parallel reasoning paths via superposition, rank should track them, and truncating Z to low rank should hurt accuracy on tasks whose solutions plausibly require multiple components. Across four training regimes of a matrix-CODI model (three on ProsQA, one on GSM8K-Aug below the learning threshold), the rank-k projection ablation curve is flat to within 0.6 percentage points. A three-seed replication yields 81.0 +/- 2.0 percentage points accuracy while the final effective rank of Z spans {4, 12, 13}; the loss does not reward any particular rank. To test whether rank-blindness arises from the flatten-then-project readout alone, we trained four readouts: a bilinear reparametrization, a bilinear-plus-GELU readout nonlinear in Z, an SVD-augmented readout feeding singular values through an MLP, and a quadratic readout in Z Z^T. All four rank-k curves remain flat (Spearman p-values 0.63, 0.14, 0.82, 0.46). The flat curves persist for readouts nonlinear in Z. A linear probe on Z underperforms a raw pretrained hidden state at target prediction (AUC 0.673 vs. 0.846). A negative control on vanilla GPT-2 SFT (no matrix bottleneck, no Z, three seeds, n=500) reproduces a flat rank-k curve under the same intervention paradigm with pooled-mean range 0.20pp, and a random-h sensitivity floor lands at the same accuracy: the rank-k ablation alone conflates rank-blindness with position-irrelevance.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Prompting and chain-of-thought reasoning, one of 17 directions in LLM reasoning. Builds on 17 of its 19 references that are in the map. First posted on arXiv 2 Sep 2026.
Builds on
17 of 19 references.
Closest ancestors · 7
Same direction, newest first
- 2026The Illusion of Superposition? A Principled Analysis of Latent Thinking in Language Models7 Apr
- 2026Dynamics Within Latent Chain-of-Thought: An Empirical Study of Causal Structure9 Feb
- 2025Latent Chain-of-Thought? Decoding the Depth-Recurrent Transformer2 Jul
- 2025Continuous Chain of Thought Enables Parallel Exploration and Reasoning29 May
- 2025Reasoning by Superposition: A Theoretical Perspective on Chain of Continuous Thought18 May
- 2025CODI: Compressing Chain-of-Thought into Continuous Space via Self-Distillation28 Feb
- 2024Training Large Language Models to Reason in a Continuous Latent Space9 Dec
Other references · 10
- 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 419
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 405
- 2017Implicit Regularization in Matrix Factorizationcited by 322
- 2020Implicit Regularization in Deep Learning May Not Be Explainable by Normscited by 119
- 2025LLM Latent Reasoning as Chain of Superpositioncited by 39
- 2024Weight decay induces low-rank attention layerscited by 27
- 2024Optimal ablation for interpretabilitycited by 25
- 2022Addressing Token Uniformity in Transformers via Singular Value Transformationcited by 14
- 2026On State Reduction in Linear Attentioncited by 6
- 2026State Rank Dynamics in Linear Attention LLMscited by 3
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Do Latent-CoT Models Think Step-by-Step? A Mechanistic Study on Sequential Reasoning Tasks
- 2026Think Wider: Mitigating Latent Rank Collapse in Implicit Chain-of-Thought Reasoning
- 2026Reasoning Theater: Disentangling Model Beliefs from Chain-of-Thought
- 2026Concept Subspaces Compute Beyond the Logit Lens: A Weights-Only Test for Locating Representations Upstream of Readout
- 2026What Makes Effective Supervision in Latent Chain-of-Thought: An Information-Theoretic Analysis
- 2026Emergent Search and Backtracking in Latent Reasoning Models
- 2026LoRi: Low-Rank Distillation for Implicit Reasoning
- 2026Invariant Reasoning Directions in Latent Trajectories of Language Models