×

An adaptive physics-informed neural network with two-stage learning strategy to solve partial differential equations. (English) Zbl 07814757

Summary: Physics-Informed Neural Network (PINN) represents a new approach to solve Partial Differential Equations (PDEs). PINNs aim to solve PDEs by integrating governing equations and the initial/boundary conditions (I/BCs) into a loss function. However, the imbalance of the loss function caused by parameter settings usually makes it difficult for PINNs to converge, e.g. because they fall into local optima. In other words, the presence of balanced PDE loss, initial loss and boundary loss may be critical for the convergence. In addition, existing PINNs are not able to reveal the hidden errors caused by non-convergent boundaries and conduction errors caused by the PDE near the boundaries. Overall, these problems have made PINN-based methods of limited use on practical situations. In this paper, we propose a novel physics-informed neural network, i.e. an adaptive physics-informed neural network with a two-stage training process. Our algorithm adds spatio-temporal coefficient and PDE balance parameter to the loss function, and solve PDEs using a two-stage training process: pre-training and formal training. The pre-training step ensures the convergence of boundary loss, whereas the formal training process completes the solution of PDE by balancing various loss functions. In order to verify the performance of our method, we consider the imbalanced heat conduction and Helmholtz equations often appearing in practical situations. The Klein-Gordon equation, which is widely used to compare performance, reveals that our method is able to reduce the hidden errors. Experimental results confirm that our algorithm can effectively and accurately solve models with unbalanced loss function, hidden errors and conduction errors. The codes developed in this manuscript are publicy available at https://github.com/callmedrcom/ATPINN.

MSC:

68T07 Artificial neural networks and deep learning
35A25 Other special methods applied to PDEs
65N99 Numerical methods for partial differential equations, boundary value problems
65M99 Numerical methods for partial differential equations, initial value and time-dependent initial-boundary value problems
Full Text: DOI

References:

