Sam Babichenko

Reading notes / Benign overfitting

Uniform convergence of interpolators

Koehler, Zhou, Sutherland and Srebro, 2021; Zhou, Koehler, Sutherland and Srebro, 2021

Context. Bartlett, Long, Lugosi and Tsigler (2020) proved benign overfitting for the minimum ℓ2\ell_2-norm interpolator through its explicit formula, and uniform convergence seemed unable to (Nagarajan and Kolter, and several impossibility results). Two companion papers on Gaussian linear regression show that it can. Koehler, Zhou, Sutherland and Srebro (KZSS) take the supremum over interpolators only and prove the Zhou et al. conjecture sup⁡∥w∥2≤B, L^(w)=0L(w)≤B2 E∥x∥22/n+o(1)\sup_{\|w\|_2\le B,\ \hat L(w)=0}L(w)\le B^2\,\mathbb E\|x\|_2^2/n+o(1) with leading constant exactly one, then bound the minimal norm. Zhou, Koehler, Sutherland and Srebro (ZKSS) extend the bound to every ww as an optimistic rate, root risk at most root training loss plus width over n\sqrt n, again with constant one (Srebro et al.'s general rate carries a factor of order 2⋅105log⁡3n2\cdot10^5\log^3n). The constant one matters because, for an interpolating class, the complexity term squared is about σ2\sigma^2, so any larger factor leaves a constant times σ2\sigma^2. Among interpolators, risk is at most (norm times the expected dual norm of the low-variance part of xx)2/n^2/n, and fitting the noise takes norm about σn/E∥Σ21/2H∥∗\sigma\sqrt n/\mathbb E\|\Sigma_2^{1/2}H\|_*, so the minimal-norm interpolator's risk is about σ2\sigma^2.

Setting. Y=Xw∗+ξY=Xw^*+\xi, with i.i.d. rows N(0,Σ)N(0,\Sigma), d≥nd\ge n, ξ∼N(0,σ2In)\xi\sim N(0,\sigma^2I_n) independent of XX, and w∗w^* arbitrary; L^(w)=1n∥Y−Xw∥22\hat L(w)=\frac1n\|Y-Xw\|_2^2 and L(w)=σ2+∥w−w∗∥Σ2L(w)=\sigma^2+\|w-w^*\|_\Sigma^2. w^\hat w is the minimal-norm interpolator for a norm ∥⋅∥\|\cdot\| with dual ∥⋅∥∗\|\cdot\|_*. With H∼N(0,Id)H\sim N(0,I_d), the Gaussian width and radius of a set are W(S)=Esup⁡s∈S∣⟨s,H⟩∣W(S)=\mathbb E\sup_{s\in S}|\langle s,H\rangle| and rad(S)=sup⁡s∈S∥s∥2\mathrm{rad}(S)=\sup_{s\in S}\|s\|_2, and WΣ(S)=W(Σ1/2S)W_\Sigma(S)=W(\Sigma^{1/2}S), which for symmetric KK makes WΣ(K)/nW_\Sigma(K)/\sqrt n the Rademacher complexity Rn\mathcal R_n. A covariance split Σ=Σ1⊕Σ2\Sigma=\Sigma_1\oplus\Sigma_2 partitions the eigenvectors into a few large-variance directions and a tail. Effective ranks are r(Σ)=TrΣ/∥Σ∥opr(\Sigma)=\mathrm{Tr}\Sigma/\|\Sigma\|_{op} and R(Σ)=(TrΣ)2/Tr(Σ2)R(\Sigma)=(\mathrm{Tr}\Sigma)^2/\mathrm{Tr}(\Sigma^2), and for a general norm r∥⋅∥(Σ)=(E∥Σ1/2H∥∗sup⁡∥w∥≤1∥w∥Σ)2,R∥⋅∥(Σ)=(E∥Σ1/2H∥∗E∥v∗∥Σ)2, r_{\|\cdot\|}(\Sigma)=\Big(\frac{\mathbb E\|\Sigma^{1/2}H\|_*}{\sup_{\|w\|\le1}\|w\|_\Sigma}\Big)^2,\qquad R_{\|\cdot\|}(\Sigma)=\Big(\frac{\mathbb E\|\Sigma^{1/2}H\|_*}{\mathbb E\|v^*\|_\Sigma}\Big)^2, with v∗v^* the least-Σ\Sigma-norm subgradient of ∥Σ1/2H∥∗\|\Sigma^{1/2}H\|_*; R∥⋅∥R_{\|\cdot\|} compares how well v∗v^* fits a Gaussian with how much risk it adds, and for ℓ2\ell_2, r−1≤r∥⋅∥2≤rr-1\le r_{\|\cdot\|_2}\le r and R∥⋅∥2≥(1−4/r)RR_{\|\cdot\|_2}\ge(1-4/\sqrt r)R, the direction the consistency conditions use (the ratio R∥⋅∥2/RR_{\|\cdot\|_2}/R tends to 1 only if r(Σ2)r(\Sigma^2) also diverges, not from r→∞r\to\infty alone).

Why the impossibility results do not apply. Two-sided uniform convergence over a ball fails for a concrete reason (an illustration, not from the papers): a ball containing w^\hat w also contains 2w∗−w^2w^*-\hat w, which interpolates the noise-flipped labels Xw∗−ξXw^*-\xi and has residual 2ξ2\xi, so the two-sided gap is at least about 3σ23\sigma^2. Requiring L^=0\hat L=0 excludes it (the Nagarajan-Kolter mechanism). Bartlett and Long's impossibility covers bounds that cannot use the training error and target L−σ2L-\sigma^2; these bounds use L^\hat L and control L−L^L-\hat L, which is σ2\sigma^2 at a consistent interpolator. Consistency also needs radius (1+o(1))∥w^∥(1+o(1))\|\hat w\|; radius α∥w^∥\alpha\|\hat w\| gives about α2σ2\alpha^2\sigma^2.

Optimistic rate (ZKSS, Theorems 1 and 2). Let FF be continuous with ⟨w−w∗,x⟩≤F(w)\langle w-w^*,x\rangle\le F(w) for all ww, with probability 1−δ′1-\delta' over a single fresh x∼N(0,Σ)x\sim N(0,\Sigma). If n≥196log⁡(12/δ)n\ge196\log(12/\delta), there is β1≤14log⁡(12/δ)/n\beta_1\le14\sqrt{\log(12/\delta)/n} such that with probability 1−2(δ′+δ)1-2(\delta'+\delta), uniformly over all ww, L(w)≤(1+β1)(L^(w)+F(w)n)2.L(w)\le(1+\beta_1)\Big(\sqrt{\hat L(w)}+\frac{F(w)}{\sqrt n}\Big)^2 . The complexity enters only through a bound at one Gaussian point. For a compact KK, any split, δ≤1/4\delta\le1/4 and β2=32(log⁡(1/δ)/n+rank(Σ1)/n)≤1\beta_2=32\big(\sqrt{\log(1/\delta)/n}+\sqrt{\mathrm{rank}(\Sigma_1)/n}\big)\le1, this becomes, uniformly over dilations α≥0\alpha\ge0 and w∈αKw\in\alpha K, L(w)≤(1+β2)(L^(w)+αWΣ2(K)n+[∥w∗∥Σ2+α rad(Σ21/2K)]2log⁡(32/δ)n)2,L(w)\le(1+\beta_2)\Big(\sqrt{\hat L(w)}+\frac{\alpha W_{\Sigma_2}(K)}{\sqrt n}+\big[\|w^*\|_{\Sigma_2}+\alpha\,\mathrm{rad}(\Sigma_2^{1/2}K)\big]\sqrt{\frac{2\log(32/\delta)}{n}}\Big)^2, which is (L^+Rn)2(\sqrt{\hat L}+\mathcal R_n)^2 with constant one, the width taken over the tail covariance only.

