mlsanityData to 7 Oct 2026

Stability and Generalization in Looped Transformers

arXiv 2604.15259 ↗First posted 16 Apr 2026cs.LG cs.AIcited by 9

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

  1. 2025Exact Expressive Power of Transformers with Padding25 May
  2. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Models28 Jan

Other references · 14

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  2. 2020On Layer Normalization in the Transformer Architecturecited by 906
  3. 2019Deep Equilibrium Modelscited by 603
  4. 2016Adaptive Computation Time for Recurrent Neural Networkscited by 479
  5. 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 372
  6. 2020Query-Key Normalization for Transformerscited by 266
  7. 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 223
  8. 2025Hierarchical Reasoning Modelcited by 167
  9. 2021PonderNet: Learning to Pondercited by 156
  10. 2023Looped Transformers are Better at Learning Learning Algorithmscited by 135
  11. 2022End-to-end Algorithm Synthesis with Recurrent Networks: Logical Extrapolation Without Overthinkingcited by 33
  12. 2022Path Independent Equilibrium Models Can Better Exploit Test-Time Computationcited by 33
2 more
  1. 2025Peri-LN: Revisiting Normalization Layer in the Transformer Architecturecited by 32
  2. 2024On the Inductive Bias of Stacking Towards Improving Reasoningcited by 18

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 9.

9 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Adaptive Loops and Memory in Transformers: Think Harder or Know More?
  2. 2026Training-Free Looped Transformers
  3. 2026SMELT: Scaling Laws for Compute-Matched MoE Looped Transformers
  4. 2026Right Direction, Wrong Step: Geometric Analysis of Finite-Step Failure in Looped Transformers
  5. 2026Looped Transformers as Optimizers
  6. 2026Quantizing Looped Transformers: Feedback Exposure and Calibration Blindness
  7. 2026Dynamical phase selection controls compute scaling in looped transformers
  8. 2026Looped Transformers with Source-Centered State Evolution