Stability and Generalization in Looped Transformers
On the map: LLM reasoning → Prompting and chain-of-thought reasoning
Abstract (arXiv)
Looped transformers promise test-time compute scaling by spending more iterations on harder problems, but it remains unclear which architectural choices let them extrapolate to harder problems at test time rather than memorize training-specific solutions. We introduce a fixed-point based framework for analyzing looped architectures along three axes of stability -- reachability, input-dependence, and geometry -- and use it to characterize when fixed-point iteration yields meaningful predictions. Theoretically, we prove that looped networks without recall have countable fixed points and cannot achieve strong input-dependence at any spectral regime, while recall combined with outer normalization reliably produces a regime in which fixed points are simultaneously reachable, locally smooth in the input, and supported by stable backpropagation. Empirically, we train single-layer looped transformers on chess, sudoku, and prefix-sums and find that downstream performance tracks the framework's predictions across tasks and architectural configurations. We additionally introduce internal recall, a novel recall placement variant, and show that it becomes competitive with -- and on sudoku, substantially better than -- standard recall placement once outer normalization is applied.
Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Placed in Prompting and chain-of-thought reasoning, one of 17 directions in LLM reasoning. Builds on 16 of its 27 references that are in the map. First posted on arXiv 16 Apr 2026.
Builds on
16 of 27 references.
Closest ancestors · 2
Same direction, newest first
- 2025Exact Expressive Power of Transformers with Padding25 May
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Models28 Jan
Other references · 14
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2020On Layer Normalization in the Transformer Architecturecited by 906
- 2019Deep Equilibrium Modelscited by 603
- 2016Adaptive Computation Time for Recurrent Neural Networkscited by 479
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 372
- 2020Query-Key Normalization for Transformerscited by 266
- 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 223
- 2025Hierarchical Reasoning Modelcited by 167
- 2021PonderNet: Learning to Pondercited by 156
- 2023Looped Transformers are Better at Learning Learning Algorithmscited by 135
- 2022End-to-end Algorithm Synthesis with Recurrent Networks: Logical Extrapolation Without Overthinkingcited by 33
- 2022Path Independent Equilibrium Models Can Better Exploit Test-Time Computationcited by 33
2 more
Built on by
Citing papers per year: 2026: 9.
9 papers.
- 2026Fixed-Point Reasoners: Stable and Adaptive Deep Looped Transformerscited by 21
- 2026Equilibrium Reasoners: Learning Attractors Enables Scalable Reasoningcited by 12
- 2026Solve the Loop: Attractor Models for Language and Reasoningcited by 11
- 2026Dense Supervision Is Not Enough: The Readout Blind Spot in Looped Language Modelscited by 5
- 2026Think Shallow, Solve Deep: Controlling Recurrent Dynamics for Reliable Test-Time Depthcited by 5
- 2026Loop the Loopies!cited by 4
- 2026Block-Based Double Decoderscited by 0
- 2026Beyond the Training Horizon: Mechanisms and Limits of Length Generalization in Looped Transformerscited by 0
- 2026Improving Test-Time Scaling with Adaptive Looped Transformerscited by 0
- 2026Improving Test-Time Scaling with Adaptive Looped Transformers28 Sep 2026
- 2026Beyond the Training Horizon: Mechanisms and Limits of Length Generalization in Looped Transformers27 Sep 2026
- 2026Think Shallow, Solve Deep: Controlling Recurrent Dynamics for Reliable Test-Time Depth18 Aug 2026
- 2026Loop the Loopies!17 Jul 2026
- 2026Fixed-Point Reasoners: Stable and Adaptive Deep Looped Transformers16 Jun 2026
- 2026Dense Supervision Is Not Enough: The Readout Blind Spot in Looped Language Models12 Jun 2026
- 2026Equilibrium Reasoners: Learning Attractors Enables Scalable Reasoning20 May 2026
- 2026Solve the Loop: Attractor Models for Language and Reasoning12 May 2026
- 2026Block-Based Double Decoders11 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026Adaptive Loops and Memory in Transformers: Think Harder or Know More?
- 2026Training-Free Looped Transformers
- 2026SMELT: Scaling Laws for Compute-Matched MoE Looped Transformers
- 2026Right Direction, Wrong Step: Geometric Analysis of Finite-Step Failure in Looped Transformers
- 2026Looped Transformers as Optimizers
- 2026Quantizing Looped Transformers: Feedback Exposure and Calibration Blindness
- 2026Dynamical phase selection controls compute scaling in looped transformers
- 2026Looped Transformers with Source-Centered State Evolution