acceptodds
Under review as a conference paper at ICLR 2027

Closing the Loop: Practical Training Recipes for Looped Language Models

Abstract

Looped language models increase effective depth by repeatedly applying a shared block of layers, but existing large-scale recipes require multi-stage training over trillions of tokens, and it remains difficult to separate the benefits of recurrence from differences in data and training. In this work, we establish practical recipes for looped language models, with three main results. (1) We develop a compute-efficient from-scratch training pipeline that reduces the training budget from 7.7T tokens in Ouro to 310B tokens while retaining strong reasoning performance. A two-stage recipe—pretraining followed by high-quality midtraining—together with learning-rate warmup and stronger exit-gate regularization enables stable recurrent training without the multi-stage schedule used in prior work. (2) We establish an end-to-end recipe spanning pretraining and reasoning-oriented midtraining. Under controlled comparisons, our 1.4B LoopLM outperforms a parameter-matched dense model trained on exactly the same data and token budget on all 12 evaluated benchmarks, including +14 points on GSM8K, +10 on MATH, and +22 on DROP. At matched inference compute, it approaches a 3.9B dense model on mathematical reasoning and reading comprehension while using only 36% as many parameters. (3) We introduce a minimal recipe for converting pretrained dense models into looped ones: a single learned input-mixing scalar and a smoothed exit loss are sufficient, with no step-specific parameters. Applied to Qwen3-1.7B-Base, the resulting Looped Qwen improves over an identically continued dense baseline on every evaluated benchmark across two independent data regimes, with statistically clear gains on GSM8K, MATH, and MMLU-Pro on the curated mixture. Together, these results make looped language models substantially cheaper to train from scratch and practical to introduce into existing pretrained checkpoints, while isolating the gains due to recurrence itself.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.