mlsanityData to 7 Oct 2026

Thinking Deeper, Not Longer: Memory-Efficient Test-Time Reasoning with Depth-Recurrent Transformers for Compositional Generalization

arXiv 2603.21676 ↗First posted 23 Mar 2026cs.LG cs.AI cs.CLcited by 1

On the map: LLM reasoning → Test-time compute and efficient thinking

Abstract (arXiv)

Standard Transformers have a fixed computational depth, limiting their ability to generalize to tasks that require variable-depth reasoning. The usual remedy, Chain-of-Thought (CoT), spends tokens to reason, inflating the key--value cache and making latency grow with the step count, so memory becomes the limiting cost when reasoning is served over large query batches. We study a depth-recurrent Transformer that decouples computational depth from parameter count by iterating a shared-weight block, so that each added reasoning step costs flat memory and linear latency, with no token generation. Three ingredients keep the recurrence stable for 20+ thinking steps: a silent thinking objective that supervises only the final output, LayerScale initialization, and an identity-biased gate that opens a gradient highway across steps. We characterize it on three compositional domains with decreasing structural bias: graph reachability (adjacency masking), nested boolean logic (relative positioning), and unstructured relational text (no positional cue). We find a computational frontier: accuracy climbs once the thinking-step count meets the task's complexity, reaching near-perfect performance on the two structured tasks and a lower plateau on unstructured text. How it climbs depends on the structural bias---abruptly from chance on the graph task, gradually on the other two. Depth recurrence extrapolates beyond the training range: it succeeds on the graph task where fixed-depth models barely extrapolate, and on the two sequence tasks comes within two points of fixed-depth Transformers that use 4--6.4× more parameters. On the graph task, whose adjacency mask makes propagation depth verifiable, intermediate per-step supervision---a standard recipe for deep iterative models---consistently harms this extrapolation. We release the code for reproducibility.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 19 of its 26 references that are in the map. First posted on arXiv 23 Mar 2026.

Builds on

19 of 26 references.

Closest ancestors · 2

Same direction, newest first

  1. 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameters6 Aug
  2. 2023Think before you speak: Training Language Models With Pause Tokens3 Oct

Other references · 17

  1. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  2. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  3. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
  4. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  5. 2022Large Language Models are Zero-Shot Reasonerscited by 4,323
  6. 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,772
  7. 2016Stacked Hourglass Networks for Human Pose Estimationcited by 913
  8. 2020On Layer Normalization in the Transformer Architecturecited by 906
  9. 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 763
  10. 2021Going deeper with Image Transformerscited by 759
  11. 2018Universal Transformerscited by 613
  12. 2019Deep Equilibrium Modelscited by 603
5 more
  1. 2016Adaptive Computation Time for Recurrent Neural Networkscited by 479
  2. 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 411
  3. 2014Deeply-Supervised Netscited by 364
  4. 2023Towards Revealing the Mystery behind Chain of Thought: A Theoretical Perspectivecited by 293
  5. 2019CLUTRR: A Diagnostic Benchmark for Inductive Reasoning from Textcited by 155

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026Loop, Think, & Generalize: Implicit Reasoning in Recurrent-Depth Transformers
  2. 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approach
  3. 2026Not All Thinking is Created Equal: Latent Reasoning Discovers a Recurrent Search Algorithm for Depth Generalization
  4. 2025Latent Chain-of-Thought? Decoding the Depth-Recurrent Transformer
  5. 2025A Little Depth Goes a Long Way: The Expressive Power of Log-Depth Transformers
  6. 2025Beyond Memorization: Extending Reasoning Depth with Recurrence, Memory and Test-Time Compute Scaling
  7. 2026Boule or Baguette? A Study on Task Topology, Length Generalization, and the Benefit of Reasoning Traces
  8. 2025Hydra: A Modular Architecture for Efficient Long-Context Reasoning