Optimizing the Optimizer for Physics-Informed Neural Networks and Kolmogorov-Arnold Networks

TL;DR

This paper explores advanced quasi-Newton optimization methods, such as SSBFGS and SSBroyden, to improve the training efficiency and accuracy of Physics-Informed Neural Networks and Kolmogorov-Arnold Networks for solving complex PDEs.

Contribution

It introduces and systematically evaluates self-scaled quasi-Newton optimizers, demonstrating significant accuracy improvements over traditional methods in SciML applications.

Findings

Enhanced training efficiency with self-scaled quasi-Newton methods.

Achieved state-of-the-art accuracy on multiple PDE benchmarks.

Demonstrated effectiveness for Deep Operator Network architectures.

Abstract

Physics-Informed Neural Networks (PINNs) have revolutionized the computation of PDE solutions by integrating partial differential equations (PDEs) into the neural network's training process as soft constraints, becoming an important component of the scientific machine learning (SciML) ecosystem. More recently, physics-informed Kolmogorv-Arnold networks (PIKANs) have also shown to be effective and comparable in accuracy with PINNs. In their current implementation, both PINNs and PIKANs are mainly optimized using first-order methods like Adam, as well as quasi-Newton methods such as BFGS and its low-memory variant, L-BFGS. However, these optimizers often struggle with highly non-linear and non-convex loss landscapes, leading to challenges such as slow convergence, local minima entrapment, and (non)degenerate saddle points. In this study, we investigate the performance of Self-Scaled BFGS (SSBFGS), Self-Scaled Broyden (SSBroyden) methods and other advanced quasi-Newton schemes, including BFGS and L-BFGS with different line search strategies. These methods dynamically rescale updates based on historical gradient information, thus enhancing training efficiency and accuracy. We systematically compare these optimizers using both PINNs and PIKANs on key challenging PDEs, including the Burgers, Allen-Cahn, Kuramoto-Sivashinsky, Ginzburg-Landau, and Stokes equations. Additionally, we evaluate the performance of SSBFGS and SSBroyden for Deep Operator Network (DeepONet) architectures, demonstrating their effectiveness for data-driven operator learning. Our findings provide state-of-the-art results with orders-of-magnitude accuracy improvements without the use of adaptive weights or any other enhancements typically employed in PINNs.