Loss Landscape Features That Make Adam Stall: Definitions, Estimators, and the Preconditioned Hessian View
Organizations: Texas State University
Abstract
Across implicit-neural-representation (INR) architectures and analytic benchmarks we observe that a thoroughly tuned Adam (especially its learning rate (lr), e.g. in a hyperparameter sweep from to ) can potentially reach a very low loss even on ill-conditioned loss landscape or converge at a plateau far above the loss attained by second-order methods. This report defines the measured metrics that help determine if Adam can mitigate the ill-conditioning on a given loss landscape. We provide the indicators by which each outcome is determined, that are: the condition number of the Hessian and of the Adam-preconditioned Hessian (with the derivation from Adam's update rule), the diagonal mass that distinguishes axis-aligned from cross-coupled ill-conditioning, the negative spectral mass estimated by stochastic Lanczos quadrature, and the gradient energy fractions over curvature bands, including the flat fraction that indicates the Adam stall. A worked out example and an illustration show the reasons why a diagonal preconditioning by Adam can remove axis-aligned ill-conditioning by rescaling and why it cannot do the same if the ill-conditioning is cross coupled. In addition, we present a case study of FINER image fitting architecture that goes over the whole loss landscape analysis framework: the fitting architecture description, reasons due to which its landscape stalls Adam at saddles, the measured PSNR values through our tuned baselines to the --,dB results of the blockwise second order methods, the error maps behind those numbers, and description of the benefits such image fitting accuracy gives in practice.
Figures & tables
| Benchmark | reduction | neg. mass | flat frac | outcome | |||
|---|---|---|---|---|---|---|---|
| 35k multi-freq. | 4.9e2 | 3.1e1 | 16 | .04 | .92 | .000 | Adam at precision floor |
| SIREN 2D | 1.6e3 | 9.9e1 | 16 | .09 | .09 | .000 a | comparable |
| WIRE 2D | 1.2e5 | 8.4e2 | 143 | .65 | .00 | .000 | comparable |
| FINER 2D | 5.4e4 | 6.1e1 | 887 | .98 | .19 b | .000 | ARC dB |
| SDF-SIREN | 8.9e3 | 2.4e1 | 372 | .06 | .67 | .000 c | comparable |
| SDF-FINER | 1.0e3 | 6.9 | 146 | .58 | .68 | .001 | Adam error 5.5 |