Uniform convergence of interpolators (KZSS, Theorem 1 and Corollaries 1 to 3). Restricted to L^(w)=0\hat L(w)=0, the width form is KZSS's Theorem 1 (they prove it directly with a constant C1≤66C_1\le66 in place of 32): sup⁡w∈K,L^(w)=0L(w)≤1+βn[W(Σ21/2K)+(rad(Σ21/2K)+∥w∗∥Σ2)2log⁡(32/δ)]2\sup_{w\in K,\hat L(w)=0}L(w)\le\frac{1+\beta}{n}\big[W(\Sigma_2^{1/2}K)+(\mathrm{rad}(\Sigma_2^{1/2}K)+\|w^*\|_{\Sigma_2})\sqrt{2\log(32/\delta)}\big]^2, with w∗∈Kw^*\in K not required. The width is the main term and the radius its fluctuation, and Σ1\Sigma_1 enters only through β\beta. For a norm ball of radius B≥∥w∗∥B\ge\|w^*\|, the bound is (1+γ)(B E∥Σ21/2H∥∗)2/n(1+\gamma)(B\,\mathbb E\|\Sigma_2^{1/2}H\|_*)^2/n with γ\gamma adding a log⁡(1/δ)/r∥⋅∥(Σ2)\sqrt{\log(1/\delta)/r_{\|\cdot\|}(\Sigma_2)} term, which for ℓ2\ell_2 is (1+γ)B2Tr(Σ2)/n(1+\gamma)B^2\mathrm{Tr}(\Sigma_2)/n. With no structural assumption it is (1+γ)B2Tr(Σ)/n(1+\gamma)B^2\mathrm{Tr}(\Sigma)/n with γ≲(log⁡(1/δ)/n)1/4\gamma\lesssim(\log(1/\delta)/n)^{1/4}, the Zhou et al. conjecture for Gaussian data.

