English

Training-Free Looped Transformers

Machine Learning 2026-05-25 v1 Numerical Analysis Numerical Analysis Machine Learning

Abstract

We introduce training-free looped transformers, in which a lightweight inference-time wrapper loops a contiguous mid-stack block of layers of a frozen checkpoint without additional fine-tuning, continued training, or architectural changes. Unlike prior looped transformer methods that train with the looped structure end-to-end, we retrofit recurrence onto pretrained models at test time. We show that naive block reapplication usually degrades performance, highlighting the importance of the loop application strategy. Motivated by viewing a pre-norm transformer block as a forward Euler step on an ODE, we instead treat looping as a refinement of the same approximation, replacing one large update with smaller damped sub-steps. Across seven dense, sparse MoE, and MLA+MoE model families, our method improves Qwen3-4B-Instruct by +2.64 pp on MMLU-Pro, Qwen3-30B-A3B-Instruct by +1.14 pp on CommonsenseQA, and Moonlight-16B-A3B-Instruct by +1.20 pp on OpenBookQA.

Cite

@article{arxiv.2605.23872,
  title  = {Training-Free Looped Transformers},
  author = {Lizhang Chen and Jonathan Li and Chen Liang and Ni Lao and Qiang Liu},
  journal= {arXiv preprint arXiv:2605.23872},
  year   = {2026}
}