Symmetry group based domain decomposition to enhance physics-informed neural networks for solving partial differential equations

TL;DR

This paper introduces a symmetry group based domain decomposition method to improve physics-informed neural networks (PINN) for solving PDEs, effectively handling interfaces and enhancing accuracy in forward and inverse problems.

Contribution

It proposes a novel symmetry group based domain decomposition strategy that enhances PINN performance by leveraging Lie symmetries to improve solution accuracy and interface handling.

Findings

Significant accuracy improvements in solving PDEs with symmetry-based domain decomposition.

Effective handling of interfaces between sub-domains in PINN training.

Successful application to Korteweg-de Vries and viscous fluid equations.

Abstract

Domain decomposition provides an effective way to tackle the dilemma of physics-informed neural networks (PINN) which struggle to accurately and efficiently solve partial differential equations (PDEs) in the whole domain, but the lack of efficient tools for dealing with the interfaces between two adjacent sub-domains heavily hinders the training effects, even leads to the discontinuity of the learned solutions. In this paper, we propose a symmetry group based domain decomposition strategy to enhance the PINN for solving the forward and inverse problems of the PDEs possessing a Lie symmetry group. Specifically, for the forward problem, we first deploy the symmetry group to generate the dividing-lines having known solution information which can be adjusted flexibly and are used to divide the whole training domain into a finite number of non-overlapping sub-domains, then utilize the PINN and the symmetry-enhanced PINN methods to learn the solutions in each sub-domain and finally stitch them to the overall solution of PDEs. For the inverse problem, we first utilize the symmetry group acting on the data of the initial and boundary conditions to generate labeled data in the interior domain of PDEs and then find the undetermined parameters as well as the solution by only training the neural networks in a sub-domain. Consequently, the proposed method can predict high-accuracy solutions of PDEs which are failed by the vanilla PINN in the whole domain and the extended physics-informed neural network in the same sub-domains. Numerical results of the Korteweg-de Vries equation with a translation symmetry and the nonlinear viscous fluid equation with a scaling symmetry show that the accuracies of the learned solutions are improved largely.