Norm bound and benign overfitting (KZSS, Theorems 2 to 5). If, with probability 1−δ/41-\delta/4, ∥v∗∥Σ2≤(1+ϵ1)E∥v∗∥Σ2\|v^*\|_{\Sigma_2}\le(1+\epsilon_1)\mathbb E\|v^*\|_{\Sigma_2} and ∥Pv∗∥2≤1+ϵ2\|Pv^*\|^2\le1+\epsilon_2 (PP the projection onto span Σ2\mathrm{span}\,\Sigma_2), then with probability 1−δ1-\delta ∥w^∥≤∥w∗∥+(1+ϵ)1/2 σ nE∥Σ21/2H∥∗,ϵ≲log⁡(1/δ)r∥⋅∥(Σ2)+log⁡(1/δ)n+(1+ϵ1)2nR∥⋅∥(Σ2)+ϵ2, \|\hat w\|\le\|w^*\|+(1+\epsilon)^{1/2}\,\sigma\,\frac{\sqrt n}{\mathbb E\|\Sigma_2^{1/2}H\|_*},\qquad \epsilon\lesssim\sqrt{\tfrac{\log(1/\delta)}{r_{\|\cdot\|}(\Sigma_2)}}+\sqrt{\tfrac{\log(1/\delta)}{n}}+(1+\epsilon_1)^2\frac{n}{R_{\|\cdot\|}(\Sigma_2)}+\epsilon_2, with no rank(Σ1)\mathrm{rank}(\Sigma_1) term; for ℓ2\ell_2, given R(Σ2)≳log⁡2(1/δ)R(\Sigma_2)\gtrsim\log^2(1/\delta), the noise part is σn/Tr(Σ2)\sigma\sqrt{n/\mathrm{Tr}(\Sigma_2)} with nlog⁡(1/δ)/R(Σ2)n\log(1/\delta)/R(\Sigma_2) in ϵ\epsilon. Combined with the ball bound, L(w^)≤(1+γ)(1+ϵ)(σ+∥w∗∥ E∥Σ21/2H∥∗/n)2L(\hat w)\le(1+\gamma)(1+\epsilon)\big(\sigma+\|w^*\|\,\mathbb E\|\Sigma_2^{1/2}H\|_*/\sqrt n\big)^2. So in ℓ2\ell_2, L(w^)→σ2L(\hat w)\to\sigma^2 under the benign overfitting conditions: some sequence of splits with rank(Σ1)=o(n)\mathrm{rank}(\Sigma_1)=o(n), ∥w∗∥2Tr(Σ2)/n→0\|w^*\|_2\sqrt{\mathrm{Tr}(\Sigma_2)/n}\to0 (no aliasing) and n/R(Σ2)→0n/R(\Sigma_2)\to0. They contain Bartlett et al.'s conditions, with TrΣ2\mathrm{Tr}\Sigma_2 in place of their TrΣ\mathrm{Tr}\Sigma. For basis pursuit (splits with Σ2\Sigma_2 diagonal) a single rank r1(Σ)=(E∥Σ1/2H∥∞)2/max⁡iΣiir_1(\Sigma)=(\mathbb E\|\Sigma^{1/2}H\|_\infty)^2/\max_i\Sigma_{ii} suffices, giving consistency with junk features when log⁡d=ω(n)\log d=\omega(n), and for Σ=Id\Sigma=I_d the null risk σ2+∥w∗∥22\sigma^2+\|w^*\|_2^2 up to 1+o(1)1+o(1).

What the optimistic rate adds (ZKSS, Theorems 3 to 12). Along the regularization path, sup⁡R≥∥w∗∥2L(w^R)→σ2\sup_{R\ge\|w^*\|_2}L(\hat w_R)\to\sigma^2 under the benign conditions with fixed σ>0\sigma\gt0, for w^R=arg⁡min⁡∥w∥≤RL^(w)\hat w_R=\arg\min_{\|w\|\le R}\hat L(w): every ridge solution with ∥w^λ∥2≥∥w∗∥2\|\hat w_\lambda\|_2\ge\|w^*\|_2 is consistent, so once the signal is fit, how much noise is fit does not matter. Optimally tuned ridge (Corollary 3) is consistent with only the first two conditions, so n/R(Σ2)→0n/R(\Sigma_2)\to0 is needed only to make tuning unnecessary; for LASSO this gives the slow rate, and a compatibility condition gives σ2klog⁡d/n\sigma^2k\log d/n for sparse w∗w^* plus a (β1+ϵ)σ2(\beta_1+\epsilon)\sigma^2 term (exact recovery at σ=0\sigma=0), which Theorem 11 removes for the constrained LASSO. With well-chosen FF, Theorem 1 recovers OLS's σ2/(1−γ)\sigma^2/(1-\gamma) (γ=d/n<1\gamma=d/n\lt1) and isotropic min-norm interpolation's (1−1/γ)∥w∗∥22+σ2γ/(γ−1)(1-1/\gamma)\|w^*\|_2^2+\sigma^2\gamma/(\gamma-1), both sides of the double-descent curve from one inequality. With the local width of Kr={w∈K:∥w−w∗∥Σ≤r}K_r=\{w\in K:\|w-w^*\|_\Sigma\le r\} as complexity (Theorems 9 and 10), the bound places the ERM's error between two radii, a nonasymptotic analogue of the Moreau-envelope characterization of Thrampoulidis et al. The ERM over a closed convex class with F=p ∥w−w∗∥ΣF=\sqrt p\,\|w-w^*\|_\Sigma, p/n≤0.999p/n\le0.999 and n≥Clog⁡(2/δ)n\ge C\log(2/\delta), attains the parametric rate (1+τ)σ2p/n(1+\tau)\sigma^2p/n, while for OLS no bound that sees only L^(w)\hat L(w) beats n−1/4n^{-1/4} slack (Theorem 12).

