中国物理B ›› 2024, Vol. 33 ›› Issue (2): 20203-020203.doi: 10.1088/1674-1056/ad0bf4

• • 上一篇    下一篇

MetaPINNs: Predicting soliton and rogue wave of nonlinear PDEs via the improved physics-informed neural networks based on meta-learned optimization

Yanan Guo(郭亚楠)1,2,3,†, Xiaoqun Cao(曹小群)1,2,‡, Junqiang Song(宋君强)1,2, and Hongze Leng(冷洪泽)1   

  1. 1 College of Meteorology and Oceanography, National University of Defense Technology, Changsha 410073, China;
    2 College of Computer, National University of Defense Technology, Changsha 410073, China;
    3 Naval Aviation University, Huludao 125001, China
  • 收稿日期:2023-09-25 修回日期:2023-11-06 接受日期:2023-11-13 出版日期:2024-01-16 发布日期:2024-01-25
  • 通讯作者: Yanan Guo, Xiaoqun Cao E-mail:guoyn14@lzu.edu.cn;caoxiaoqun@nudt.edu.cn
  • 基金资助:
    Project supported by the National Natural Science Foundation of China (Grant Nos. 42005003 and 41475094).

MetaPINNs: Predicting soliton and rogue wave of nonlinear PDEs via the improved physics-informed neural networks based on meta-learned optimization

Yanan Guo(郭亚楠)1,2,3,†, Xiaoqun Cao(曹小群)1,2,‡, Junqiang Song(宋君强)1,2, and Hongze Leng(冷洪泽)1   

  1. 1 College of Meteorology and Oceanography, National University of Defense Technology, Changsha 410073, China;
    2 College of Computer, National University of Defense Technology, Changsha 410073, China;
    3 Naval Aviation University, Huludao 125001, China
  • Received:2023-09-25 Revised:2023-11-06 Accepted:2023-11-13 Online:2024-01-16 Published:2024-01-25
  • Contact: Yanan Guo, Xiaoqun Cao E-mail:guoyn14@lzu.edu.cn;caoxiaoqun@nudt.edu.cn
  • Supported by:
    Project supported by the National Natural Science Foundation of China (Grant Nos. 42005003 and 41475094).

摘要: Efficiently solving partial differential equations (PDEs) is a long-standing challenge in mathematics and physics research. In recent years, the rapid development of artificial intelligence technology has brought deep learning-based methods to the forefront of research on numerical methods for partial differential equations. Among them, physics-informed neural networks (PINNs) are a new class of deep learning methods that show great potential in solving PDEs and predicting complex physical phenomena. In the field of nonlinear science, solitary waves and rogue waves have been important research topics. In this paper, we propose an improved PINN that enhances the physical constraints of the neural network model by adding gradient information constraints. In addition, we employ meta-learning optimization to speed up the training process. We apply the improved PINNs to the numerical simulation and prediction of solitary and rogue waves. We evaluate the accuracy of the prediction results by error analysis. The experimental results show that the improved PINNs can make more accurate predictions in less time than that of the original PINNs.

关键词: physics-informed neural networks, gradient-enhanced loss function, meta-learned optimization, nonlinear science

Abstract: Efficiently solving partial differential equations (PDEs) is a long-standing challenge in mathematics and physics research. In recent years, the rapid development of artificial intelligence technology has brought deep learning-based methods to the forefront of research on numerical methods for partial differential equations. Among them, physics-informed neural networks (PINNs) are a new class of deep learning methods that show great potential in solving PDEs and predicting complex physical phenomena. In the field of nonlinear science, solitary waves and rogue waves have been important research topics. In this paper, we propose an improved PINN that enhances the physical constraints of the neural network model by adding gradient information constraints. In addition, we employ meta-learning optimization to speed up the training process. We apply the improved PINNs to the numerical simulation and prediction of solitary and rogue waves. We evaluate the accuracy of the prediction results by error analysis. The experimental results show that the improved PINNs can make more accurate predictions in less time than that of the original PINNs.

Key words: physics-informed neural networks, gradient-enhanced loss function, meta-learned optimization, nonlinear science

中图分类号:  (Numerical simulation; solution of equations)

  • 02.60.Cb
07.05.Mh (Neural networks, fuzzy logic, artificial intelligence) 02.30.Jr (Partial differential equations) 05.45.Yv (Solitons)