A failure-informed multi-stage training algorithm for three-component nonlinear Schrödinger equation.

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. • We explore a deep learning method for solving three-component nonlinear Schrödinger equation. • We proposed an improved PINNS by time-adaptive strategy, failure-informed enrichment technique, and pre-fixed technique. • The data-driven double-valley dark soliton solutions are derived by the proposed method. • The proposed method can be easily extended to the physics-informed DeepONets. [ABSTRACT FROM AUTHOR]