Challenges in Training PINNs: A Loss Landscape Perspective
Pratik Rathore, Weimu Lei, Zachary Frangella, Lu Lu, Madeleine Udell
Abstract
This paper explores challenges in training Physics-Informed Neural Networks (PINNs), emphasizing the role of the loss landscape in the training process. We examine difficulties in minimizing the PINN loss function, particularly due to ill-conditioning caused by differential operators in the residual term. We compare gradient-based optimizers Adam, L-BFGS, and their combination Adam+L-BFGS, showing the superiority of Adam+L-BFGS, and introduce a novel second-order optimizer, NysNewton-CG (NNCG), which significantly improves PINN performance. Theoretically, our work elucidates the connection between ill-conditioned differential operators and ill-conditioning in the PINN loss and shows the benefits of combining first- and second-order optimization methods. Our work presents valuable insights and more powerful optimization strategies for training PINNs, which could improve the utility of PINNs for solving difficult partial differential equations.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers29
- Gradient Alignment in Physics-informed Neural Networks: A Second-Order Optimization PerspectiveSifan Wang, Ananyae Kumar Bhartari, Bowen Li, Paris PerdikarisNeurIPS 2025 · 100 citations
- RoPINN: Region Optimized Physics-Informed Neural NetworksHaixu Wu, Huakun Luo, Yuezhou Ma, Jianmin Wang et al.NeurIPS 2024 · 50 citations
- FP64 is All You Need: Rethinking Failure Modes in Physics-Informed Neural NetworksChenhui Xu, Dancheng Liu, Amir Nassereldine, Jinjun XiongNeurIPS 2025 · 18 citations
- Improving Energy Natural Gradient Descent through Woodbury, Momentum, and RandomizationAndrés Guzmán-Cordero, Felix Dangel, Gil Goldshlager, Marius ZeinhoferNeurIPS 2025 · 17 citations
- A Regularized Newton Method for Nonconvex Optimization with Global and Local Complexity GuaranteesYuhao Zhou, Jintao Xu, Bingrui Li, Chenglong Bao et al.NeurIPS 2025 · 8 citations
Builds on9
- Fourier Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola Borislavov Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.ICLR 2021 · 3,911 citations
- Characterizing possible failure modes in physics-informed neural networksAditi S. Krishnapriyan, Amir Gholami, Shandian Zhe, Robert M. Kirby et al.NeurIPS 2021 · 1,421 citations
- When Do Neural Networks Outperform Kernel Methods?Behrooz Ghorbani, Song Mei, Theodor Misiakiewicz, Andrea MontanariNeurIPS 2020 · 217 citations
- On the linearity of large non-linear models: when and why the tangent kernel is constantChaoyue Liu, Libin Zhu, Mikhail BelkinNeurIPS 2020 · 183 citations
- Aiming towards the minimizers: fast convergence of SGD for overparametrized problemsChaoyue Liu, Dmitriy Drusvyatskiy, Mikhail Belkin, Damek Davis et al.NeurIPS 2023 · 31 citations
Related papers
- MultiAdam: Parameter-wise Scale-invariant Optimizer for Multiscale Training of Physics-informed Neural NetworksJiachen Yao, Chang Su, Zhongkai Hao, Songming Liu et al.ICML 2023 · 27 citations
- An operator preconditioning perspective on training in physics-informed machine learningTim De Ryck, Florent Bonnet, Siddhartha Mishra, Emmanuel de BézenacICLR 2024 · 28 citations
- Kronecker-Factored Approximate Curvature for Physics-Informed Neural NetworksFelix Dangel, Johannes Müller, Marius ZeinhoferNeurIPS 2024 · 31 citations
- Mitigating Gradient Pathology in PINNs through Aligned ConstraintYichen Luo, Peiyu Zhu, Dongxiao Hu, Jia Wang et al.ICML 2026
- Gradient Descent Finds the Global Optima of Two-Layer Physics-Informed Neural NetworksYihang Gao, Yiqi Gu, Michael NgICML 2023 · 12 citations