Step 1 (both papers)

The worst predictor's risk is a Gaussian max problem. Whiten and center at w∗w^*: with X=ZΣ1/2X=Z\Sigma^{1/2} and v=Σ1/2(w−w∗)v=\Sigma^{1/2}(w-w^*), the residual is ξ−Zv\xi-Zv and L=σ2+∥v∥2L=\sigma^2+\|v\|^2. For interpolators, KZSS split Zv=Z1w1+Z2w2Zv=Z_1w_1+Z_2w_2 along the covariance split and get sup⁡L−σ2=dΦ=max⁡{∥w1∥2+∥w2∥2: Z1w1+Z2w2=ξ}\sup L-\sigma^2\overset{d}{=}\Phi=\max\{\|w_1\|^2+\|w_2\|^2:\ Z_1w_1+Z_2w_2=\xi\}. ZKSS write the failure of the optimistic rate as positivity of Φ=sup⁡v inf⁡∥λ∥2=1 ⟨Zv,λ⟩+nσ2+n∥v∥221+β−⟨ξ,λ⟩−F~(v),\Phi=\sup_v\ \inf_{\|\lambda\|_2=1}\ \langle Zv,\lambda\rangle+\sqrt{\frac{n\sigma^2+n\|v\|_2^2}{1+\beta}}-\langle\xi,\lambda\rangle-\tilde F(v), using −∥a∥2=inf⁡∥λ∥=1⟨λ,−a⟩-\|a\|_2=\inf_{\|\lambda\|=1}\langle\lambda,-a\rangle. Either way the random matrix enters through one bilinear term, and the rest is deterministic given ξ\xi (and, in KZSS, Z1Z_1).

Step 2 (both)

Gordon's theorem replaces the matrix by two vectors, in the direction that needs no convexity. Gordon's theorem bounds the lower tail of a min-max of ⟨u,Zw⟩+ψ\langle u,Zw\rangle+\psi by that of the auxiliary problem with ∥w∥⟨G,u⟩+∥u∥⟨H,w⟩\|w\|\langle G,u\rangle+\|u\|\langle H,w\rangle in its place, for any continuous ψ\psi on compact sets; only the upper tail needs convexity (the CGMT). The bad event is the upper tail of a sup-inf, which after negation is exactly the convexity-free half, so KK and FF can be arbitrary. Optimizing the multiplier's direction (after truncation), Pr⁡(Φ>t)≤2Pr⁡(ϕ≥t)\Pr(\Phi\gt t)\le2\Pr(\phi\ge t) with ϕ=sup⁡v n(σ2+∥v∥22)1+β−∥ ∥v∥2G−ξ∥2+⟨H,v⟩−F~(v),\phi=\sup_v\ \sqrt{\frac{n(\sigma^2+\|v\|_2^2)}{1+\beta}}-\big\|\,\|v\|_2G-\xi\big\|_2+\langle H,v\rangle-\tilde F(v), and for KZSS the constraint ∥ξ−Z1w1−∥w2∥G∥2≤⟨w2,H⟩\|\xi-Z_1w_1-\|w_2\|G\|_2\le\langle w_2,H\rangle. One technical point there is that w1w_1 appears only in the deterministic term, so the theorem does not apply verbatim; embedding such variables as extra coordinates scaled by ϵ\epsilon and letting ϵ→0\epsilon\to0 recovers it, since Gordon's theorem uses no quantitative smoothness of ψ\psi.

Step 3 (both)

The auxiliary problem is controlled by a few scalar events, which is why the constant is one. On the events that ∥G∥\|G\| and ∥ξ∥\|\xi\| concentrate and ξ\xi is nearly orthogonal to GG, ∥∥v∥G−ξ∥22≥(1−ρ)(1−α)2n(∥v∥2+σ2),\big\|\|v\|G-\xi\big\|_2^2\ge(1-\rho)(1-\alpha)^2n(\|v\|^2+\sigma^2), so 1+β=[(1−ρ)(1−α)2]−11+\beta=[(1-\rho)(1-\alpha)^2]^{-1} makes the loss part of ϕ\phi nonpositive for every vv at once, and the hypothesis ⟨H,v⟩≤F~(v)\langle H,v\rangle\le\tilde F(v), at x=Σ1/2Hx=\Sigma^{1/2}H, makes the rest nonpositive. In KZSS the same expansion of the squared constraint gives ⟨w2,H⟩2≳n(σ2+∥w1∥2+∥w2∥2)\langle w_2,H\rangle^2\gtrsim n(\sigma^2+\|w_1\|^2+\|w_2\|^2), with the Z1Z_1 block handled as in Step 4: the projection onto HH must account for both the noise and the norm. The proof never applies symmetrization or chaining to the class; in ZKSS's Theorem 1 the only slack is the concentration of two norms and one inner product.

Step 4 (both)

