RAR-PINN algorithm for the data-driven vector-soliton solutions and parameter discovery of coupled nonlinear equations

Abstract

This work aims to provide an effective deep learning framework to predict the vector-soliton solutions of the coupled nonlinear equations and their interactions. The method we propose here is a physics-informed neural network (PINN) combining with the residual-based adaptive refinement (RAR-PINN) algorithm. Different from the traditional PINN algorithm which takes points randomly, the RAR-PINN algorithm uses an adaptive point-fetching approach to improve the training efficiency for the solutions with steep gradients. A series of experiment comparisons between the RAR-PINN and traditional PINN algorithms are implemented to a coupled generalized nonlinear Schrödinger (CGNLS) equation as an example. The results indicate that the RAR-PINN algorithm has faster convergence rate and better approximation ability, especially in modeling the shape-changing vector-soliton interactions in the coupled systems. Finally, the RAR-PINN method is applied to perform the data-driven discovery of the CGNLS equation, which shows the dispersion and nonlinear coefficients can be well approximated.

References in corpus (11)

• Adam: A Method for Stochastic Optimization
• TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems
• DeepXDE: A deep learning library for solving differential equations
• Hidden Physics Models: Machine Learning of Nonlinear Partial Differential Equations
• Optical Confinement of a Bose-Einstein Condensate
• Gradient-enhanced physics-informed neural networks for forward and inverse PDE problems
• Machine Learning of Linear Differential Equations using Gaussian Processes
• SciANN: A Keras/Tensorflow wrapper for scientific computations and physics-informed deep learning using artificial neural networks
• Bright-dark solitons and their collisions in mixed N-coupled nonlinear Schrödinger equations
• Physics-informed neural networks method in high-dimensional integrable systems
• Bright solitons in a spin-tensor-momentum-coupled Bose-Einstein condensate