SS-ESOAP: Self-Scaled Adaptive Preconditioning for Physics-Informed Learning

arXiv:2608.29448 · cs.LG, cs.AI, math.OC, stat.ML · Submitted 2026-08-29 · Read on arXiv

cs.LG, cs.AI, math.OC, stat.ML

Submitted: 2026-08-29

Updated: 2026-08-29

Comments: 31 pages, 12 figures, 16 tables. Submitted to NeurIPS 2026

License: http://creativecommons.org/licenses/by-sa/4.0/

The gist: Physics-informed neural networks (PINNs) often face ill-conditioned objectives that limit high-accuracy training.

Terminology

Abstract

Physics-informed neural networks (PINNs) often face ill-conditioned objectives that limit high-accuracy training. Dense quasi-Newton methods improve local conditioning but require expensive optimizer state, while Kronecker-factored methods such as SOAP scale to larger networks but rely on periodic basis updates. We introduce, which augments SOAP-style preconditioning with a scalar secant-energy correction adapted to Kronecker geometry and an adaptive basis update followed by variance-state downscaling. We characterize the directional secant matching induced by the scalar correction and give a bound on variance-state mismatch across basis changes. Across eight PDE benchmarks, attains the lowest final residual on six, including Burgers and Boussinesq, while SOAP-family baselines perform better on Gray-Scott and Ginzburg-Landau. On Boussinesq, reaches a residual of 10-5 in 4.1 hours with 9.2 GB peak VRAM, while Adam does not reach this target within 14 hours. Three-seed L squared and H 1 errors on four representative PDEs support the link between lower residuals and improved solution accuracy. These results position as a scalable option for stiff, high-accuracy physics-informed training, rather than a uniform replacement for existing optimizers.

Related papers