mlsanityData to 7 Oct 2026

Inner Loop Inference for Pretrained Transformers: Unlocking Latent Capabilities Without Training

arXiv 2602.14759 ↗First posted 16 Feb 2026cs.LG cs.AIcited by 2

Not on the map. Why?

Abstract (arXiv)

Deep Learning architectures, and in particular Transformers, are conventionally viewed as a composition of layers. These layers are actually often obtained as the sum of two contributions: a residual path that copies the input and the output of a Transformer block. As a consequence, the inner representations (i.e. the input of these blocks) can be interpreted as iterative refinement of a propagated latent representation. Under this lens, many works suggest that the inner space is shared across layers, meaning that tokens can be decoded at early stages. Mechanistic interpretability even goes further by conjecturing that some layers act as refinement layers. Following this path, we propose inference-time inner looping, which prolongs refinement in pretrained off-the-shelf language models by repeatedly re-applying a selected block range. Across multiple benchmarks, inner looping yields modest but consistent accuracy improvements. Analyses of the resulting latent trajectories suggest more stable state evolution and continued semantic refinement. Overall, our results suggest that additional refinement can be obtained through simple test-time looping, extending computation in frozen pretrained models.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 16 of its 19 references that are in the map. First posted on arXiv 16 Feb 2026.

Builds on

16 of 19 references.

References · 16

  1. 2017Attention Is All You Needcited by 59,932
  2. 2024The Llama 3 Herd of Modelscited by 14,192
  3. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  4. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  5. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  6. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  7. 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,928
  8. 2022In-context Learning and Induction Headscited by 930
  9. 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 763
  10. 2018Universal Transformerscited by 613
  11. 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 579
  12. 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 372
4 more
  1. 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 223
  2. 2024The Remarkable Robustness of LLMs: Stages of Inference?cited by 138
  3. 2021Stabilizing Equilibrium Models by Jacobian Regularizationcited by 71
  4. 2024Mechanistic?cited by 25

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.