Sam Babichenko

Reading notes / Benign overfitting

Benign, Tempered, or Catastrophic

Mallinar, Simon, Abedsoltan, Pandit, Belkin and Nakkiran, 2022

Context. Benign-overfitting theory reaches Bayes risk only with dimension growing with nn or a very slowly decaying spectrum. Networks trained to interpolation on fixed-dimension data behave differently: their test error neither diverges nor reaches Bayes risk as n→∞n\to\infty, and it grows with the label noise. The paper names this middle regime tempered and uses kernel regression to show which spectral property selects each regime: eigendecay slower than every power law is benign, a power law is tempered, and faster than every power law is catastrophic. On these notes' reading of the formula (not the paper's wording), the learnabilities of ridgeless kernel regression sum to exactly nn over the eigenmodes, and noise is amplified by the inverse of the share of that total on partially learned modes. For λi=i−α\lambda_i=i^{-\alpha} that share tends to 1/α1/\alpha, so the risk tends to ασ2\alpha\sigma^2.

Setting: the taxonomy. A procedure with expected risk RnR_n is benign if lim⁡Rn=R∗\lim R_n=R^*, the Bayes risk, tempered if lim⁡Rn∈(R∗,∞)\lim R_n\in(R^*,\infty), and catastrophic if lim⁡Rn=∞\lim R_n=\infty (random guessing, 1−1/K1-1/K, for KK-class classification). The diagnostic is the noise profile, the limiting clean test risk as a function of the training-label noise: flat at Bayes when benign, a finite increasing curve when tempered. Examples: kk-NN with k∼log⁡nk\sim\log n is benign, 1-NN tempered, random-features regression at p=np=n catastrophic.

Setting: kernel regression and the risk estimate. Labels are yi=f∗(xi)+ηiy_i=f^*(x_i)+\eta_i, ηi∼N(0,σ2)\eta_i\sim N(0,\sigma^2), KR with ridge δ≥0\delta\ge0 is f^(x)=K(x,Dn)(K(Dn,Dn)+δIn)−1Y\hat f(x)=K(x,D_n)(K(D_n,D_n)+\delta I_n)^{-1}Y (with δ\delta, not nδn\delta), and f∗=∑iviϕif^*=\sum_iv_i\phi_i in the Mercer eigenbasis (λi,ϕi)(\lambda_i,\phi_i). The paper analyzes the eigenlearning estimate of Simon et al. (equivalent to Bordelon et al. and others), cited without derivation, nonrigorous, expected to become exact as n→∞n\to\infty: En=E0(∑i(1−Li)2vi2+σ2),E0=nn−∑jLj2,Li=λiλi+κ,∑iλiλi+κ+δκ=n. \mathcal E_n=\mathcal E_0\Big(\sum_i(1-\mathcal L_i)^2v_i^2+\sigma^2\Big),\qquad \mathcal E_0=\frac{n}{n-\sum_j\mathcal L_j^2},\qquad \mathcal L_i=\frac{\lambda_i}{\lambda_i+\kappa},\qquad \sum_i\frac{\lambda_i}{\lambda_i+\kappa}+\frac{\delta}{\kappa}=n. Li∈[0,1]\mathcal L_i\in[0,1] is the learnability of mode ii, κ\kappa the effective regularization, E0≥1\mathcal E_0\ge1 the overfitting coefficient. Assumption A.1: eigenvalues strictly decreasing with finite trace, ∑ivi2<∞\sum_iv_i^2<\infty, and (e) the target weight on modes below λ\lambda vanishes as λ→0\lambda\to0 (weight outside the RKHS counts as noise).

Theorem 3.1 (KR trichotomy). Under Assumption A.1 with σ2>0\sigma^2>0: (a) if δ>0\delta>0, or if δ=0\delta=0 and λi=i−1log⁡−αi\lambda_i=i^{-1}\log^{-\alpha}i (i≥2i\ge2) for some α>1\alpha>1, then lim⁡nEn=σ2\lim_n\mathcal E_n=\sigma^2 (benign); (b) if δ=0\delta=0 and λi=i−α\lambda_i=i^{-\alpha}, α>1\alpha>1, then lim⁡nEn=ασ2\lim_n\mathcal E_n=\alpha\sigma^2, so the excess risk is (α−1)σ2(\alpha-1)\sigma^2 (tempered); (c) if δ=0\delta=0 and λi=i−log⁡i\lambda_i=i^{-\log i}, or more generally λiλi+1≥i−log⁡i(i+1)−log⁡(i+1)\dfrac{\lambda_i}{\lambda_{i+1}}\ge\dfrac{i^{-\log i}}{(i+1)^{-\log(i+1)}} for all ii, then lim⁡nEn=∞\lim_n\mathcal E_n=\infty (catastrophic). Clause (a) with δ=0\delta=0 is the benign condition of Bartlett et al. (2020), and (b) says the power laws they showed are not benign are tempered.

