Trans-Net: A transferable pretrained neural networks based on temporal domain decomposition for solving partial differential equations

被引:5
作者
Zhang, Dinglei [1 ]
Li, Ying [1 ]
Ying, Shihui [2 ]
机构
[1] Shanghai Univ, Sch Comp Engn & Sci, Shanghai 200444, Peoples R China
[2] Shanghai Univ, Sch Sci, Dept Math, Shanghai 200444, Peoples R China
关键词
Partial differential equations; Deep neural network; Temporal domain decomposition; Pretrained model; DEEP LEARNING FRAMEWORK; SCHEMES; FLOW;
D O I
10.1016/j.cpc.2024.109130
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Physics -Informed Neural Networks (PINNs) has provided a novel direction for solving partial differential equations (PDEs) and has achieved significant advancements in the field of scientific computing. PINNs effectively incorporate the physical constraints of equations into the loss function, enabling neural networks to learn and approximate the behavior of physical systems by optimizing the loss function. According to the existing research, we propose a transferable pretrained neural networks framework based on temporal domain decomposition to solve partial differential equations. Specifically, we divide the domain into multiple subdomains according to time, each of which is solved using an individual neural network. Subsequently, the trained subnetwork and the predicted values at the common boundary are used as the pretrained model and initial values for the next subdomain, respectively. This not only improves the prediction accuracy of the subnetwork, but also reduces the training time. Finally, we present a series of classical numerical experiments including one-dimensional, two-dimensional, and three-dimensional partial differential equations. The experimental results indicate that the proposed method outperforms existing approaches in terms of accuracy and efficiency. Relevant data and code can be accessed on GitHub via https://github .com /karry -Zhang /Trans -Net.
引用
收藏
页数:12
相关论文
共 37 条
[1]   SPECTRAL AND FINITE-DIFFERENCE SOLUTIONS OF THE BURGERS-EQUATION [J].
BASDEVANT, C ;
DEVILLE, M ;
HALDENWANG, P ;
LACROIX, JM ;
OUAZZANI, J ;
PEYRET, R ;
ORLANDI, P ;
PATERA, AT .
COMPUTERS & FLUIDS, 1986, 14 (01) :23-41
[2]   Stable second-order schemes for the space-fractional Cahn-Hilliard and Allen-Cahn equations [J].
Bu, Linlin ;
Mei, Liquan ;
Hou, Yan .
COMPUTERS & MATHEMATICS WITH APPLICATIONS, 2019, 78 (11) :3485-3500
[3]   NEURAL-NETWORK-BASED APPROXIMATIONS FOR SOLVING PARTIAL-DIFFERENTIAL EQUATIONS [J].
DISSANAYAKE, MWMG ;
PHANTHIEN, N .
COMMUNICATIONS IN NUMERICAL METHODS IN ENGINEERING, 1994, 10 (03) :195-201
[4]   Research paper Modeling density-driven flow in porous media by physics-informed neural networks for CO2 sequestration [J].
Du, Honghui ;
Zhao, Ze ;
Cheng, Haojia ;
Yan, Jinhui ;
He, QiZhi .
COMPUTERS AND GEOTECHNICS, 2023, 159
[5]  
Du Y, 2015, PROC CVPR IEEE, P1110, DOI 10.1109/CVPR.2015.7298714
[6]  
Eymard R, 2000, HDBK NUM AN, V7, P713
[7]   MULTILAYER FEEDFORWARD NETWORKS ARE UNIVERSAL APPROXIMATORS [J].
HORNIK, K ;
STINCHCOMBE, M ;
WHITE, H .
NEURAL NETWORKS, 1989, 2 (05) :359-366
[8]   Enhancing PINNs for solving PDEs via adaptive collocation point movement and adaptive loss weighting [J].
Hou, Jie ;
Li, Ying ;
Ying, Shihui .
NONLINEAR DYNAMICS, 2023, 111 (16) :15233-15261
[9]   An Augmented Lagrangian Deep Learning Method for Variational Problems with Essential Boundary Conditions [J].
Huang, Jianguo ;
Wang, Haoqin ;
Zhou, Tao .
COMMUNICATIONS IN COMPUTATIONAL PHYSICS, 2022, 31 (03) :966-986
[10]   Extended Physics-Informed Neural Networks (XPINNs): A Generalized Space-Time Domain Decomposition Based Deep Learning Framework for Nonlinear Partial Differential Equations [J].
Jagtap, Ameya D. ;
Karniadakis, George Em .
COMMUNICATIONS IN COMPUTATIONAL PHYSICS, 2020, 28 (05) :2002-2041