Principle-Driven Fiber Transmission Model Based on PINN Neural Network

In this paper, a novel principle-driven fiber transmission model based on physical induced neural network (PINN) is proposed. Unlike data-driven models which regard fiber transmission problem as data regression tasks, this model views it as an equation solving problem. Instead of adopting input signals and output signals which are calculated by SSFM algorithm in advance before training, this principle-driven PINN based fiber model adopts frames of time and distance as its inputs and the corresponding real and imaginary parts of NLSE solutions as its outputs. By taking into account of pulses and signals before transmission as initial conditions and fiber physical principles as NLSE in the design of loss functions, this model will progressively learn the transmission rules. Therefore, it can be effectively trained without the data labels, referred as the pre-calculated signals after transmission in data-driven models. Due to this advantage, SSFM algorithm is no longer needed before the training of principle-driven fiber model which can save considerable time consumption. Through numerical demonstration, the results show that this principle-driven PINN based fiber model can handle the prediction tasks of pulse evolution, signal transmission and fiber birefringence for different transmission parameters of fiber telecommunications.

[1]  Shilin Xiao,et al.  Fast and Accurate Optical Fiber Channel Modeling Using Generative Adversarial Network , 2020, Journal of Lightwave Technology.

[2]  Xue Chen,et al.  Data-driven Optical Fiber Channel Modeling: A Deep Learning Approach , 2020, Journal of Lightwave Technology.

[3]  Léon Bottou,et al.  Stochastic Gradient Descent Tricks , 2012, Neural Networks: Tricks of the Trade.

[4]  Alexander J. Smola,et al.  Parallelized Stochastic Gradient Descent , 2010, NIPS.

[5]  Dimitrios I. Fotiadis,et al.  Artificial neural networks for solving ordinary and partial differential equations , 1997, IEEE Trans. Neural Networks.

[6]  Elsayed Elsayed Azzouz,et al.  Modulation recognition using artificial neural networks , 1997, Signal Process..

[7]  Reza Nejabati,et al.  Field trial of Machine-Learning-assisted and SDN-based Optical Network Planning with Network-Scale Monitoring Database , 2017, 2017 European Conference on Optical Communication (ECOC).

[8]  Jorge Nocedal,et al.  On the limited memory BFGS method for large scale optimization , 1989, Math. Program..

[9]  Matthias Hein,et al.  Variants of RMSProp and Adagrad with Logarithmic Regret Bounds , 2017, ICML.

[10]  Polina Bayvel,et al.  End-to-End Deep Learning of Optical Fiber Communications , 2018, Journal of Lightwave Technology.

[11]  B. Herbst,et al.  Split-step methods for the solution of the nonlinear Schro¨dinger equation , 1986 .

[12]  J. van Leeuwen,et al.  Neural Networks: Tricks of the Trade , 2002, Lecture Notes in Computer Science.

[13]  Govind P. Agrawal,et al.  Nonlinear Fiber Optics , 1989 .

[14]  Zhenming Yu,et al.  Universal Fiber Models based on PINN Neural Network , 2020, 2020 Asia Communications and Photonics Conference (ACP) and International Conference on Information Photonics and Optical Communications (IPOC).

[15]  Simon Günter,et al.  A Stochastic Quasi-Newton Method for Online Convex Optimization , 2007, AISTATS.

[16]  Chao Lu,et al.  An Optical Communication's Perspective on Machine Learning and Its Applications , 2019, Journal of Lightwave Technology.

[17]  C. Menyuk,et al.  Optimization of the split-step Fourier method in modeling optical-fiber communications systems , 2003 .

[18]  A.E. Willner,et al.  Optical Performance Monitoring Using Artificial Neural Networks Trained With Eye-Diagram Parameters , 2009, IEEE Photonics Technology Letters.

[19]  Paris Perdikaris,et al.  Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations , 2019, J. Comput. Phys..

[20]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[21]  Michael I. Jordan,et al.  A Linearly-Convergent Stochastic L-BFGS Algorithm , 2015, AISTATS.