mlsanityData to 7 Oct 2026

Adaptive Depth in Looped Transformers: Diagnosing Learned Halting Gates and Trajectory Readouts

arXiv 2607.20519 ↗First posted 8 Jul 2026cs.LGcited by 7

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

  1. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  2. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  3. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  4. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
  5. 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,772
  6. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,114
  7. 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,793
  8. 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 763
  9. 2019Deep Equilibrium Modelscited by 603
  10. 2016Adaptive Computation Time for Recurrent Neural Networkscited by 479
  11. 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 372
  12. 2020DeeBERT: Dynamic Early Exiting for Accelerating BERT Inferencecited by 322
19 more
  1. 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decodingcited by 302
  2. 2024Quiet-STaR: Language Models Can Teach Themselves to Think Before Speakingcited by 292
  3. 2020BERT Loses Patience: Fast and Robust Inference with Early Exitcited by 267
  4. 2020DynaBERT: Dynamic BERT with Adaptive Width and Depthcited by 246
  5. 2024Mixture-of-Depths: Dynamically allocating compute in transformer-based language modelscited by 242
  6. 2019Depth-Adaptive Transformercited by 227
  7. 2023Looped Transformers as Programmable Computerscited by 224
  8. 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 223
  9. 2022Exploring Length Generalization in Large Language Modelscited by 204
  10. 2021PonderNet: Learning to Pondercited by 156
  11. 2025Scaling Latent Reasoning via Looped Language Modelscited by 145
  12. 2023Looped Transformers are Better at Learning Learning Algorithmscited by 135
  13. 2025Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computationcited by 120
  14. 2023Fast and Robust Early-Exiting Framework for Autoregressive Language Models with Synchronized Parallel Decodingcited by 80
  15. 2026LoopFormer: Elastic-Depth Looped Transformers for Latent Reasoning via Shortcut Modulationcited by 48
  16. 2026Loop, Think, & Generalize: Implicit Reasoning in Recurrent-Depth Transformerscited by 34
  17. 2025Physics of Language Models: Part 4.1, Architecture Design and the Magic of Canon Layerscited by 32
  18. 2026AdaPonderLM: Gated Pondering Language Models with Token-Wise Adaptive Depthcited by 9
  19. 2026The Stepwise Informativeness Assumption: Why are Entropy Dynamics and Reasoning Correlated in LLMs?cited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 7.

7 papers.