Risk and cross validation in ridge regression with correlated samples
Not on the map. Why?
Abstract (arXiv)
Recent years have seen substantial advances in our understanding of high-dimensional ridge regression, but existing theories assume that training examples are independent. By leveraging techniques from random matrix theory and free probability, we provide sharp asymptotics for the in- and out-of-sample risks of ridge regression when the data points have arbitrary correlations. We demonstrate that in this setting, the generalized cross validation estimator (GCV) fails to correctly predict the out-of-sample risk. However, in the case where the noise residuals have the same correlations as the data points, one can modify the GCV to yield an efficiently-computable unbiased estimator that concentrates in the high-dimensional limit, which we dub CorrGCV. We further extend our asymptotic analysis to the case where the test point has nontrivial correlations with the training set, a setting often encountered in time series forecasting. Assuming knowledge of the correlation structure of the time series, this again yields an extension of the GCV estimator, and sharply characterizes the degree to which such test points yield an overly optimistic prediction of long-time risk. We validate the predictions of our theory across a variety of high dimensional data.
Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 34 of its 77 references that are in the map. First posted on arXiv 8 Aug 2024.
Builds on
34 of 77 references.
References · 34
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2018On Lazy Training in Differentiable Programmingcited by 671
- 2019Surprises in High-Dimensional Ridgeless Least Squares Interpolationcited by 508
- 2021Explaining Neural Scaling Lawscited by 353
- 2017High-dimensional dynamics of generalization error in neural networkscited by 238
- 2020Spectrum Dependent Learning Curves in Kernel Regression and Wide Neural Networkscited by 200
- 2020Spectral Bias and Task-Model Alignment Explain Generalization in Kernel Regression and Infinitely Wide Neural Networkscited by 166
- 2015High-Dimensional Asymptotics of Prediction: Ridge Regression and Classificationcited by 146
- 2020Generalisation error in learning with random features and the hidden manifold modelcited by 126
- 2021Learning curves of generic features maps for realistic datasets with a teacher-student modelcited by 126
- 2024A Dynamical Model of Neural Scaling Lawscited by 105
22 more
- 2022A Solvable Model of Neural Scaling Lawscited by 101
- 2021Generalization Error Rates in Kernel Regression: The Crossover from the Noiseless to Noisy Regimecited by 95
- 2020Understanding Double Descent Requires a Fine-Grained Bias-Variance Decompositioncited by 75
- 2022Universality of empirical risk minimizationcited by 73
- 2021Cross-validation: what does it estimate and how well does it do it?cited by 65
- 2020Kernel Alignment Risk Estimator: Risk Prediction from Training Datacited by 54
- 2024Asymptotic theory of in-context learning by linear attentioncited by 54
- 2023High-dimensional analysis of double descent for linear regression with random projectionscited by 47
- 2024Scaling and renormalization in high-dimensional regressioncited by 45
- 2020What causes the test error? Going beyond bias-variance via ANOVAcited by 28
- 2021Covariate Shift in High-Dimensional Random Feature Regressioncited by 24
- 2024A non-asymptotic theory of Kernel Ridge Regression: deterministic equivalents, test error, and GCV estimatorcited by 22
- 2021Out-of-Distribution Generalization in Kernel Regressioncited by 21
- 2024Optimal Ridge Regularization for Out-of-Distribution Predictioncited by 18
- 2016Statistical Mechanics of High-Dimensional Inferencecited by 13
- 2023Learning curves for deep structured Gaussian feature modelscited by 13
- 2019Cross-Validation for Correlated Datacited by 11
- 2023Asymptotically free sketched ridge ensembles: Risks, cross-validation, and tuningcited by 11
- 2016Distributed Generalized Cross-Validation for Divide-and-Conquer Kernel Ridge Regression and its Asymptotic Optimalitycited by 9
- 2021Statistical Neuroscience in the Single Trial Limitcited by 4
- 2024High-dimensional analysis of ridge regression for non-identically distributed data with a variance profilecited by 4
- 2024ROTI-GCV: Generalized Cross-Validation for right-ROTationally Invariant Datacited by 4
Built on by
Citing papers per year: 2024: 1; 2025: 5; 2026: 2.
8 papers.
- 2025On the Mechanisms of Weak-to-Strong Generalization: A Theoretical Perspectivecited by 7
- 2025Two-Point Deterministic Equivalence for Stochastic Gradient Dynamics in Linear Modelscited by 6
- 2025Pretrain-Test Task Alignment Governs Generalization in In-Context Learningcited by 6
- 2024Asymptotics of Linear Regression with Linearly Dependent Datacited by 4
- 2025Generalisation and benign over-fitting for linear regression onto random functional covariatescited by 1
- 2025A Ridge Too Far: Correcting Over-Shrinkage via Negative Regularizationcited by 0
- 2026Sequential Correlations Change In-Context Learning: Effective Context Length and Architectural Mismatchcited by 0
- 2026A Simple Approximation to the Distribution of the Ridge Regression Estimatorcited by 0
- 2026A Simple Approximation to the Distribution of the Ridge Regression Estimator3 Aug 2026
- 2026Sequential Correlations Change In-Context Learning: Effective Context Length and Architectural Mismatch4 Jul 2026
- 2025Pretrain-Test Task Alignment Governs Generalization in In-Context Learning30 Sep 2025
- 2025A Ridge Too Far: Correcting Over-Shrinkage via Negative Regularization24 Aug 2025
- 2025Generalisation and benign over-fitting for linear regression onto random functional covariates19 Aug 2025
- 2025On the Mechanisms of Weak-to-Strong Generalization: A Theoretical Perspective23 May 2025
- 2025Two-Point Deterministic Equivalence for Stochastic Gradient Dynamics in Linear Models7 Feb 2025
- 2024Asymptotics of Linear Regression with Linearly Dependent Data4 Dec 2024