From Growing to Looping: A Unified View of Iterative Computation in LLMs
Abstract
Lay Summary
Large language models excel at many tasks, but getting them to reason through multi-step problems, like math or logic puzzles, remains hard. Two training techniques have each been shown to help: looping (recycling the same set of network layers multiple times in sequence) and depth growing (starting with a small model and progressively duplicating its middle layers until full size). Despite their different appearances, it was unclear whether they work for the same reason. We show they do: both push the model to refine its understanding step by step, like a student who rechecks their work, rather than solving everything in one pass. Measurements of how much each layer contributes to the final answer confirm that both approaches shift critical computation toward later layers in the same periodic, structured way. Building on this, we show the two techniques can be combined: looping the middle section of a depth-grown model at test time, without any extra training, can double accuracy on certain reasoning tasks. This unification suggests a simple recipe "grow first, loop later" for building more capable reasoning models without proportionally increasing training or inference cost.