When do data mixtures improve scaling laws? Insights from high-dimensional regression
Not on the map. Why?
Abstract (arXiv)
Modern machine learning systems are trained on mixtures of data from different domains, and choosing the right mixture can substantially improve downstream performance. Despite an extensive literature on data mixing and reweighting, existing work is largely empirical and it remains unclear when auxiliary data genuinely improves scaling laws rather than merely providing more samples. To gain insight into this question, we study a high-dimensional mixed-data regression model with a shared regression function, heterogeneous covariances and noise levels, and dataset sizes that may grow at different rates. We establish the minimax risk under an ellipsoidal parameter constraint for the general covariance structure and derive deterministic equivalents for the test error of ridge regression under commutative covariances. We then specialize to a target domain and an auxiliary domain with aligned power-law covariance spectra, where the theory yields explicit scaling laws in terms of spectral decay, target regularity, and the relative growth of the two datasets. These laws identify regimes in which combining data mixtures provably yields a faster scaling rate than using either dataset alone. In particular, improving the scaling law requires a specific interplay between spectra and relative sample sizes of the domains. Our numerical experiments on language models exhibit the same qualitative phenomenon: appropriate data mixtures yield a faster decrease in target-domain test loss than training on either domain alone.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 7 of its 70 references that are in the map. First posted on arXiv 29 Sep 2026.
Builds on
7 of 70 references.
References · 7
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2022A Solvable Model of Neural Scaling Lawscited by 116
- 2024A non-asymptotic theory of Kernel Ridge Regression: deterministic equivalents, test error, and GCV estimatorcited by 28
- 2024BiMix: A Bivariate Data Mixing Law for Language Model Pretrainingcited by 25
- 2024Generalization error of min-norm interpolators in transfer learningcited by 16
- 2026Scaling Laws for Mixture Pretraining Under Data Constraintscited by 6
Built on by
Not cited yet.