NSNO: Neumann Series Neural Operator for Solving Helmholtz Equations in Inhomogeneous Medium

Fukai Chen,Ziyang Liu,Guochang Lin,Junqing Chen,Zuoqiang Shi
2024-01-24
Abstract:In this paper, we propose Neumann Series Neural Operator (NSNO) to learn the solution operator of Helmholtz equation from inhomogeneity coefficients and source terms to solutions. Helmholtz equation is a crucial partial differential equation (PDE) with applications in various scientific and engineering fields. However, efficient solver of Helmholtz equation is still a big challenge especially in the case of high wavenumber. Recently, deep learning has shown great potential in solving PDEs especially in learning solution operators. Inspired by Neumann series in Helmholtz equation, we design a novel network architecture in which U-Net is embedded inside to capture the multiscale feature. Extensive experiments show that the proposed NSNO significantly outperforms the state-of-the-art FNO with at least 60\% lower relative $L^2$-error, especially in the large wavenumber case, and has 50\% lower computational cost and less data requirement. Moreover, NSNO can be used as the surrogate model in inverse scattering problems. Numerical tests show that NSNO is able to give comparable results with traditional finite difference forward solver while the computational cost is reduced tremendously.
Numerical Analysis
What problem does this paper attempt to address?
This paper focuses on efficient solving of the wave equation - the Helmholtz equation in inhomogeneous medium. The Helmholtz equation has wide applications in physics and engineering fields, such as acoustics, electromagnetics, and medical imaging. However, efficient solvers are still challenging, especially for cases that require repeated solving with different coefficients and source terms, such as inverse scattering problems. The paper proposes a new neural network operator called "Niemann Series Neural Operator" (NSNO) for learning the operator of the Helmholtz equation from inhomogeneous coefficients and source terms. Inspired by the Niemann series in the Helmholtz equation, they design a network architecture that embeds U-Net to capture multi-scale features. Experiments show that NSNO significantly outperforms the state-of-the-art FNO with at least 60% reduction in relative L2 error, while reducing the computational cost by 50% and the data requirement, especially in high wave number cases. In addition, NSNO can also serve as a proxy model for inverse scattering problems, providing results comparable to traditional finite difference forward solvers but with greatly improved speed. Overall, the paper addresses the issues that existing methods fail to fully consider: mapping both coefficients and source terms to the operator of the solution simultaneously, and handling training instability caused by high-frequency oscillations in the solution. These problems are effectively resolved through NSNO and the combination of U-Net in the network architecture.