首页 > AI前沿 > How Much Is One Recurrence Worth? Iso-Depth Scaling Laws for Looped Language Models

How Much Is One Recurrence Worth? Iso-Depth Scaling Laws for Looped Language Models

arXiv自然语言 2026-04-23 05:51 5 阅读 查看原文

Introduction

We measure how much one recurrence is worth to a looped transformer in equivalent unique parameters. From an iso-depth pretraining sweep across recurrence counts $r \in \{1, 2, 4, 8\}$ spanning ${\sim}50\times$ in training compute, we fit a joint scaling law $L = E + A\,(N_\text{once} + r^{\varphi} N_\text{rec})^{-α} + B\,D^{-β}$ and measure a recurrence-equivalence exponent $\varphi = 0.46$. Intuitively, $\varphi$ tells us whether looping a block $r$ times is equivalent in validation loss to $r$ unique blocks of a non-looped model (full equivalence, $\varphi{=}1$) or to a single block run repeatedly with no capacity gain ($\varphi{=}0$).

Our $\varphi = 0.46$ sits in between, so replacing unique blocks with shared recurrences increases validation loss at matched training compute. For example, at $r{=}4$ a 410M looped model performs on par with a 580M non-looped model, but incurs the training cost of a 1B non-looped one.

Case Studies

We demonstrate the utility of $\varphi$ as a diagnostic tool on two case studies:

  • Truncated backpropagation lowers $\varphi$ to 0.38, indicating that the loop mechanism is poorly trained under truncation, even though validation loss decreases.
  • Hyperconnections raise $\varphi$ to 0.65, a genuine capacity gain.

Diagnostic Utility

Our method separates true loop improvements from training-side gains, a distinction raw validation loss cannot make.

Code is available at https://github.com/kschwethelm/looped-lm-scaling.