Looped Flows: A Training Method to Stabilize Recurrent Compute in Looped Models

burkov · x · 2026-09-13

Looped models reuse the same parameters to iteratively update a shared hidden state, gaining extra inference-time compute without growing model size. But training them is hard: full backprop through many recurrent steps is costly and unstable, so gradients are usually truncated after one or a few updates, leaving early steps unsupervised and recurrences diverging or settling into wrong solutions.

The proposed looped flows approach trains a stateful denoiser on a sequence of local denoising objectives at progressively lower noise levels, with shared noise across steps to encourage temporal consistency of the recurrent state; inference then integrates the resulting probability flow.

Related event: Looped Flows: Denoising-Trained Recurrent Reasoning Tops Loop-Model ARC-AGI(6 posts)→

Original post →

More from Research

Research channel →