Adaptive Depth in Looped Transformers: Diagnosing Learned Halting Gates and Trajectory Readouts
Not on the map. Why?
Abstract (arXiv)
Looped Transformers increase test-time computation by repeatedly applying a shared recurrent block. Learned halting objectives in looped Transformers typically use a single exit distribution both as the inference-time stopping rule and as the training-time weighting of per-depth losses. This entangles exit selection with trajectory formation: the gate not only chooses which recurrent state to use, but also determines how strongly each intermediate state is supervised. Consequently, poor adaptive-compute performance can arise from the readout, the induced trajectory, or their interaction. We study adaptive depth in looped Transformers through this trajectory--readout lens, across controlled synthetic tasks (modular arithmetic and binary parity) and large-scale Ouro-1.4B and 2.6B checkpoints. We find that fixed-prior depth supervision, which shapes the trajectory without an input-dependent halting policy, produces difficulty-aware trajectories whose intermediate states expose useful stopping signals, and that simple post-hoc confidence readouts often match or outperform learned linear and MLP gates. Fitting gates on frozen trajectories localizes the failure: it appears to stem mainly from the trajectory induced by joint gate training rather than from limited gate expressivity. The same pattern is present in Ouro evaluations, where pretrained ponder gates are competitive but not uniformly Pareto-optimal, and measured latency confirms that the resulting reductions in average exit depth translate into practical inference-time savings. Our systematic diagnostic evaluation reframes adaptive depth in looped Transformers as a joint problem of trajectory formation and exit readout, rather than gate learning alone, highlighting a distinction that prior learned-halting work has often left implicit.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Builds on 31 of its 32 references that are in the map. First posted on arXiv 8 Jul 2026.
Builds on
31 of 32 references.
References · 31
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,772
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,114
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,793
- 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 763
- 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
- 2020DeeBERT: Dynamic Early Exiting for Accelerating BERT Inferencecited by 322
19 more
- 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decodingcited by 302
- 2024Quiet-STaR: Language Models Can Teach Themselves to Think Before Speakingcited by 292
- 2020BERT Loses Patience: Fast and Robust Inference with Early Exitcited by 267
- 2020DynaBERT: Dynamic BERT with Adaptive Width and Depthcited by 246
- 2024Mixture-of-Depths: Dynamically allocating compute in transformer-based language modelscited by 242
- 2019Depth-Adaptive Transformercited by 227
- 2023Looped Transformers as Programmable Computerscited by 224
- 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 223
- 2022Exploring Length Generalization in Large Language Modelscited by 204
- 2021PonderNet: Learning to Pondercited by 156
- 2025Scaling Latent Reasoning via Looped Language Modelscited by 145
- 2023Looped Transformers are Better at Learning Learning Algorithmscited by 135
- 2025Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computationcited by 120
- 2023Fast and Robust Early-Exiting Framework for Autoregressive Language Models with Synchronized Parallel Decodingcited by 80
- 2026LoopFormer: Elastic-Depth Looped Transformers for Latent Reasoning via Shortcut Modulationcited by 48
- 2026Loop, Think, & Generalize: Implicit Reasoning in Recurrent-Depth Transformerscited by 34
- 2025Physics of Language Models: Part 4.1, Architecture Design and the Magic of Canon Layerscited by 32
- 2026AdaPonderLM: Gated Pondering Language Models with Token-Wise Adaptive Depthcited by 9
- 2026The Stepwise Informativeness Assumption: Why are Entropy Dynamics and Reasoning Correlated in LLMs?cited by 3
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026The Ignition Is Real, and It Lives at the Readout: Latent composition, difficulty-clocked ignition, and the interface-constituted commit in a recurrent-depth reasonercited by 2
- 2026Looped Language Models Improve Compositional Tool Callingcited by 1
- 2026FlexLoop: Depth-Elastic Looped Policies for Adaptive Test-Time Computation in Deep RLcited by 0
- 2026Improving Test-Time Scaling with Adaptive Looped Transformerscited by 0
- 2026What Makes Recurrence Effective in Looped Language Models?cited by 0
- 2026LatentHarness: Learning Latent Actions for Memory and Reasoning via Counterfactual Policy Distillationcited by 0
- 2026Shared Weights, Selected Computations: How Looped Transformers Route What Each Loop Doescited by 0
- 2026Shared Weights, Selected Computations: How Looped Transformers Route What Each Loop Does30 Sep 2026
- 2026LatentHarness: Learning Latent Actions for Memory and Reasoning via Counterfactual Policy Distillation30 Sep 2026
- 2026What Makes Recurrence Effective in Looped Language Models?29 Sep 2026
- 2026Improving Test-Time Scaling with Adaptive Looped Transformers28 Sep 2026
- 2026FlexLoop: Depth-Elastic Looped Policies for Adaptive Test-Time Computation in Deep RL28 Sep 2026
- 2026Looped Language Models Improve Compositional Tool Calling17 Aug 2026
- 2026The Ignition Is Real, and It Lives at the Readout: Latent composition, difficulty-clocked ignition, and the interface-constituted commit in a recurrent-depth reasoner4 Aug 2026