mlsanityData to 14 Oct 2026

The Gradient Does Not See Rank: Rank-Indifference in Matrix-CODI on ProsQA

arXiv 2609.03090 ↗First posted 2 Sep 2026cs.LGcited by 2

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

  1. 2026The Illusion of Superposition? A Principled Analysis of Latent Thinking in Language Models7 Apr
  2. 2026Dynamics Within Latent Chain-of-Thought: An Empirical Study of Causal Structure9 Feb
  3. 2025Latent Chain-of-Thought? Decoding the Depth-Recurrent Transformer2 Jul
  4. 2025Continuous Chain of Thought Enables Parallel Exploration and Reasoning29 May
  5. 2025Reasoning by Superposition: A Theoretical Perspective on Chain of Continuous Thought18 May
  6. 2025CODI: Compressing Chain-of-Thought into Continuous Space via Self-Distillation28 Feb
  7. 2024Training Large Language Models to Reason in a Continuous Latent Space9 Dec

Other references · 10

  1. 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 419
  2. 2019Implicit Regularization in Deep Matrix Factorizationcited by 405
  3. 2017Implicit Regularization in Matrix Factorizationcited by 322
  4. 2020Implicit Regularization in Deep Learning May Not Be Explainable by Normscited by 119
  5. 2025LLM Latent Reasoning as Chain of Superpositioncited by 39
  6. 2024Weight decay induces low-rank attention layerscited by 27
  7. 2024Optimal ablation for interpretabilitycited by 25
  8. 2022Addressing Token Uniformity in Transformers via Singular Value Transformationcited by 14
  9. 2026On State Reduction in Linear Attentioncited by 6
  10. 2026State Rank Dynamics in Linear Attention LLMscited by 3

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Do Latent-CoT Models Think Step-by-Step? A Mechanistic Study on Sequential Reasoning Tasks
  2. 2026Think Wider: Mitigating Latent Rank Collapse in Implicit Chain-of-Thought Reasoning
  3. 2026Reasoning Theater: Disentangling Model Beliefs from Chain-of-Thought
  4. 2026Concept Subspaces Compute Beyond the Logit Lens: A Weights-Only Test for Locating Representations Upstream of Readout
  5. 2026What Makes Effective Supervision in Latent Chain-of-Thought: An Information-Theoretic Analysis
  6. 2026Emergent Search and Backtracking in Latent Reasoning Models
  7. 2026LoRi: Low-Rank Distillation for Implicit Reasoning
  8. 2026Invariant Reasoning Directions in Latent Trajectories of Language Models