Which Optimizer Works Best for Physics-Informed Neural Networks and Kolmogorov-Arnold Networks?
| Title: | Which Optimizer Works Best for Physics-Informed Neural Networks and Kolmogorov-Arnold Networks? |
|---|---|
| Authors: | Kiyani, Elham, Shukla, Khemraj, Urbán, Jorge F., Darbon, Jérôme, Karniadakis, George Em |
| Publication Year: | 2025 |
| Collection: | Computer Science Mathematics |
| Subject Terms: | Computer Science - Machine Learning, Computer Science - Artificial Intelligence, Mathematics - Optimization and Control |
| More Details: | 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. In its current implementation, PINNs 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 Broyden (SSBroyden) methods and other advanced quasi-Newton schemes, including BFGS and L-BFGS with different line search strategies approaches. These methods dynamically rescale updates based on historical gradient information, thus enhancing training efficiency and accuracy. We systematically compare these optimizers on key challenging linear, stiff, multi-scale and non-linear PDEs benchmarks, including the Burgers, Allen-Cahn, Kuramoto-Sivashinsky, and Ginzburg-Landau equations, and extend our study to Physics-Informed Kolmogorov-Arnold Networks (PIKANs) representation. Our findings provide insights into the effectiveness of second-order optimization strategies in improving the convergence and accurate generalization of PINNs for complex PDEs by orders of magnitude compared to the state-of-the-art. Comment: 33 pages, 27 figures |
| Document Type: | Working Paper |
| Access URL: | http://arxiv.org/abs/2501.16371 |
| Accession Number: | edsarx.2501.16371 |
| Database: | arXiv |
| Description not available. |