SS-ESOAP: Self-Scaled Adaptive Preconditioning for Physics-Informed Learning
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
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: Local-Consistency Optimisation for Bayesian Neural Networks
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks