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

2021/07/19 by Wang, Yiran, Li, Zhen · 1 citation
#FOS: Computer and information sciences #FOS: Mathematics #Machine Learning (cs.LG) #Numerical Analysis (math.NA)

paper · doi:10.48550/arxiv.2107.08593

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.

Cited by

• Machine learning approach to single-shot multiparameter estimation for the non-linear Schrödinger equation

Related

• A clever neural network in solving inverse problems of Schrödinger equation
• Convolutional neural networks in phase space and inverse problems
• Physics-oriented learning of nonlinear Schrödinger equation: optical fiber loss and dispersion profile identification
• Machine learning approach to single-shot multiparameter estimation for the non-linear Schrödinger equation
• Physics-Based Deep Learning for Fiber-Optic Communication Systems