A Failure-Informed Multi-Stage Training Algorithm for Three-Component Nonlinear Schrödinger Equation

Yawen Wu,Liming Ling,Yubin Huang
DOI: https://doi.org/10.1016/j.camwa.2024.06.012
IF: 3.218
2024-01-01
Computers & Mathematics with Applications
Abstract:In this paper, we propose a failure-informed multi-stage training algorithm for approximating the double-valley dark soliton solutions of the three-component nonlinear Schrödinger equation (NLSE). Although the physics-informed neural network (PINN) method has achieved remarkable results in many equations, it is ineffective when directly applied to the considered equation due to the non-vanishing background of dark solitons and the complexity of the multi-valley structure. To improve the approximation ability of the PINN, the algorithm we proposed generates effective samples adaptively based on Gaussian mixture distribution by combining the time-adaptive strategy, failure-informed enrichment technique, and pre-fixed technique. Furthermore, the algorithm is extended to learn the solution operator of three-component NLSE within a specialized solution space. The results of numerical simulation demonstrate that the improved algorithm has better approximation ability and faster convergence rate in approximating the solution and solution operator of the considered equation while using a smaller training set.
What problem does this paper attempt to address?