The covariance split puts the width on the tail, at a relative error rank Σ1/n\sqrt{\mathrm{rank}\,\Sigma_1/n}. Bound ⟨w−w∗,x⟩\langle w-w^*,x\rangle by a head term ∥w−w∗∥Σ1∥H1∥2≤L(w) (rankΣ1+O(log⁡))\|w-w^*\|_{\Sigma_1}\|H_1\|_2\le\sqrt{L(w)}\,(\sqrt{\mathrm{rank}\Sigma_1}+O(\sqrt{\log})) plus a Gaussian term in w∗w^* plus sup⁡u∈Σ21/2K∣⟨u,H⟩∣≤WΣ2(K)+rad2log⁡(32/δ)\sup_{u\in\Sigma_2^{1/2}K}|\langle u,H\rangle|\le W_{\Sigma_2}(K)+\mathrm{rad}\sqrt{2\log(32/\delta)} by Gaussian concentration, and use the sum (+∞+\infty off KK) as FF in Theorem 1. Moving the head term to the left side multiplies L\sqrt L by 1−O(rankΣ1/n)1-O(\sqrt{\mathrm{rank}\Sigma_1/n}), negligible only when rankΣ1=o(n)\mathrm{rank}\Sigma_1=o(n); in KZSS the head cannot absorb noise because the o(n)o(n)-dimensional range of Z1Z_1 is nearly orthogonal to ξ\xi and GG, and the near isometry of Z1/nZ_1/\sqrt n on span Σ1\mathrm{span}\,\Sigma_1 turns what it does contribute into the term ∥w1∥2\|w_1\|^2. As a result, the relative error rad/width is computed on Σ2\Sigma_2, where it is small even if spiked eigenvalues make it large on Σ\Sigma. Dilations need no extra work, since the supremum over αK\alpha K is α\alpha times that over KK.

Step 5 (KZSS)

At L^=0\hat L=0 the noise term cancels exactly. The auxiliary constraint gives ϕ≤1+βn(sup⁡u∈Σ21/2K∣⟨u,H⟩∣+∣⟨Σ21/2w∗,H⟩∣)2−σ2\phi\le\frac{1+\beta}{n}\big(\sup_{u\in\Sigma_2^{1/2}K}|\langle u,H\rangle|+|\langle\Sigma_2^{1/2}w^*,H\rangle|\big)^2-\sigma^2, and that −σ2-\sigma^2 cancels the Bayes risk in L=σ2+ΦL=\sigma^2+\Phi. This cancellation is essential, since σ2+(width)2/n\sigma^2+(\text{width})^2/n would be about 2σ22\sigma^2 at the min-norm radius. For a norm ball, duality gives width B E∥Σ21/2H∥∗B\,\mathbb E\|\Sigma_2^{1/2}H\|_* and radius Bsup⁡∥w∥≤1∥w∥Σ2B\sup_{\|w\|\le1}\|w\|_{\Sigma_2}, a relative r∥⋅∥(Σ2)−1/2r_{\|\cdot\|}(\Sigma_2)^{-1/2} correction. With no spectral assumption, take Σ1\Sigma_1 to be the top kk directions: k∥Σ2∥op≤TrΣk\|\Sigma_2\|_{op}\le\mathrm{Tr}\Sigma, so the radius adds a relative error log⁡(1/δ)/k\sqrt{\log(1/\delta)/k} and the split k/n\sqrt{k/n}, and k≍nlog⁡(1/δ)k\asymp\sqrt{n\log(1/\delta)} balances them at (log⁡(1/δ)/n)1/4(\log(1/\delta)/n)^{1/4}.

Step 6 (KZSS)

The minimal norm is at most ∥w∗∥\|w^*\| plus the minimal norm needed to interpolate pure noise, and that problem is convex. By the triangle inequality, min⁡Xw=Y∥w∥≤∥w∗∥+min⁡Zw=ξ∥Σ−1/2w∥\min_{Xw=Y}\|w\|\le\|w^*\|+\min_{Zw=\xi}\|\Sigma^{-1/2}w\|; the crude additive ∥w∗∥\|w^*\| later becomes the aliasing term. Now the target is an upper bound on a minimum, the direction that needs convexity, and here it holds: a norm under a linear constraint. The CGMT bounds it by the auxiliary problem min⁡{∥Σ−1/2w∥: ∥ξ−∥w∥2G∥2≤⟨H,w⟩}\min\{\|\Sigma^{-1/2}w\|:\ \|\xi-\|w\|_2G\|_2\le\langle H,w\rangle\}.

Step 7 (KZSS)

