NSGA-PINN: A Multi-Objective Optimization Method for Physics-Informed Neural Network Training

Binghang Lu,Christian B. Moya,Guang Lin
2023-03-07
Abstract:This paper presents NSGA-PINN, a multi-objective optimization framework for effective training of Physics-Informed Neural Networks (PINNs). The proposed framework uses the Non-dominated Sorting Genetic Algorithm (NSGA-II) to enable traditional stochastic gradient optimization algorithms (e.g., ADAM) to escape local minima effectively. Additionally, the NSGA-II algorithm enables satisfying the initial and boundary conditions encoded into the loss function during physics-informed training precisely. We demonstrate the effectiveness of our framework by applying NSGA-PINN to several ordinary and partial differential equation problems. In particular, we show that the proposed framework can handle challenging inverse problems with noisy data.
Machine Learning
What problem does this paper attempt to address?
The paper aims to address the issues encountered during the training of Physics-Informed Neural Networks (PINNs), particularly how to effectively optimize highly non-convex loss functions to avoid getting trapped in local minima and accurately satisfy initial and boundary conditions. The authors propose a multi-objective optimization framework called NSGA-PINN, which combines the Non-dominated Sorting Genetic Algorithm II (NSGA-II) and traditional stochastic gradient descent algorithms (such as Adam) to achieve these goals. Specifically, the main contributions of the NSGA-PINN method are as follows: 1. **Escaping Local Minima**: By utilizing the NSGA-II algorithm, the traditional stochastic gradient descent algorithm can more effectively escape local minima. 2. **Accurately Satisfying Constraints**: The NSGA-II algorithm ensures that the initial and boundary conditions encoded in the loss function are accurately satisfied during training. 3. **Solving Inverse Problems and Handling Noisy Data**: The paper demonstrates that the NSGA-PINN framework can effectively handle inverse problems that include noisy data. To validate the effectiveness of the proposed framework, the authors conducted experiments on several Ordinary Differential Equation (ODE) and Partial Differential Equation (PDE) problems, particularly for inverse problems with noisy data. The results show that the NSGA-PINN algorithm can converge to the optimal solution while exhibiting good generalization ability and robustness to noise.