Consequences for named kernels (heuristic). Only the spectral tail matters, so each named kernel is matched to a clause by its known tail (not a formal corollary: spherical-harmonic spectra are degenerate and only asymptotically of the stated form). The Laplace kernel on a dd-manifold and the ReLU NTK on the sphere have λi∼i−(d+1)/d\lambda_i\sim i^{-(d+1)/d}, so clause (b) with α=1+1/d\alpha=1+1/d would give excess σ2/d\sigma^2/d; the paper claims only Θ(1/d)\Theta(1/d), tempered and approaching benign as dd grows. The Gaussian kernel has λi∼e−ci2/d\lambda_i\sim e^{-ci^{2/d}}, whose ratio eventually beats that of clause (c), so ridgeless Gaussian KR is catastrophic. Early stopping acts as a ridge, so early-stopped wide networks should be benign, and since nearly any kernel on the sphere is the NTK of some activation, NTK-regime networks can be tempered or catastrophic.

Empirical results. On pure-noise labels, ridge Gaussian KR goes to 0, ridgeless Laplace KR plateaus with excess about 1/d1/d, and ridgeless Gaussian KR diverges. Synthetic KR with exact spectra i−αi^{-\alpha} reaches ≈ασ2\approx\alpha\sigma^2. Interpolating networks (ResNets, VGG, MLPs) have noise profiles converging to a nonzero, roughly linear curve; early-stopped versions approach benign.

Step 1

Where the estimate comes from (background the paper cites). KR is linear regression on the eigenfeatures; universality replaces the ϕi(xj)\phi_i(x_j) by i.i.d. Gaussians, a first-order deterministic equivalent of the resolvent gives the expected learned fraction diag(Li)\mathrm{diag}(\mathcal L_i), and the second-order one says the fluctuations inflate bias and variance by a common factor, which implicit differentiation of the κ\kappa equation identifies as E0=n ∂κ/∂δ\mathcal E_0=n\,\partial\kappa/\partial\delta. Everything after this is rigorous analysis of the formula; the link to actual KR rests on universality, checked numerically.

Step 2

κ\kappa is bounded below by a multiple of λn(1+γ)\lambda_{n(1+\gamma)} and above by the per-sample tail mass, so the bias vanishes and only E0\mathcal E_0 matters. In the κ\kappa equation, bound each learnability by 1 or by λi/κ\lambda_i/\kappa, split near nn, and get γ λn(1+γ) ≤ κ ≤ 1nγ(δ+∑i≥n(1−γ)λi), \gamma\,\lambda_{n(1+\gamma)}\ \le\ \kappa\ \le\ \frac{1}{n\gamma}\Big(\delta+\sum_{i\ge n(1-\gamma)}\lambda_i\Big), the upper bound for γ∈(0,1)\gamma\in(0,1), the lower for any γ>0\gamma>0. The upper bound gives κ=O(1/n)\kappa=O(1/n), so every fixed mode is eventually learned, and A.1(e) makes the target weight on the rest small; the bias vanishes, and the regime is decided by lim⁡E0∈[1,∞]\lim\mathcal E_0\in[1,\infty] whatever the target.

Step 3