One feasible point, built from a dual-norm subgradient, gives the norm bound (Lemma 8). An upper bound on a minimum needs one feasible point: the direction that fits the low-variance Gaussian best per unit norm, scaled to fit the noise. Up to 1+o(1)1+o(1), the constraint is n(σ2+∥Σ1/2w∥22)≤⟨H,Σ1/2w⟩2n(\sigma^2+\|\Sigma^{1/2}w\|_2^2)\le\langle H,\Sigma^{1/2}w\rangle^2. Try w=s Pv∗w=s\,Pv^*: since Σ1/2P=Σ21/2\Sigma^{1/2}P=\Sigma_2^{1/2} (orthogonal spans), the fit is s∥Σ21/2H∥∗s\|\Sigma_2^{1/2}H\|_*, the risk term s∥v∗∥Σ2s\|v^*\|_{\Sigma_2}, and the point is feasible for s2≈σ2(∥Σ21/2H∥∗2n−∥v∗∥Σ22)−1. s^2\approx\sigma^2\Big(\frac{\|\Sigma_2^{1/2}H\|_*^2}{n}-\|v^*\|_{\Sigma_2}^2\Big)^{-1}. Gaussian concentration puts ∥Σ21/2H∥∗\|\Sigma_2^{1/2}H\|_* within a relative r∥⋅∥(Σ2)−1/2r_{\|\cdot\|}(\Sigma_2)^{-1/2} of its mean, and the subtracted term is a relative (1+ϵ1)2n/R∥⋅∥(Σ2)(1+\epsilon_1)^2n/R_{\|\cdot\|}(\Sigma_2), so RR compares the fit of the noise-fitting direction with the risk it adds, and rank Σ1\mathrm{rank}\,\Sigma_1 never enters because the construction avoids the large-variance directions. In ℓ2\ell_2 everything reduces to traces: Bernstein turns the R∥⋅∥R_{\|\cdot\|} term into nlog⁡(1/δ)/R(Σ2)n\log(1/\delta)/R(\Sigma_2), and Gaussian Poincaré replaces E∥Σ21/2H∥2\mathbb E\|\Sigma_2^{1/2}H\|_2 by TrΣ2\sqrt{\mathrm{Tr}\Sigma_2}.

Step 8 (KZSS; ZKSS Corollary 1)

