acceptodds
Under review as a conference paper at ICLR 2027

Scaling to Tens of Thousands of Test-Time Iterations with Loop-Native Attention Residuals

Abstract

In this paper, we argue that looped Transformers need their own residual connections to prevent performance degradation as the number of iterations grows. We observe that increasing loop iterations can reduce reasoning accuracy: noisy state updates overwrite correct intermediate deductions and even undo completed solutions. Under the standard carry-last rule, each new output replaces the recurrent state, leaving subsequent iterations to recover lost information from an already degraded representation. We introduce InfiLoop, a loop-native residual connection that learns which past computations to retain and how much to accept from each new update. InfiLoop combines content-based weighting with learned temporal decay to maintain a running summary of recurrent states. An exact streaming recurrence keeps its persistent aggregation memory constant as the loop count grows. The resulting adaptive update suppresses unreliable proposals and preserves useful intermediate states. Across extensive reasoning tasks, a 7M-parameter InfiLoop model outperforms existing recursive architectures, reaching 97.9% exact accuracy on Sudoku-Extreme, and 13.6% pass@2 on ARC-AGI-2. It is worth noting that on Sudoku-Extreme, InfiLoop scales test-time looping beyond 20,000 effective steps, effectively translating greater computational depth into higher reasoning capacity.

Then back it, or bet against it.

Related papers

Open the market on this paper to see 7 more related papers.