From Cross-Validation to SURE: Asymptotic Risk of Tuned Regularized Estimators
Authors: Karun Adusumilli, Maximilian Kasy, Ashia Wilson
Organizations: Department of Economics, University of Pennsylvania. · Department of Economics, University of Oxford. · Department of Electrical Engineering and Computer Science, MIT.
We derive the asymptotic risk function of regularized empirical risk minimization (ERM) estimators tuned by n-fold cross-validation (CV). The out-of-sample prediction loss of such estimators converges in distribution to the squared-error loss (risk function) of shrinkage estimators in the normal means model, tuned by Stein's unbiased risk estimate (SURE). This risk function provides a more fine-grained picture of predictive performance than uniform bounds on worst-case regret, which are common in learning theory: it quantifies how risk varies with the true parameter. As key intermediate steps, we show that (i) n-fold CV converges uniformly to SURE, and (ii) while SURE typically has multiple local minima, its global minimum is generically well separated. Well-separation ensures that uniform convergence of CV to SURE translates into convergence of the tuning parameter chosen by CV to that chosen by SURE.
Figures & tables
Figure 1: Risk function for JS-shrinkage, dimension 10
Figure 2: Examples of multi-modality of SURE
Result
Assumptions invoked
Builds on
Lemma 1 (Lipschitz gλ )
convexity of π
—
Lemma 2 (Influence-function approx.)
1, 2, 4.1, 4.2, 5.2
Lemma 1
Lemma 3 (Limiting squared-error loss)
1, 2, 5.3
—
Corollary 2 (Asymptotic distribution)
1, 4.1
Lemmas 1 , 2
Lemma 4 (CV → SURE)
1, 2, 3, 5.1–5.4
Lemmas 1 , 2 , Cor. 2
Lemma 5 (Tuning-parameter convergence)
1, 3
Lemmas 2 , 4 , Cor. 2
Table 1: Assumptions invoked by each result.
Loss
Penalty
p
θ0
A
D1
Linear
Ridge
10
(1,…,1)
I
D2
Linear
Lasso
10
(3,−3,2,0,…,0)
I
D3
Logit
Ridge
10
(1,…,1)
I
D4
Logit
Lasso
10
(3,−3,2,0,…,0)
I
D5
Linear
Ridge
10
1.5⋅(1,…,1)
diag(40(j−1)/9)
Table 2: Simulation designs.
Design
ΔIF
ΔCV
Δtune
Rn
R
ΔR
(Lem. 2)
(Lem. 4)
(Lem. 5)
(Thm. 1)
(limit)
∣Rn−R∣
D1 Linear / Ridge
0.028
0.208
0.052
3.110
3.084
0.026
D2 Linear / Lasso (sparse)
0.028
0.226
0.111
3.885
3.789
0.096
D3 Logit / Ridge
0.032
0.309
0.055
3.062
3.075
0.013
D4 Logit / Lasso (sparse)
0.038
0.297
0.127
3.731
3.776
0.046
D5 Linear / Ridge (anisotropic)
0.028
0.152
0.036
4.827
4.733
0.094
Table 3: Approximation diagnostics at n=400 .
Figure 3: Convergence of the approximations as n grows (designs identified in the legend panel, lower right). Top left: the influence-function error ΔIF (Lemma 2 ), ∝n−1 . Top right: the cross-validation gap ΔCV (Lemma 4 ), ∝n−1/2 . Bottom left: the finite-sample risk Rn converging to the limit R (dotted lines; Theorem 1 ).
Figure 4: Convergence of the risk function (linear–Ridge design). Along the slice θ0=t⋅v (with v a fixed unit vector, so ∥θ0∥=t ), the finite-sample risk Rn(θ0) of the CV-tuned estimator approaches the SURE-tuned limit R(θ0) ; by n=200 the curves are nearly indistinguishable from the limit.