AMStraMGRAM: Adaptive Multi-cutoff Strategy Modification for ANaGRAM
Nilo Schwencke, Cyriaque Rousselot, Alena Shilova, Cyril Furtlehner
TL;DR
The paper tackles the challenge of efficiently training physics-informed neural networks (PINNs) with natural-gradient-inspired updates. It analyzes ANaGRAM, revealing a flattening phenomenon tied to singular-value cutoffs and introduces AMStraMGRAM, an adaptive multi-cutoff strategy that leverages reconstruction-error–spectrum interactions to reach target precision efficiently. The authors provide a spectral-theoretic framework linking cutoff regularization to generalized Green's functions, and demonstrate substantial performance gains on benchmark PDEs while also noting overfitting risks tied to sampling. This work advances mesh-free PINN optimization by coupling adaptive spectral regularization with a geometric view of the empirical tangent space, offering practical improvements and theoretical grounding. Future directions include convergence guarantees, residual-based stabilization, and co-design of sampling strategies to further boost robustness and accuracy.
Abstract
Recent works have shown that natural gradient methods can significantly outperform standard optimizers when training physics-informed neural networks (PINNs). In this paper, we analyze the training dynamics of PINNs optimized with ANaGRAM, a natural-gradient-inspired approach employing singular value decomposition with cutoff regularization. Building on this analysis, we propose a multi-cutoff adaptation strategy that further enhances ANaGRAM's performance. Experiments on benchmark PDEs validate the effectiveness of our method, which allows to reach machine precision on some experiments. To provide theoretical grounding, we develop a framework based on spectral theory that explains the necessity of regularization and extend previous shown connections with Green's functions theory.
