Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Challenges in Training PINNs: A Loss Landscape Perspective

About

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.

Pratik Rathore, Weimu Lei, Zachary Frangella, Lu Lu, Madeleine Udell• 2024

Related benchmarks

TaskDatasetResultRank
PDE solvingHelmholtz 1d (test)
Relative MSE0.847
15
PDE solvingPoisson 1d (test)
Relative MSE0.118
15
PDE solvingDarcy-Flow 2d (test)
Relative MSE0.838
15
PDE solvingNLRD 1d+time (test)
Relative MSE0.757
15
PDE solvingHeat 2d+time (test)
Rel. MSE0.61
9
Showing 5 of 5 rows

Other info

Follow for update