E0\mathcal E_0 measures how spread out the learnability is. The κ\kappa equation is a conservation law, ∑iLi=n−δ/κ≤n\sum_i\mathcal L_i=n-\delta/\kappa\le n, so ∑Li2≤n\sum\mathcal L_i^2\le n and E0≥1\mathcal E_0\ge1. In the ridgeless case it lets one replace nn by ∑Li\sum\mathcal L_i: 1E0=1−1n∑iLi2 =δ=0 ∑iLi(1−Li)∑iLi. \frac1{\mathcal E_0}=1-\frac1n\sum_i\mathcal L_i^2\ \overset{\delta=0}{=}\ \frac{\sum_i\mathcal L_i(1-\mathcal L_i)}{\sum_i\mathcal L_i}. Benign means ∑Li2=o(n)\sum\mathcal L_i^2=o(n), the total learnability spread thinly over many barely learned modes. Catastrophic means ∑Li(1−Li)=o(n)\sum\mathcal L_i(1-\mathcal L_i)=o(n): learnability becomes a 0/1 step, nn modes each fit their own noise, and nothing averages it out.

Step 4

With a fixed ridge, E0→1\mathcal E_0\to1. The ridge keeps κ≥δ/n\kappa\ge\delta/n. Bound the first nγn\gamma modes trivially and the rest by λi2/κ2≤n2λi2/δ2\lambda_i^2/\kappa^2\le n^2\lambda_i^2/\delta^2; since a monotone summable sequence has iλi→0i\lambda_i\to0, 1n∑Li2→γ\frac1n\sum\mathcal L_i^2\to\gamma at most, and γ\gamma is arbitrary. The convention matters here. With nδn\delta in place of δ\delta, κ≥δ\kappa\ge\delta, every Li≤λ1/(λ1+δ)<1\mathcal L_i\le\lambda_1/(\lambda_1+\delta)<1, and the bias no longer vanishes.

Step 5

For ridgeless λi=1/(ilog⁡αi)\lambda_i=1/(i\log^\alpha i) the tail acts as a ridge, and E0→1\mathcal E_0\to1. The tail beyond nn carries mass ≍log⁡1−αn≫nλn\asymp\log^{1-\alpha}n\gg n\lambda_n, so it pushes κ\kappa well above λn\lambda_n: feeding κ≥λ2n\kappa\ge\lambda_{2n} back into the κ\kappa equation with an integral comparison gives κ≳[(α−1)nlog⁡α−1n]−1\kappa\gtrsim[(\alpha-1)n\log^{\alpha-1}n]^{-1}, a factor ≍log⁡n\asymp\log n above λn\lambda_n. Bounding the first n/log⁡log⁡nn/\log\log n modes trivially then leaves ∑Li2=o(n)\sum\mathcal L_i^2=o(n), so E0→1\mathcal E_0\to1, slowly.

Step 6

For λi=i−α\lambda_i=i^{-\alpha}, E0→α\mathcal E_0\to\alpha. Sums become integrals, and the answer is a ratio of two integrals that does not involve κ\kappa: with Ip=∫0∞(1+xα)−pdxI_p=\int_0^\infty(1+x^\alpha)^{-p}dx and i=κ−1/αxi=\kappa^{-1/\alpha}x, the κ\kappa equation gives n≈κ−1/αI1n\approx\kappa^{-1/\alpha}I_1 and ∑Li2≈κ−1/αI2\sum\mathcal L_i^2\approx\kappa^{-1/\alpha}I_2, so ∑Li2/n→I2/I1\sum\mathcal L_i^2/n\to I_2/I_1. The paper states both Beta integrals without derivation; a simpler route, these notes' and not the paper's, is one integration by parts, whose boundary term vanishes because α>1\alpha>1: I1=∫0∞αxα(1+xα)2dx=α(I1−I2) ⟹ I2I1=1−1α. I_1=\int_0^\infty\frac{\alpha x^\alpha}{(1+x^\alpha)^2}dx=\alpha(I_1-I_2)\ \Longrightarrow\ \frac{I_2}{I_1}=1-\frac1\alpha. So a share 1/α1/\alpha of the learnability sits on partially learned modes. Only the exponent matters, since the profile 1/(1+(i/i∗)α)1/(1+(i/i^*)^\alpha) is scale-free with steepness α\alpha.

Step 7