[1] S. AMINI NIAKI, E. HAGHIGHAT, T. CAMPBELL, A. POURSARTIP, AND R. VAZIRI, Physics-informed neural network for modelling the thermochemical curing process of composite-tool systems during manufacture, Comput. Methods Appl. Mech. Engrg. 384 (2021), 113959. · Zbl 1506.74123
[2] F. ARNOLD AND R. KING, State-space modeling for control based on physics-informed neural networks, Eng. Appl. Artif. Intell. 101 (2021), 104195.
[3] A. G. BAYDIN, B. A. PEARLMUTTER, A. A. RADUL, AND J. M. SISKIND, Automatic differen-tiation in machine learning: A survey, J. Mach. Learn. Res. 18 (2018), 1-43. · Zbl 06982909
[4] A. BIHLO AND R. O. POPOVYCH, Physics-informed neural networks for the shallow-water equations on the sphere, arXiv.2104.00615 (2021).
[5] S. BUOSO, T. JOYCE, AND S. KOZERKE, Personalising left-ventricular biophysical models of the heart using parametric physics-informed neural networks, Med Image Anal. 71 (2021), 102066.
[6] Z. CHANG, K. LI, X. ZOU, AND X. XIANG, High order deep neural network for solving high frequency partial differential equations, Commun. Comput. Phys. 31 (2022), 370-397. · Zbl 1538.65590
[7] Y. CHEN, L. LU, G. E. KARNIADAKIS, AND L. DAL NEGRO, Physics-informed neural networks for inverse problems in nano-optics and metamaterials, Opt. Express 28 (2020), 11618-11633.
[8] F. S. COSTABAL, Y. YANG, P. PERDIKARIS, D. E. HURTADO, AND E. KUHL, Physics-informed neural networks for cardiac activation mapping, Front. Phys. 8 (2020).
[9] W. E AND B. YU, The deep Ritz method: A deep learning-based numerical algorithm for solving variational problems, Commun. Math. Stat. 6 (2018), 1-12. · Zbl 1392.35306
[10] J. N. FUHG AND N. BOUKLAS, The mixed deep energy method for resolving concentration features in finite strain hyperelasticity, J. Comput. Phys. 451 (2022), 110839. · Zbl 07517152
[11] Z. GAO, L. YAN, AND T. ZHOU, Failure-informed adaptive sampling for pinns, arXiv:2210. 00279 (2022).
[12] E. HAGHIGHAT AND R. JUANES, Sciann: A keras/tensorflow wrapper for scientific computa-tions and physics-informed deep learning using artificial neural networks, Comput. Methods Appl. Mech. Engrg. 373 (2021), 113552. · Zbl 1506.65251
[13] Q. Z. HE AND A. M. TARTAKOVSKY, Physics-informed neural network method for forward and backward advection-dispersion equations, Water Resour. Res. 57 (2021).
[14] J. HUANG, H. WANG, AND T. ZHOU, An augmented Lagrangian deep learning method for variational problems with essential boundary conditions, Commun. Comput. Phys. 31 (2022), 966-986. · Zbl 1482.65226
[15] A. D. JAGTAP, K. KAWAGUCHI, AND G. E. KARNIADAKIS, Adaptive activation functions ac-celerate convergence in deep and physics-informed neural networks, J. Comput. Phys. 404 (2020), 109136. · Zbl 1453.68165
[16] A. D. JAGTAP, E. KHARAZMI, AND G. E. KARNIADAKIS, Conservative physics-informed neu-ral networks on discrete domains for conservation laws: Applications to forward and inverse problems, Comput. Methods Appl. Mech. Engrg. 365 (2020), 113028. · Zbl 1442.92002
[17] X. JIN, S. CAI, H. LI, AND G. E. KARNIADAKIS, NSFnets (Navier-Stokes flow nets): Physics-informed neural networks for the incompressible Navier-Stokes equations, J. Comput. Phys. 426 (2021), 109951. · Zbl 07510065
[18] P. KAMSING, P. TORTEEKA, W. BOONPOOK, AND C. CAO, Deep neural learning adaptive sequential Monte Carlo for automatic image and speech recognition, Appl. Comput. Intell. Soft Comput. 2020 (2020), 8866259:1-8866259:9.
[19] E. KHARAZMI, Z. ZHANG, AND G. E. M. KARNIADAKIS, hp-VPINNs: Variational physics-informed neural networks with domain decomposition, Comput. Methods Appl. Mech. En-grg. 374 (2021), 113547. · Zbl 1506.68105
[20] G. KISSAS, Y. YANG, E. HWUANG, W. R. WITSCHEY, J. A. DETRE, AND P. PERDIKARIS, Machine learning in cardiovascular flows modeling: Predicting arterial blood pressure from non-invasive 4d flow MRI data using physics-informed neural networks, Comput. Methods Appl. Mech. Engrg. 358 (2020), 112623. · Zbl 1441.76149
[21] Y. LI, H. ZHANG, C. BERMUDEZ, Y. CHEN, B. A. LANDMAN, AND Y. VOROBEYCHIK, Anatomical context protects deep learning from adversarial perturbations in medical imag-ing, Neurocomputing 379 (2020), 370-378.
[22] Y. LIAO AND P. MING, Deep nitsche method: Deep Ritz method with essential boundary conditions, Commun. Comput. Phys. 29 (2021), 1365-1384. · Zbl 1473.65309
[23] D. LIU AND Y. WANG, Multi-fidelity physics-constrained neural network and its application in materials modeling, J. Mech. Design 141 (2019), 1-13.
[24] C. LO, C. CHEN, AND R. ZHONG, A review of digital twin in product design and development, Adv. Eng. Inform. 48 (2021), 101297.
[25] B. LU, D. XU, AND B. HUANG, Deep-learning-based anomaly detection for lace defect inspec-tion employing videos in production line, Adv. Eng. Inform. 51 (2022), 101471.
[26] L. LU, X. MENG, Z. MAO, AND G. E. KARNIADAKIS, Deepxde: A deep learning library for solving differential equations, SIAM Rev. 63 (2021), 208-227. · Zbl 1459.65002
[27] L. LU, R. PESTOURIE, W. YAO, Z. WANG, F. VERDUGO, AND S. G. JOHNSON, Physics-informed neural networks with hard constraints for inverse design, SIAM J. Sci. Comput. 43 (2021), B1105-B1132. · Zbl 1478.35242
[28] L. LYU, K. WU, R. DU, AND J. CHEN, Enforcing exact boundary and initial conditions in the deep mixed residual method, CSIAM Trans. Appl. Math. 2 (2021), 748-775.
[29] L. LYU, Z. ZHANG, M. CHEN, AND J. CHEN, Mim: A deep mixed residual method for solving high-order partial differential equations, J. Comput. Phys. 452 (2022), 110930. · Zbl 07517751
[30] L. MCCLENNY AND U. BRAGA-NETO, Self-adaptive physics-informed neural networks using a soft attention mechanism, arXiv.2009.04544 (2020).
[31] K. S. MCFALL AND J. R. MAHAN, Artificial neural network method for solution of boundary value problems with exact satisfaction of arbitrary boundary conditions, IEEE trans. neural netw. 20 (2009), 1221-1233.
[32] R. G. NASCIMENTO, K. FRICKE, AND F. A. VIANA, A tutorial on solving ordinary differential equations using Python and hybrid physics-informed neural network, Eng. Appl. Artif. Intell. 96 (2020), 103996.
[33] M. RAISSI, Deep hidden physics models: Deep learning of nonlinear partial differential equa-tions, J. Mach. Learn. Res. 19 (2018), 932-955. · Zbl 1439.68021
[34] M. RAISSI, P. PERDIKARIS, AND G. KARNIADAKIS, Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations, J. Comput. Phys. 378 (2019), 686-707. · Zbl 1415.68175
[35] C. RAO, H. SUN, AND Y. LIU, Physics informed deep learning for computational elastody-namics without labeled data, J. Eng. Mech. 147 (2020), 1-19.
[36] J. REN, T. ZHOU, Y. RONG, Y. MA, AND R. AHMAD, Feature-based modeling for industrial processes in the context of digital twins: A case study of HVOF process, Adv. Eng. Inform. 51 (2022), 101486.
[37] J.-C. REN, D. LIU, AND Y. WAN, Modeling and application of Czochralski silicon single crys-tal growth process using hybrid model of data-driven and mechanism-based methodologies, J. Process Control 104 (2021), 74-85.
[38] S. SHEN, H. LU, M. SADOUGHI, C. HU, V. NEMANI, A. THELEN, K. WEBSTER, M. DARR, J. SIDON, AND S. KENNY, A physics-informed deep learning approach for bearing fault detection, Eng. Appl. Artif. Intell. 103 (2021), 104295.
[39] K. SHUKLA, P. C. DI LEONI, J. BLACKSHIRE, D. SPARKMAN, AND G. E. KARNIADAKIS, Physics-informed neural network for ultrasound nondestructive quantification of surface breaking cracks, J. Nondestr. Eval. 39 (2020).
[40] K. SHUKLA, A. D. JAGTAP, J. L. BLACKSHIRE, D. SPARKMAN, AND G. E. KARNIADAKIS, A physics-informed neural network for quantifying the microstructure properties of polycrys-talline nickel using ultrasound data, arXiv.2103.14104 (2021).
[41] J. SIRIGNANO, J. F. MACART, AND J. B. FREUND, DPM: A deep learning PDE augmentation method with application to large-eddy simulation, J. Comput. Phys. 423 (2020), 109811. · Zbl 07508424
[42] J. SIRIGNANO AND K. SPILIOPOULOS, DGM: A deep learning algorithm for solving partial differential equations, J. Comput. Phys. 375 (2018), 1339-1364. · Zbl 1416.65394
[43] L. SUN, H. GAO, S. PAN, AND J.-X. WANG, Surrogate modeling for fluid flows based on physics-constrained deep learning without simulation data, Comput. Methods Appl. Mech. Engrg. 361 (2020), 112732. · Zbl 1442.76096
[44] K. TANG, X. WAN, AND C. YANG, DAS-PINNs: A deep adaptive sampling method for solving high-dimensional partial differential equations, arXiv:2112.14038 (2022).
[45] S. TANG, Y. ZHU, AND S. YUAN, An improved convolutional neural network with an adapt-able learning rate towards multi-signal fault diagnosis of hydraulic piston pump, Adv. Eng. Inform. 50 (2021), 101406.
[46] Y. WAN, D. LIU, C. C. LIU, AND J. C. REN, Data-driven model predictive control of Cz silicon single crystal growth process with V/G value soft measurement model, IEEE Trans. Semicond. Manuf. 34 (2021), 420-428.
[47] S. WANG, Y. TENG, AND P. PERDIKARIS, Understanding and mitigating gradient pathologies in physics-informed neural networks, SIAM J. Sci. Comput. 43(5) (2021), A3055-A3081. · Zbl 1530.68232
[48] C. L. WIGHT AND J. ZHAO, Solving Allen-Cahn and Cahn-Hilliard equations using the adap-tive physics informed neural networks, Commun. Comput. Phys. 29 (2020), 930-954. · Zbl 07419706
[49] L. YANG, X. MENG, AND G. E. KARNIADAKIS, B-PINNs: Bayesian physics-informed neural networks for forward and inverse PDE problems with noisy data, J. Comput. Phys. 425 (2021), 109913. · Zbl 07508507
[50] Y. YANG AND P. PERDIKARIS, Adversarial uncertainty quantification in physics-informed neural networks, J. Comput. Phys. 394 (2019), 136-152. · Zbl 1452.68171
[51] A. YAZDANI, L. LU, M. RAISSI, AND G. E. KARNIADAKIS, Systems biology informed deep learning for inferring parameters and hidden dynamics, PLoS Comput. Biol. 16 (2020).
[52] D. ZHANG, L. LU, L. GUO, AND G. E. KARNIADAKIS, Quantifying total uncertainty in physics-informed neural networks for solving forward and inverse stochastic problems, J. Comput. Phys. 397 (2019). · Zbl 1454.65008
[53] L.-S. ZHANG, J.-J. TAO, G.-H. WANG, AND X.-J. ZHENG, Experimental study on the origin of lobe-cleft structures in a sand storm, Acta Mech. Sin. 37 (2021), 47-52.
[54] C. ZHU, Deep Learning in Natural Language Processing, Machine Reading Comprehension, 2021.
[55] N. ZOBEIRY AND K. D. HUMFELD, A physics-informed machine learning approach for solving heat transfer equation in advanced manufacturing and engineering applications, Eng. Appl. Artif. Intell. 101 (2021), 104232.
This reference list is based on information provided by the publisher or from digital mathematics libraries. Its items are heuristically matched to zbMATH identifiers and may contain data conversion errors. In some cases that data have been complemented/enhanced by data from zbMATH Open. This attempts to reflect the references listed in the original paper as accurately as possible without claiming completeness or a perfect matching.