Abstract:Solving high-wavenumber and heterogeneous Helmholtz equations presents a long-standing challenge in scientific computing. In this paper, we introduce a deep learning-enhanced multigrid solver to address this issue. By conducting error analysis on standard multigrid applied to a discrete Helmholtz equation, we devise a strategy to handle errors with different frequencies separately.
What problem does this paper attempt to address?
This paper attempts to address the long - standing challenge in scientific computing, namely, solving the Helmholtz equation with high wavenumbers and inhomogeneous media. Specifically, the paper mainly focuses on the following three issues:
1. **Solving large - scale linear systems**: The discretized Helmholtz equation requires sufficient grid points to represent each wavelength, especially in the case of high wavenumbers, which will lead to a very large linear system.
2. **Indefiniteness problem**: The discretized Helmholtz equation is indefinite, which makes many numerical methods ineffective or even leads to divergence.
3. **Scalability problem in inhomogeneous media**: For the Helmholtz equation in inhomogeneous media, existing methods applicable to constant wavenumbers may encounter scalability problems.
To solve these problems, the authors propose a deep - learning - enhanced multigrid (MG) solver called Wave - ADR - NS. This solver processes error components at different frequencies through the following strategies:
- For error components far from the wavenumber, use the standard multigrid method for smoothing.
- For error components close to the wavenumber, use another multigrid V - cycle to solve an advection - diffusion - reaction (ADR) equation on a coarse scale.
In addition, the parameters of Wave - ADR - NS are learned through unsupervised training, and numerical experiments show that this method can effectively solve the two - dimensional inhomogeneous Helmholtz equation with wavenumbers up to 2000. Compared with the classical multigrid preconditioner and existing deep - learning - based multigrid preconditioners, Wave - ADR - NS exhibits superior performance.
### Formula Summary
The Helmholtz equation has the following form:
\[
-\Delta u - k^2 u = g
\]
where \( k=\frac{\omega}{c(x)} \) is the wavenumber, \( \omega = 2\pi f \) is the angular frequency, \( f \) is the frequency, \( c(x) \) is the wave speed, and \( g(x)=\delta(x - x_0) \) is the Dirac delta function.
The discretized Helmholtz operator can be represented as:
\[
\frac{1}{h^2}\begin{bmatrix}
0 & - 1 & 0\\
-1 & 4 - k^2h^2 & -1\\
0 & -1 & 0
\end{bmatrix}
\]
The eigenvalues and eigenvectors are respectively:
\[
\lambda_j=\frac{4}{h^2}\sin^2\left(\frac{j\pi h}{2}\right)-k^2,\quad j = 1,\ldots,N
\]
\[
v_j=\{\sin(ij\pi h)\}_{i = 1}^N,\quad j = 1,\ldots,N
\]
Through these formulas, the authors conduct a detailed error analysis and design the Wave - ADR - NS solver to effectively handle error components at different frequencies.