Bennett-whitened gradient trust region
Implementation & benchmark of arXiv:2608.15874 — Self-normalised Bennett inequalities for Hilbert-valued martingales
Source paper: Self-normalised Bennett inequalities for Hilbert-valued martingales arXiv:2608.15874 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Use the paper's self-normalized martingale bound to monitor cumulative stochastic gradient noise in covariance-whitened coordinates. Convert its time-uniform confidence boundary into a trust-region multiplier: retain the normal optimizer update while the observed noise is within the boundary, and shrink or clip the update after an exceedance.
Formulas
Mathematical statement
For a martingale M_n=sum_{i=1}^n X_i with conditionally mean-zero increments X_i, increment norm bounded by one, and predictable covariance V_n=sum_{i=1}^n E[X_i X_i^T | F_{i-1}], the paper constructs the nonnegative supermartingale S_n=exp{rho h(||(V_n+rho I)^(-1/2)M_n||/sqrt(rho))-(1/2)log det(I+rho^(-1)V_n)}. Here rho>0 is a ridge parameter, I is the identity, det is the determinant, and h(u)=(1+u)log(1+u)-u is the Bennett rate function. Ville's inequality gives, with probability at least 1-delta simultaneously for every n, rho h(z_n/sqrt(rho))-(1/2)log det(I+rho^(-1)V_n)<=log(1/delta), where z_n=||(V_n+rho I)^(-1/2)M_n||. Define r_n as the largest solution of rho h(r_n/sqrt(rho))-(1/2)log det(I+rho^(-1)V_n)=log(1/delta). In training, X_i is clipped minibatch gradient noise, M_n is its cumulative sum, and V_n is estimated predictably from previous minibatches. The exact theorem uses the true covariance; the MVP uses a diagonal covariance estimate.
Implementation notes
(1) Integrate this controller into the optimizer immediately after computing a minibatch gradient g_t and before the parameter update. Maintain a gradient mean estimate mu_t and define the centered noise proxy x_t=clip(g_t-mu_{t-1},1), using global-norm or layerwise clipping so the theorem's bounded-increment assumption is approximately enforced. Maintain cumulative noise M_t=M_{t-1}+x_t. For the first MVP use diagonal covariance: estimate q_t from the previous minibatch, for example q_t=(g_{t-1}-mu_{t-2}) squared elementwise, and maintain V_t=V_{t-1}+q_t. The one-step delay makes V_t predictable before x_t is added. (2) Pseudocode: initialize M=0, V=0, mu=0; compute g; compute x=clip(g-mu,1); update M=M+x; update V using the delayed previous squared residual; calculate z=sqrt(sum_j M_j^2/(V_j+rho)); calculate D=0.5*sum_j log(1+V_j/rho); solve rho*((1+r/sqrt(rho))*log(1+r/sqrt(rho))-r/sqrt(rho))-D=log(1/delta) for r by bisection; multiply the base SGD, Adam, or AdamW step by min(1,r/max(z,epsilon)). Update mu after the covariance proxy is formed. (3) The Bennett function, determinant correction, whitening, and root equation are mathematical components transplanted from the paper. Diagonal covariance estimation, clipping threshold, delta, rho, and any multiplier floor are empirical approximations. (4) First experiment: compare SGD, AdamW, gradient-norm clipping, and this controller on a small MLP and a ViT-Tiny trained on CIFAR-10, using identical batches, learning-rate schedules, and compute. Log z_t/r_t, the fraction of downscaled steps, gradient norm, loss variance, divergence events, time to a fixed validation loss, and final accuracy. The method succeeds if it reduces instability and loss variance while matching or improving accuracy; a concrete target is at least 20 percent lower loss variance or fewer divergence events at equal training FLOPs.
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 a diagonal, delayed-covariance Bennett-whitened trust-region controller and numerically verified the root equation to machine precision. In 400 bounded-noise paths, only 0.25% crossed the boundary, so the controller downscaled just 0.13% of training steps; matched toy regression showed nearly identical but slightly worse final loss (0.1735 vs 0.1723) and loss variance (0.001890 vs 0.001856), with no stability win.
- Agent confidence
- 9/10
- Baseline
- SGD: final loss 0.172262, best loss 0.082479, last-50 loss variance 0.001856, 768 steps, no divergence
- Idea
- Bennett controller: final loss 0.173519, best loss 0.083908, last-50 loss variance 0.001890, 768 steps, 0.13% downscaled steps, median z/r 0.00382, no divergence; math root error 0 and boundary-crossing path fraction 0.0025
Limitations: Only a small synthetic regression MLP was tested; no CIFAR-10, ViT, AdamW, gradient-clipping control, or multi-seed evaluation was run. The covariance is a diagonal one-step-delayed proxy rather than the theorem's true predictable covariance, and CUDA runtime fallback was not stress-tested.
How to run: python3 experiment.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Artifacts
- experiment.py 5.9 KB View Raw
- report.md 1.3 KB View
- results.json 0.8 KB View Raw