Learning to Compute Is Not Learning to Retain or Reuse in Looped Transformers
Not on the map. Why?
Abstract (arXiv)
Looped transformers compute by applying one block repeatedly, so they can run for more loops than they were trained with. Recurrent reasoners are known to overthink: iterating past the trained depth can turn a correct answer into a wrong one. Such a late error can reflect failed computation, loss of retention, or a state that later computation cannot reuse. We separate these cases in in-context Mealy-machine execution, where a simulator supplies intermediate targets and exact counterfactuals for hidden-state interventions. Models trained to answer at loop 32 are fully correct there but keep only 22% of these answers at loop 64. Retention depends on what training asks for: supervising one randomly chosen position per example, instead of always the last, keeps 99.8%, and randomizing the number of training loops also helps; truncating gradients, as in progressive training against overthinking, does not. The computation itself is exact: position k is computed at loop k, and swapping its loop-k hidden vector for the one from another input redirects every later position to that input's exact continuation (237/237 pairs in each of ten models). Retention, however, is not reuse. Even in models that keep nearly all answers, holding each position k fixed from loop k drops accuracy to 4--6%, whereas holding it from loop k+4 is harmless. A released recurrent maze network trained to avoid overthinking behaves the same way. Learning to compute is thus not learning to retain, and avoiding overthinking does not make a state reusable when its answer first appears.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 15 of its 16 references that are in the map. First posted on arXiv 14 Oct 2026.
Builds on
15 of 16 references.
References · 15
- 2018Universal Transformerscited by 647
- 2021Causal Abstractions of Neural Networkscited by 422
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 415
- 2022Transformers Learn Shortcuts to Automatacited by 268
- 2023Looped Transformers as Programmable Computerscited by 237
- 2023Finding Alignments Between Interpretable Causal Variables and Distributed Neural Representationscited by 209
- 2021Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networkscited by 117
- 2024Looped Transformers for Length Generalizationcited by 99
- 2022Path Independent Equilibrium Models Can Better Exploit Test-Time Computationcited by 38
- 2025(How) Do Language Models Track State?cited by 34
- 2025Finite State Automata Inside Transformers with Chain-of-Thought: A Mechanistic Study on State Trackingcited by 18
- 2026Adaptive Depth in Looped Transformers: Diagnosing Learned Halting Gates and Trajectory Readoutscited by 12
3 more
- 2026When Does Recurrence Become an Algorithm? Convergence Selection in Weight-Tied Looped Transformerscited by 6
- 2026Looped Transformers under the Jacobian Lens: Does the Global Workspace Survive Recurrence?cited by 6
- 2026Stabilizing Extrapolation in Looped Transformers via Learned Stochastic Stoppingcited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Adaptive Loops and Memory in Transformers: Think Harder or Know More?
- 2026Step-resolved data attribution for looped transformers
- 2026Stability and Generalization in Looped Transformers
- 2026Shared Weights, Selected Computations: How Looped Transformers Route What Each Loop Does
- 2026Beyond the Training Horizon: Mechanisms and Limits of Length Generalization in Looped Transformers
- 2026Training-Free Looped Transformers
- 2026Looped Actor: Depth-Recurrent Reasoning Models for Reinforcement Learning
- 2026Chain-of-Thought and Compressed Looped Transformers: A Memory-Budget Separation