Setting the ball radius to the norm bound gives benign overfitting. Take B=∥w∗∥+(1+ϵ)1/2σn/E∥Σ21/2H∥∗B=\|w^*\|+(1+\epsilon)^{1/2}\sigma\sqrt n/\mathbb E\|\Sigma_2^{1/2}H\|_*. The norm bound puts w^\hat w in the ball (for consistency, its hypothesis on ∥v∗∥Σ2\|v^*\|_{\Sigma_2} follows from Markov's inequality), the ball bound applies, and the width cancels, leaving σ\sigma plus an aliasing term. In ℓ2\ell_2, R≤r2R\le r^2, so n/R(Σ2)→0n/R(\Sigma_2)\to0 also forces the radius correction to vanish. Moving the eigenvalues within a factor τ\tau of the top into Σ1\Sigma_1 moves only o(n)o(n) directions and makes r(Σ2)=ω(n)r(\Sigma_2)=\omega(n), so Bartlett et al.'s split loses nothing for consistency.

Step 9 (KZSS)

For ℓ1\ell_1 a single effective rank suffices, and folding the signal into the noise handles isotropic data. Every subgradient of ∥⋅∥∞\|\cdot\|_\infty lies in the convex hull of signed basis vectors, so ∥v∗∥Σ2≤max⁡i(Σ2)ii\|v^*\|_{\Sigma_2}\le\max_i\sqrt{(\Sigma_2)_{ii}} deterministically and one rank suffices. For Σ=I\Sigma=I the aliasing term ∥w∗∥12log⁡d/n\|w^*\|_1\sqrt{2\log d/n} diverges in the needed regime log⁡d≫n\log d\gg n, so rewrite Y=XSc0+ξ′Y=X_{S^c}0+\xi' with ξ′=XSwS∗+ξ\xi'=X_Sw_S^*+\xi, Gaussian and independent of XScX_{S^c}; the bound then has no ∥w∗∥1\|w^*\|_1, but the target risk becomes σ2+∥w∗∥22\sigma^2+\|w^*\|_2^2.

Step 10 (ZKSS)

The risk is flat along the path, by convex interpolation between w∗w^* and the interpolator, and tuning needs no norm bound. The optimistic bound is affine along the segment from w∗w^* to an interpolator w0w_0 and about σ\sigma at both ends. Since Xw0=YXw_0=Y, the point wa=(1−a)w∗+aw0w_a=(1-a)w^*+aw_0 has residual (1−a)ξ(1-a)\xi, so L^(wa)=(1−a)L^(w∗)\sqrt{\hat L(w_a)}=(1-a)\sqrt{\hat L(w^*)}, and a convex FF (here affine in ∥w∥\|w\|) satisfies F(wa)≤(1−a)F(w∗)+aF(w0)F(w_a)\le(1-a)F(w^*)+aF(w_0); the constrained ERM at the matching norm does at least as well on both terms. For the tuned estimator, KKT gives some λ∗\lambda^* with w^λ∗=arg⁡min⁡{∥w∥:L^(w)≤∥ξ∥22/n}\hat w_{\lambda^*}=\arg\min\{\|w\|:\hat L(w)\le\|\xi\|_2^2/n\}, and w∗w^* is feasible, so ∥w^λ∗∥≤∥w∗∥\|\hat w_{\lambda^*}\|\le\|w^*\| with training error about σ2\sigma^2, and the dilated width bound applies without any interpolator norm bound.

Step 11 (ZKSS)

Sharp constants come from solving the quadratic that Theorem 1 produces. When FF depends on ww through ∥w−w∗∥Σ\|w-w^*\|_\Sigma, the bound is a quadratic inequality in the excess risk, and its roots give two-sided bounds. For OLS, Cauchy-Schwarz gives F≈d eF\approx\sqrt d\,e, e=L−σ2e=\sqrt{L-\sigma^2}, so Theorem 1 reads σ2+e2≤(L^+γ e)2\sigma^2+e^2\le(\sqrt{\hat L}+\sqrt\gamma\,e)^2 up to a factor 1+ϵ1+\epsilon, ϵ≲log⁡(36/δ)/n\epsilon\lesssim\sqrt{\log(36/\delta)/n}, and completing the square, (e−γL^1−γ)2≤11−γ(L^1−γ−σ2);\Big(e-\frac{\sqrt{\gamma\hat L}}{1-\gamma}\Big)^2\le\frac1{1-\gamma}\Big(\frac{\hat L}{1-\gamma}-\sigma^2\Big); at w^OLS\hat w_{OLS}, nL^/σ2∼χn−d2n\hat L/\sigma^2\sim\chi^2_{n-d} makes the right side O(ϵ)O(\epsilon), so e2→σ2γ/(1−γ)e^2\to\sigma^2\gamma/(1-\gamma), and the square root of that O(ϵ)O(\epsilon) slack gives the n−1/4n^{-1/4}. For isotropic interpolation, split w−w∗w-w^* along span(w∗)\mathrm{span}(w^*) and its complement, where only the complement sees the full ∥x∥≈d\|x\|\approx\sqrt d; Pythagoras makes the bound at L^=0\hat L=0 a quadratic in ∥w∥\|w\|, and the CGMT with a feasible point in span(w∗,H)\mathrm{span}(w^*,H) bounds the minimal norm.

Step 12 (ZKSS)

With the local width as complexity, the bound and the ERM meet at the minimizer of one convex function. On KrK_r, Gaussian concentration makes F≡WΣ(Kr)+Crlog⁡(2/δ)F\equiv W_\Sigma(K_r)+Cr\sqrt{\log(2/\delta)} valid, so Theorem 1 bounds L^\sqrt{\hat L} below on each sphere ∥w−w∗∥Σ=r\|w-w^*\|_\Sigma=r, while the convex half of the minimax theorem bounds min⁡KL^\min_K\sqrt{\hat L} above. They meet because ψ(r)≈σ2+r2−WΣ(Kr)/n\psi(r)\approx\sqrt{\sigma^2+r^2}-W_\Sigma(K_r)/\sqrt n is convex: convex combinations of KsK_s and KtK_t lie in the localized set at the averaged radius, so the local width is concave. Convexity of L^\sqrt{\hat L} in ww then keeps the ERM inside the interval of radii, with a union bound over a grid for the lower end.

Step 13 (ZKSS)

The n−1/4n^{-1/4} slack is unavoidable for bounds that see only L^\hat L, and the first-order condition avoids it. Along wa=w∗+a(w^OLS−w∗)w_a=w^*+a(\hat w_{OLS}-w^*), Pythagoras gives L^(wa)=L^(w^OLS)+(1−a)2∥w^OLS−w∗∥Σ^2,L(wa)−σ2=a2(L(w^OLS)−σ2),\hat L(w_a)=\hat L(\hat w_{OLS})+(1-a)^2\|\hat w_{OLS}-w^*\|_{\hat\Sigma}^2,\qquad L(w_a)-\sigma^2=a^2\big(L(\hat w_{OLS})-\sigma^2\big), so training error moves to second order and risk to first order, and a−1≍n−1/4a-1\asymp n^{-1/4} keeps the training gap at n−1/2n^{-1/2}, the size of L^(w^OLS)\hat L(\hat w_{OLS})'s own fluctuation, while the risk moves by (a2−1)(L(w^OLS)−σ2)≍n−1/4(a^2-1)(L(\hat w_{OLS})-\sigma^2)\asymp n^{-1/4}, since L(w^OLS)−σ2→σ2γ/(1−γ)>0L(\hat w_{OLS})-\sigma^2\to\sigma^2\gamma/(1-\gamma)\gt0. The ERM's fast rate also uses the first-order condition. KKT gives ∥w^−w∗∥Σ^2≤1n⟨ξ,X(w^−w∗)⟩≤∥ξ∥nF(w^)\|\hat w-w^*\|_{\hat\Sigma}^2\le\frac1n\langle\xi,X(\hat w-w^*)\rangle\le\frac{\|\xi\|}{n}F(\hat w), since Z⊤ξZ^\top\xi is Gaussian given ξ\xi; the noiseless Theorem 1 lifts the empirical norm to the population one, and with F=p ∥w−w∗∥ΣF=\sqrt p\,\|w-w^*\|_\Sigma this solves to e2≲σ2p/ne^2\lesssim\sigma^2p/n. This fails in the proportional regime, where the two norms differ by more than 1+o(1)1+o(1).

Consequences and limits. Uniform convergence explains interpolation once it is localized (to interpolators, or by training error) with leading constant one; arguments that lose a constant cannot show consistency. Benign overfitting is then a statement about norms: under the ℓ2\ell_2 conditions every interpolator within (1+o(1))(1+o(1)) of the norm bound is consistent, and the spectrum acts only through the split, with a low-rank head learned as in classical regression, no aliasing (∥w∗∥2TrΣ2=o(n)\|w^*\|^2\mathrm{Tr}\Sigma_2=o(n)), and R(Σ2)=ω(n)R(\Sigma_2)=\omega(n) so that fitting the noise adds negligible risk (at R(Σ2)≍nR(\Sigma_2)\asymp n the bound is cσ2c\sigma^2 with c>1c\gt1). For basis pursuit KZSS give the first consistency results with Gaussian covariates and σ>0\sigma\gt0, at slow rates. Limits: Gaussian, well-specified data are essential to Gordon's theorem; rates are somewhat worse than Bartlett et al.'s (rankΣ1/n\sqrt{\mathrm{rank}\Sigma_1/n} for rankΣ1/n\mathrm{rank}\Sigma_1/n, plus an r(Σ2)−1/2r(\Sigma_2)^{-1/2} term); the interpolator's norm is a separate input; no lower bound beyond L≥σ2L\ge\sigma^2 is proved for interpolators; the local-width results need convex KK.

Terms

Gaussian width W(S)W(S)
Esup⁡s∈S∣⟨s,H⟩∣\mathbb E\sup_{s\in S}|\langle s,H\rangle|; the main term, bound ≈W(Σ21/2K)2/n\approx W(\Sigma_2^{1/2}K)^2/n
Radius rad(S)\mathrm{rad}(S)
sup⁡s∈S∥s∥2\sup_{s\in S}\|s\|_2; Lipschitz constant of the supremum, sets the fluctuation
Optimistic rate
L(w)≤(1+o(1))(L^(w)+Rn)2L(w)\le(1+o(1))(\sqrt{\hat L(w)}+\mathcal R_n)^2: the gap shrinks with training error; here with constant one
Complexity functional FF
any continuous FF with ⟨w−w∗,x⟩≤F(w) ∀w\langle w-w^*,x\rangle\le F(w)\ \forall w w.h.p. for one x∼N(0,Σ)x\sim N(0,\Sigma)
Covariance split Σ1⊕Σ2\Sigma_1\oplus\Sigma_2
Partition of eigenvectors; Σ1\Sigma_1 (rank o(n)o(n)) by Wishart concentration, Σ2\Sigma_2 by Gordon
GMT vs CGMT
GMT: lower tail of a min-max for any continuous ψ\psi; CGMT adds the upper tail when ψ\psi is convex-concave
AO constraint
∥ξ−∥w∥2G∥2≤⟨H,w⟩\|\xi-\|w\|_2G\|_2\le\langle H,w\rangle, roughly σ2+∥w∥2≤⟨H,w⟩2/n\sigma^2+\|w\|^2\le\langle H,w\rangle^2/n
R∥⋅∥(Σ)R_{\|\cdot\|}(\Sigma), v∗v^*
(E∥Σ1/2H∥∗/E∥v∗∥Σ)2(\mathbb E\|\Sigma^{1/2}H\|_*/\mathbb E\|v^*\|_\Sigma)^2; v∗v^* is the least-Σ\Sigma-norm subgradient of ∥Σ1/2H∥∗\|\Sigma^{1/2}H\|_*, the noise-fitting direction that adds the least risk
Norm bound
∥w^∥≤∥w∗∥+(1+ϵ)1/2σn/E∥Σ21/2H∥∗\|\hat w\|\le\|w^*\|+(1+\epsilon)^{1/2}\sigma\sqrt n/\mathbb E\|\Sigma_2^{1/2}H\|_*; in ℓ2\ell_2, σn/TrΣ2\sigma\sqrt{n/\mathrm{Tr}\Sigma_2}
Benign conditions
rankΣ1/n→0\mathrm{rank}\Sigma_1/n\to0, ∥w∗∥TrΣ2/n→0\|w^*\|\sqrt{\mathrm{Tr}\Sigma_2/n}\to0, n/R(Σ2)→0n/R(\Sigma_2)\to0

Quiz

  1. Why does KZSS's Theorem 1 apply Gordon's theorem only to the Σ2\Sigma_2 block and handle Σ1\Sigma_1 by Wishart concentration and approximate orthogonality?
  2. With no spectral assumption, KZSS's ball bound is sup⁡L≤(1+γ)B2Tr(Σ)/n\sup L\le(1+\gamma)B^2\mathrm{Tr}(\Sigma)/n with γ\gamma only of order (log⁡(1/δ)/n)1/4(\log(1/\delta)/n)^{1/4}. Where does the fourth root come from?
  3. Why is it essential for interpolation learning that the optimistic rate have leading constant exactly one rather than an absolute constant c>1c\gt1?
  4. Optimally tuned ridge (Corollary 3) is consistent without n/R(Σ2)→0n/R(\Sigma_2)\to0. Why is that condition needed for the min-norm interpolator but not here?

Sources