cs.LGAug 23, 2026

Linear RNN Scaling Laws: When Longer Sequences Beat More Sequences

Authors: Ziyan Chen, Zhongzhu Zhou, Linhe Zheng, Peilin Liu, Ding-Xuan Zhou

Organizations: The University of Sydney · Together AI · The Pennsylvania State University

Abstract

Empirical scaling laws for autoregressive language models relate prediction loss to model size, data size, and optimization compute, but their theoretical origin is still poorly understood in sequential pretraining settings. We study this question in a tractable teacher--student model where a stable latent linear RNN generates trajectories and a sketched linear recurrent student is trained by safeguarded full-batch WSD gradient descent on next-token prediction. The sketch dimension MM plays the role of model size, while NN independent trajectories of length PP provide the training tokens. We allow the innovation and initialization covariances to have different power-law exponents αα and θθ. The induced design spectrum produces explicit approximation, optimization, and statistical scaling laws separated by spectral crossovers. When θ≥αθ\geα, the original one-scale rates M1−βαM^{1-β_α}, R(1−βα)/αR^{(1-β_α)/α}, and (NP)−1min⁡{M,R1/α}(NP)^{-1}\min\{M,R^{1/α}\} are recovered. When α−2r≤θ<αα-2r\leθ<α, the heavier initialization tail changes the rates beyond PP-dependent model and optimization crossovers. The proof uses a covariance event only internally and a globally safeguarded step size on its complement. The variance retains the factor (NP)−1(NP)^{-1}, while sequence length also suppresses the initialization transient, so NN and PP cease to be fully interchangeable in the two-scale regime.

Figures & tables

Appendix figures & tables1 asset

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

Jun 29, 2026

Smooth Scaling Laws Hide Stepwise Token Learning

Language model loss follows remarkably regular scaling laws over model and data size, yet it remains unclear why the aggregate loss should exhibit a power-law form. Existing explanations often attribute this regularity to a heavy-tailed spectrum of pattern difficulty in natural language, but this view has not been directly validated at token-level granularity in large-scale real-data training. We present a token-level framework that decomposes scaling laws into localized learning events of individual contextualized tokens. By fitting token loss trajectories with sigmoids, we show that token learning is concentrated in localized transitions, giving rise to a learning-time spectrum that dominates the scaling-law shape. Across more than one hundred pre-training runs on large and diverse real-language corpora with modern LLM architectures, scaling up to 6B parameters and 300B training tokens, the measured learning-time spectrum quantitatively reconstructs the validation loss derivative along the training-step TT, data-scale DD, and model-scale MM axes. We further show that the same signal is actionable: by reshaping the training distribution according to when tokens become learnable, we alter the optimization trajectory and achieve 11% faster validation-loss reduction. These results provide direct empirical evidence that scaling laws are governed primarily by the distribution of token-level learning times, and that this distribution can be used not only to explain scaling behavior but also to improve training performance.
Oct 13, 2026

Exact Scaling Laws for a Random-Features Bigram Model

Understanding neural scaling laws is increasingly important for theory and practice, yet first-principles explanations remain limited even in simple settings. In this work, we study capacity-limited scaling in a bottlenecked bigram model with power-law (Zipf) token statistics, viewed as an analytically tractable toy model of a large language model. Fixed random embeddings serve as a proxy for representational capacity and induce an explicit bottleneck of width, allowing us to derive scaling laws for prediction regret as a function of model width, vocabulary size, and the Zipf exponent. Our results complement recent theoretical accounts based on optimization dynamics by isolating scaling due purely to representational constraints. In the appropriate regime, the theory recovers the functional form of classical neural scaling laws and suggests that heavy-tailed language statistics may underlie both optimization- and capacity-driven scaling. Experiments natural language datasets (WikiText2, OpenWebText) closely match the predicted curves.
May 28, 2026

On the Optimizer Dependence of Neural Scaling Laws

The scaling exponent αα in neural scaling laws L(N)∝N−αL(N) \propto N^{-α} is commonly treated as a fixed constant set by architecture and data. We present evidence that αα depends systematically on the optimizer. In controlled random-feature regression experiments -- the canonical theoretical framework for neural scaling -- we measure αα across five optimizer variants and six spectral conditions. Preconditioned optimizers consistently yield steeper scaling (larger αα), with the αα-shift increasing across most of the tested spectral range, peaking near s=1.5s = 1.5, and remaining large at s=2.0s = 2.0. At s≈1.0s \approx 1.0 (characteristic of natural language), the full natural gradient achieves α≈0.31α\approx 0.31 versus α≈0.12α\approx 0.12 for gradient descent -- a 2.6×2.6\times larger fitted exponent that, within the random-feature model, compounds with each model-size doubling. Whether and how this exponent shift transfers to large-scale LLM training -- where recent evidence suggests the advantage may attenuate with scale -- remains an important open question. Our results imply that scaling-law forecasts should account for optimizer choice, and we provide a spectral diagnostic predicting when advanced optimizers will pay off.