Tail-triggered adaptive ridge head
Implementation & benchmark of arXiv:2607.09547 — High-Dimensional Interpolators Can Be Fragile: Heavy Tails and High-Dimensional Large Deviations
Source paper: High-Dimensional Interpolators Can Be Fragile: Heavy Tails and High-Dimensional Large Deviations arXiv:2607.09547 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace a fixed ridge coefficient in a neural network's final head with a controller driven by inverse spectral mass and hard-edge mass. The head can remain weakly regularized when the feature spectrum is healthy, but automatically increases ridge strength when small eigenvalues signal a high-risk interpolation regime.
Formulas
Mathematical statement
The paper's exact risk decomposition is $R_n=b_n+H_n$, with $H_n=(1/n)\sum_i\lambda_{i,n}^{-1}$. Its cutoff decomposition $H_n=M_n+S_n$ separates eigenvalues above and below $\eta_n$, where $S_n$ captures microscopic hard-edge instability. The abstract states that ridge regularization suppresses fixed right-tail deviations at an $n^2$ scale, while ridgeless regression has only $n\log n$-scale decay. Adapt this by solving a ridge head $W_\gamma=(Z^\top Z+B\gamma I_d)^{-1}Z^\top Y$, where $Z\in\mathbb{R}^{B\times d}$ is the feature matrix, $Y$ contains targets, $B$ is batch size, $d$ is feature dimension, and $\gamma\geq0$ is the ridge coefficient. Estimate $\hat H$ and $\hat S$ from the eigenvalues of $C=Z^\top Z/B+\varepsilon I_d$, and increase $\gamma$ when either statistic exceeds a baseline.
Implementation notes
Apply this to a final linear probe, classifier head, regression head, or LoRA-style adaptation head while initially freezing the backbone. Every K steps, collect a feature batch Z_t and targets Y_t. Compute C_t = Z_t.T @ Z_t / B + eps*I, obtain eigenvalues lam, and calculate H_t = mean(1/(lam+eps)) and S_t = mean((lam < eta)/(lam+eps)), using a sigmoid gate instead of the Boolean indicator if a differentiable diagnostic is needed. Maintain exponential moving averages of both statistics. During the first 100 batches, define H0 and S0 as robust medians; initialize gamma to a small value. Update with gamma = clip(gamma * exp(rho*((H-H0)/H0 + (S-S0)/max(S0,eps0))), gamma_min, gamma_max). For closed-form head fitting, solve W = CholeskySolve(Z.T@Z + B*gamma*I, Z.T@Y). For end-to-end training, use gamma*||W||_F^2 as the head regularizer and update gamma only every K steps. The mathematical quantities supplied by the paper are the inverse-trace decomposition and hard-edge separation; K, rho, thresholds, clipping bounds, EMA decay, and numerical floors are empirical. First compare against ridgeless least squares, fixed ridge, and AdamW weight decay using a width-1024 MLP on CIFAR-10 or a small tabular regression benchmark. Track clean accuracy or MSE, 99th-percentile loss under corruptions, maximum output norm, minimum feature eigenvalue, selected gamma, and throughput. Success means retaining most of the ridgeless mean performance while reducing catastrophic-loss frequency and avoiding a permanently excessive ridge coefficient.
Verification
Stage 1 · Toy mechanism gate: Failed ✗
Stage 2 · Mechanism transferred to benchmark: Not tested
Stage 2 · Practical benchmark result: Not run
Methodology: Toy-system gate first; the benchmark stage runs only after a pass. How verification works
Stage 1 — Mechanism check agent confidence 9/10
Built the spectral inverse-trace/hard-edge diagnostics, closed-form ridge head, adaptive controller, and fixed-seed synthetic regression experiment. The decomposition and ridge normal equations were verified to numerical precision, but the promised performance effect was not observed: adaptive ridge saturated at gamma=0.05 and had substantially worse hard-edge MSE (0.0160) and p99 loss (0.1022) than fixed ridge (0.0128, 0.0831) and ridgeless regression (0.0135, 0.0875).
- Agent confidence
- 9/10
- Baseline
- Ridgeless hard-edge MSE 0.01350, p99 0.08750; fixed ridge hard-edge MSE 0.01281, p99 0.08314.
- Idea
- Adaptive hard-edge MSE 0.01602, p99 0.10216, median gamma 0.05 at the upper clip; healthy MSE 0.01338 and p99 0.08562. Math split absolute error 2.27e-13 and ridge normal-equation residual 9.15e-17.
Limitations: This is a small synthetic linear-head test, not a neural backbone or CIFAR-10 experiment. It uses one-batch controller updates and does not test multi-step EMA dynamics, corruption distributions beyond the constructed hard-edge covariance, throughput, or end-to-end AdamW training.
How to run: python3 adaptive_ridge_experiment.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Artifacts
- adaptive_ridge_experiment.py 4.4 KB View Raw
- report.md 1.4 KB View
- results.json 1.7 KB View Raw