12
0

Improved Scaling Laws in Linear Regression via Data Reuse

Abstract

Neural scaling laws suggest that the test error of large language models trained online decreases polynomially as the model size and data size increase. However, such scaling can be unsustainable when running out of new data. In this work, we show that data reuse can improve existing scaling laws in linear regression. Specifically, we derive sharp test error bounds on MM-dimensional linear models trained by multi-pass stochastic gradient descent (multi-pass SGD) on NN data with sketched features. Assuming that the data covariance has a power-law spectrum of degree aa, and that the true parameter follows a prior with an aligned power-law spectrum of degree bab-a (with a>b>1a > b > 1), we show that multi-pass SGD achieves a test error of Θ(M1b+L(1b)/a)\Theta(M^{1-b} + L^{(1-b)/a}), where LNa/bL \lesssim N^{a/b} is the number of iterations. In the same setting, one-pass SGD only attains a test error of Θ(M1b+N(1b)/a)\Theta(M^{1-b} + N^{(1-b)/a}) (see e.g., Lin et al., 2024). This suggests an improved scaling law via data reuse (i.e., choosing L>NL>N) in data-constrained regimes. Numerical simulations are also provided to verify our theoretical findings.

View on arXiv
@article{lin2025_2506.08415,
  title={ Improved Scaling Laws in Linear Regression via Data Reuse },
  author={ Licong Lin and Jingfeng Wu and Peter L. Bartlett },
  journal={arXiv preprint arXiv:2506.08415},
  year={ 2025 }
}
Comments on this paper