Modified failproof physics-informed neural network framework for fast and accurate optical fiber transmission link modeling

Abstract

Physics-informed neural networks (PINNs) have recently emerged as an important and ground-breaking technique in scientific machine learning for numerous applications including in optical fiber communications. However, the vanilla/baseline version of PINNs is prone to fail under certain conditions because of the nature of the physics-based regularization term in its loss function. The use of this unique regularization technique results in a highly complex non-convex loss landscape when visualized. This leads to failure modes in PINN-based modeling. The baseline PINN works very well as an optical fiber model with relatively simple fiber parameters and for uncomplicated transmission tasks. Yet, it struggles when the modeling task becomes relatively complex, reaching very high error, for example, numerous modeling tasks/scenarios in soliton communication and soliton pulse development in special fibers such as erbium-doped dispersion compensating fibers. We implement two methods to circumvent the limitations caused by the physics-based regularization term to solve this problem, namely, the so-called scaffolding technique for PINN modeling and the progressive block learning PINN modeling strategy to solve the nonlinear Schrödinger equation (NLSE), which models pulse propagation in an optical fiber. This helps PINN learn more accurately the dynamics of pulse evolution and increases accuracy by two to three orders of magnitude. We show in addition that this error is not due to the depth or architecture of the neural network but a fundamental issue inherent to PINN by design. The results achieved indicate a considerable reduction in PINN error for complex modeling problems, with accuracy increasing by up to two orders of magnitude.