Faster than every power law, E0→∞\mathcal E_0\to\infty. When the tail beyond each index mm is o(mλm)o(m\lambda_m), the two bounds of Step 2 constrain κ\kappa so tightly that learnability becomes a step at i≈ni\approx n. Telescoping the ratio condition compares λ\lambda with e−log⁡2ie^{-\log^2i}, whose tail ratio is about m/(2log⁡m)=o(m)m/(2\log m)=o(m). Bounding L(1−L)\mathcal L(1-\mathcal L) by κ/λi\kappa/\lambda_i below nn and λi/κ\lambda_i/\kappa above, and bounding nγn\gamma indices on each side of nn trivially, gives 1/E0≤2γ+O(1/(γlog⁡n))1/\mathcal E_0\le2\gamma+O(1/(\gamma\log n)). The paper says only that the ratio condition was easier to work with than λi=O(i−log⁡i)\lambda_i=O(i^{-\log i}); a likely reason (these notes') is that a big-O bound does not control the tail ratio.

Consequences and limits. Interpolation alone does not determine the regime; the regime is set by how sharply learnability falls off at index nn, that is, by the spectral tail exponent. A heuristic consistent with all three clauses, not in the paper: E0≈\mathcal E_0\approx the local exponent −dlog⁡λ/dlog⁡i-d\log\lambda/d\log i at nn, which tends to 1, equals α\alpha, and is 2log⁡n2\log n in the three cases (numerically E0=13.8\mathcal E_0=13.8 at n=1000n=1000 for i−log⁡ii^{-\log i}), though rough when the exponent drifts. In fixed dimension, then, common kernels are tempered, smooth kernels catastrophic, and a ridge or early stopping benign, as with networks. Limits: the estimate is nonrigorous; the trichotomy is proved only for three exact spectral families; the network results are empirical, with the NTK link heuristic; and why classification noise profiles have slope near one, while KR gives α−1\alpha-1, is unexplained.

Terms

Benign / tempered / catastrophic
lim⁡nRn=R∗\lim_nR_n=R^* / ∈(R∗,∞)\in(R^*,\infty) / =∞=\infty (classification: 1−1/K1-1/K in place of ∞\infty)
Noise profile PA(σ)P_{\mathcal A}(\sigma)
lim⁡n\lim_n expected clean test risk as a function of training-label noise level
Learnability Li\mathcal L_i
λi/(λi+κ)∈[0,1]\lambda_i/(\lambda_i+\kappa)\in[0,1]: fraction of mode ii of the target that KR recovers
Effective regularization κ\kappa
root of ∑iλi/(λi+κ)+δ/κ=n\sum_i\lambda_i/(\lambda_i+\kappa)+\delta/\kappa=n; between γλn(1+γ)\gamma\lambda_{n(1+\gamma)} and (δ+tail)/(nγ)(\delta+\text{tail})/(n\gamma)
Conservation law
∑iLi=n−δ/κ\sum_i\mathcal L_i=n-\delta/\kappa; exactly nn when ridgeless
Overfitting coefficient E0\mathcal E_0
n/(n−∑Li2)=n ∂κ/∂δ≥1n/(n-\sum\mathcal L_i^2)=n\,\partial\kappa/\partial\delta\ge1; multiplies bias and noise
Ridgeless 1/E01/\mathcal E_0
∑Li(1−Li)/∑Li\sum\mathcal L_i(1-\mathcal L_i)/\sum\mathcal L_i: share of learnability on partially learned modes
Power-law limit
λi=i−α⇒E0→α\lambda_i=i^{-\alpha}\Rightarrow\mathcal E_0\to\alpha, excess risk (α−1)σ2(\alpha-1)\sigma^2
Laplace / ReLU NTK
λi∼i−(d+1)/d⇒\lambda_i\sim i^{-(d+1)/d}\Rightarrow excess Θ(σ2/d)\Theta(\sigma^2/d) (σ2/d\sigma^2/d for an exact power law): tempered
Gaussian (RBF) kernel
λi∼e−ci2/d\lambda_i\sim e^{-ci^{2/d}}, faster than any power law ⇒\Rightarrow ridgeless KR catastrophic

Quiz

  1. With δ=0\delta=0, which fact turns 1/E01/\mathcal E_0 into ∑iLi(1−Li)/∑iLi\sum_i\mathcal L_i(1-\mathcal L_i)/\sum_i\mathcal L_i?
  2. What property of a spectrum satisfying the ratio condition in clause (c) makes E0\mathcal E_0 diverge?
  3. For λi=1/(ilog⁡αi)\lambda_i=1/(i\log^\alpha i) with no ridge, why is interpolation benign?
  4. Why do the authors expect optimally early-stopped networks to be benign while the same networks trained to interpolation are tempered?

Sources