Search papers, labs, and topics across Lattice.
This paper introduces McMg, a learned phase-space multigrid preconditioner specifically designed for solving heterogeneous Helmholtz equations at high wavenumbers. By mapping residuals to corrections within an iterative framework, McMg retains crucial local wave information through a multi-channel approach that incorporates amplitude, phase, direction, and scattering coefficients at each coarse node. Numerical experiments reveal that McMg significantly reduces the number of iterations and wall-clock time compared to traditional methods, while outperforming existing neural preconditioners in high-frequency, high-contrast, and large-scale scenarios.
McMg slashes the computational burden of solving complex Helmholtz equations by leveraging learned phase-space corrections, making it a game-changer for high-frequency wave problems.
Solving heterogeneous Helmholtz equations at high wavenumbers remains challenging because the discretized operator is indefinite, pollution degrades phase accuracy, and scalar coarse-grid correction can discard the local phase and propagation-direction information carried by oscillatory errors. We propose Multi-channel Multigrid (McMg), a learned phase-space multigrid preconditioner for heterogeneous Helmholtz equations. Rather than predicting the solution directly, McMg maps residuals to corrections within an iterative framework. Its central idea is to coarsen physical space while retaining unresolved local wave information in the channel dimension: each coarse node carries a learned packet of amplitude, phase, direction, and scattering coefficients rather than a single scalar unknown. The architecture combines linear multi-channel transfer operators with locally adaptive stencils, neural PDE operators, and medium-dependent smoothers whose coefficients are generated from the wave speed. For a fixed medium, the V-cycle is linear in the residual; nonlinear physical features are computed once in a setup phase and cached, so each online iteration reduces to convolutions with fixed coefficients. We further study generalization across scales. Models trained on small domains transfer directly to larger domains and higher effective wavenumbers, and a Layer-by-Layer Progressive Finetuning (LLPF) strategy extends the support of the learned Green's operator by adding and finetuning only new coarse levels. Numerical experiments on high-frequency, high-contrast, and large-scale three-dimensional problems demonstrate that McMg requires substantially fewer iterations and less wall-clock time than strong classical baselines, while consistently outperforming existing neural preconditioners.
