We propose a physical information neural network with learning rate decay strategy (LrD-PINN) to predict the dynamics of symmetric, asymmetric, and antisymmetric solitons of the self-defocusing saturable nonlinear Schrödinger equation with the PT-symmetric potential and boost the predicted evolutionary distance by an order of magnitude. Taking symmetric solitons as an example, we explore the advantages of the learning rate decay strategy, analyze the anti-interference performance of the model, and optimize the network structure. In addition, the coefficients of the saturable nonlinearity strength and the modulation strength in the PT-symmetric potential are reconstructed from the dataset of symmetric soliton solutions. The application of more advanced machine learning techniques in the field of nonlinear optics can provide more powerful tools and richer ideas for the study of optical soliton dynamics.
© 2023 Author(s). Published under an exclusive license by AIP Publishing.