arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.02493math.NAcs.NA
收藏

一种用于高波数非均匀介质亥姆霍兹方程的神经多重网格求解器

A Neural Multigrid Solver for Helmholtz Equations with High Wavenumber and Heterogeneous Media

Chen Cui, Kai Jiang, Shi Shu

AI总结:

本文提出Wave-ADR-NS神经多重网格求解器,通过谱分析分离误差分量,结合多重网格波循环、学习相位函数和ADR循环,高效求解波数超2000的非均匀亥姆霍兹方程。

摘要:

本文提出了一种深度学习增强的多重网格求解器,用于求解高频非均匀亥姆霍兹方程。通过谱分析,我们将迭代误差分为特征分量和非特征分量。我们通过多重网格波循环消除非特征分量,该循环在每个网格上采用精心选择的平滑器。我们通过学习的相位函数以及平流-扩散-反应(ADR)方程的近似解来削弱特征分量,该方程在更粗尺度上使用另一个多重网格V循环求解,称为ADR循环。由此得到的求解器称为Wave-ADR-NS,它能够处理不同频率的误差分量,并克服了粗网格上每个波长网格点数的限制。此外,我们利用可微编程提供了一种高效实现,使Wave-ADR-NS成为一个端到端的亥姆霍兹求解器,其参数通过半监督训练学习得到。Wave-ADR-NS对不同难度的分布内和分布外速度场均表现出稳健的泛化能力。与其他多重网格方法的对比实验验证了其在求解波数超过2000的非均匀二维亥姆霍兹方程时的优越性能。

英文摘要:

In this paper, we propose a deep learning-enhanced multigrid solver for high-frequency and heterogeneous Helmholtz equations. By applying spectral analysis, we categorize the iteration error into characteristic and non-characteristic components. We eliminate the non-characteristic components by a multigrid wave cycle, which employs carefully selected smoothers on each grid. We diminish the characteristic components by a learned phase function and the approximate solution of an advection-diffusion-reaction (ADR) equation, which is solved using another multigrid V-cycle on a coarser scale, referred to as the ADR cycle. The resulting solver, termed Wave-ADR-NS, enables the handling of error components with varying frequencies and overcomes constraints on the number of grid points per wavelength on coarse grids. Furthermore, we provide an efficient implementation using differentiable programming, making Wave-ADR-NS an end-to-end Helmholtz solver that incorporates parameters learned through a semi-supervised training. Wave-ADR-NS demonstrates robust generalization capabilities for both in-distribution and out-of-distribution velocity fields of varying difficulty. Comparative experiments with other multigrid methods validate its superior performance in solving heterogeneous 2D Helmholtz equations with wavenumbers exceeding 2000.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