✗ Mechanism failed 2026

Lanczos-triggered low-rank Newton Adam

Implementation & benchmark of arXiv:2608.22145 — Loss Landscape Features That Make Adam Stall: Definitions, Estimators, and the Preconditioned Hessian View

Usefulness7/10
Difficulty6/10
Novelty6/10

Source paper: Loss Landscape Features That Make Adam Stall: Definitions, Estimators, and the Preconditioned Hessian View arXiv:2608.22145 ⓘ · analyzed Aug 29, 2026

AI-generated research hypothesis, automatically tested. Not peer-reviewed.

Idea description

Use Adam normally, but periodically estimate the spectrum of the Adam-preconditioned Hessian and add a damped low-rank Newton correction when the preconditioned curvature is strongly ill-conditioned or the gradient is concentrated in flat directions. The correction is computed only in a small Lanczos subspace, so the method targets cross-coupled ill-conditioning without materializing or inverting the full Hessian.

Formulas

$$H=\sum_{j=1}^{n}\lambda_j u_j u_j^{\top},\qquad \lambda_1\geq\lambda_2\geq\dots\geq\lambda_n,\qquad u_j^{\top}u_i=\delta_{ij}.$$
$$B=D^{-1/2}HD^{-1/2},\qquad D=\operatorname{diag}(v_t)+\epsilon I.$$
$$F_{\mathrm{flat}}=\frac{\sum_{i:\,|\mu_i|\leq\tau}(q_i^{\top}D^{-1/2}g)^2}{\|D^{-1/2}g\|_2^2},\qquad \widehat\kappa=\frac{\max_{i:\,\mu_i>\tau}\mu_i}{\max\left(\min_{i:\,\mu_i>\tau}\mu_i,\tau\right)}.$$
$$\Delta\theta=-\eta\frac{m_t}{\sqrt{v_t}+\epsilon}-\alpha D^{-1/2}Q_r(\widehat\Lambda_r+\gamma I)^{-1}Q_r^{\top}D^{-1/2}g.$$

Mathematical statement

The paper writes the Hessian as $H=\sum_{j=1}^{n}\lambda_j u_j u_j^{\top}$, where $H=\nabla^2 f(\theta)$ is the Hessian of loss $f$, $\lambda_j$ are ordered eigenvalues, and $u_j$ are orthonormal eigenvectors. Adam's diagonal second-moment preconditioner is $D=\operatorname{diag}(v_t)+\epsilon I$, with $v_t$ the exponentially averaged squared gradient and $\epsilon>0$; in Adam-scaled coordinates the relevant curvature is $B=D^{-1/2}HD^{-1/2}$. Let $(\mu_i,q_i)$ be eigenpairs of $B$, let $g=\nabla f(\theta)$, and define transformed gradient coefficients $a_i=q_i^{\top}D^{-1/2}g$. The gradient energy in a curvature band $S$ is $E_S=\sum_{i\in S}a_i^2$, with total energy $E=\|D^{-1/2}g\|_2^2$. A flat fraction is $F_{\mathrm{flat}}=E_{\{i:|\mu_i|\leq\tau\}}/E$, and a positive-curvature condition estimate is $\widehat\kappa=\max_{\mu_i>\tau}\mu_i/\max(\min_{\mu_i>\tau}\mu_i,\tau)$. The adapted update uses the damped inverse of $B$ in the estimated Lanczos subspace: $\Delta\theta_{\mathrm{LR}}=-D^{-1/2}Q_r(\widehat\Lambda_r+\gamma I)^{-1}Q_r^{\top}D^{-1/2}g$, where $Q_r=[q_1,\ldots,q_r]$, $\widehat\Lambda_r=\operatorname{diag}(\mu_1,\ldots,\mu_r)$, and $\gamma>0$ is damping. Negative eigenvalues are discarded or replaced by zero before inversion to avoid unstable saddle directions. The paper's mechanism is that diagonal scaling removes axis-aligned ill-conditioning but not cross-coupled ill-conditioning, which appears as a large condition number of $B$.

Implementation notes

Integrate this into the optimizer update, not into the network architecture. Keep Adam's usual first- and second-moment updates, and every K optimizer steps—start with K=100—run r-step Lanczos using Hessian-vector products from the current minibatch loss. First transform each HVP as $x\mapsto D^{-1/2}H(D^{-1/2}x)$, where $D=\operatorname{diag}(v_t)+\epsilon I$. Lanczos returns approximate eigenvectors $Q_r$ and eigenvalues $\mu_i$ of the symmetric operator $B$. Compute $a_i=q_i^\top D^{-1/2}g$, the total transformed gradient energy $E=\|D^{-1/2}g\|^2$, the flat fraction using threshold $\tau=10^{-3}$ times the running median of positive $|\mu_i|$, and the estimated condition number from the smallest and largest positive Ritz values. Pseudocode: (1) update Adam moments; (2) if step mod K is nonzero, apply the ordinary Adam step; (3) otherwise call Lanczos(B,r); (4) compute $F_{\mathrm{flat}}$ and $\widehat\kappa$; (5) if $F_{\mathrm{flat}}>0.3$ or $\widehat\kappa>100$, form the low-rank correction with $\gamma=0.1\max(\mu_i,\tau)$ and blend it with Adam using $\alpha\in[0.1,1]$; (6) clip the combined step to the Adam step norm times 2. Use only positive Ritz values in the inverse, and fall back to Adam if Lanczos fails or the predicted quadratic decrease is nonpositive. The mathematical quantities are estimated from HVPs; no full Hessian is constructed. Begin with a 2-layer SIREN or ReLU MLP fitting MNIST or a single $64\times64$ image, comparing Adam, Adam plus untriggered low-rank correction, and the triggered method at matched HVP/FLOP budgets. Log loss, gradient energy fractions, condition estimates, and number of corrections. Success means faster loss descent or lower final fitting error than Adam at equal wall-clock cost, especially on runs where Adam exhibits a plateau; a negative result is no improvement after accounting for the Lanczos overhead.

Verification

Mechanism failed

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 8/10

Built a matrix-free Lanczos low-rank Newton correction integrated with Adam, including damping, positive-curvature filtering, trigger metrics, correction clipping, and numerical verification. The algebra check passed with spectrum error 2.8e-14, orthogonality error 1.4e-15, and update-equivalence error 4.4e-16. On the rotated ill-conditioned quadratic, low-rank Adam reduced equal-step loss versus Adam, but cost about 2.1x more wall time and the triggered method was essentially no better than the always-correcting variant, so there is no demonstrated cost-adjusted win.

Agent confidence
8/10
Baseline
Adam: final loss 2.952, step-100 loss 45.83, step-200 loss 10.89, 0 HVPs, 0.124 s
Idea
Triggered low-rank Adam: final loss 2.166, step-100 loss 25.79, step-200 loss 7.158, 20 corrections, 160 HVPs, 0.263 s; untriggered final loss 2.162 in 0.269 s

Limitations: Only a 24-dimensional deterministic rotated quadratic was tested, not an MLP, image, MNIST, or minibatch neural-network training. The benchmark used equal optimizer steps rather than a strict equal-wall-clock or equal-FLOP comparison, and only one seed and one hyperparameter setting were evaluated.

How to run: python3 run_experiment.py

Verdict computed by deterministic test code from paired-seed statistics — not by the language model.

Artifacts

Implementation overview ⬇ Download all as ZIP 3 files · code, reports and structured results