Inverse Problem of Nonlinear Schrödinger Equation as Learning of Convolutional Neural Network

Abstract

In this work, we use an explainable convolutional neural network (NLS-Net) to solve an inverse problem of the nonlinear Schrödinger equation, which is widely used in fiber-optic communications. The landscape and minimizers of the non-convex loss function of the learning problem are studied empirically. It provides a guidance for choosing hyper-parameters of the method. The estimation error of the optimal solution is discussed in terms of expressive power of the NLS-Net and data. Besides, we compare the performance of several training algorithms that are popular in deep learning. It is shown that one can obtain a relatively accurate estimate of the considered parameters using the proposed method. The study provides a natural framework of solving inverse problems of nonlinear partial differential equations with deep learning.

arXiv comment25 pages

References in corpus (11)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• ADADELTA: An Adaptive Learning Rate Method
• Deep Convolutional Neural Network for Inverse Problems in Imaging
• A Convergence Theory for Deep Learning via Over-Parameterization
• Mathematical theory and numerical methods for Bose-Einstein condensation
• Fermionic neural-network states for ab-initio electronic structure
• Solving Many-Electron Schrödinger Equation Using Deep Neural Networks
• A mesh-free method for interface problems using the deep learning approach
• DNN Expression Rate Analysis of High-dimensional PDEs: Application to Option Pricing
• Learning to solve inverse problems using Wasserstein loss
• A Flow Model of Neural Networks