[论文解读] A Matrix-free Preconditioner for the Helmholtz Equation based on the Fast Multipole Method

Huda Ibeid, Rio Yokota+1 more|arXiv (Cornell University)|Aug 8, 2016

Electromagnetic Scattering and Analysis参考文献 27被引用 4

一句话总结

该论文提出了一种基于快速多极子方法（FMM）的无矩阵预条件子，用于求解亥姆霍兹方程，利用FMM的$Ó(N/log N)$复杂度和计算密集型内核，实现了大规模稀疏线性系统的可扩展、高效迭代求解。该方法在1024个核心下实现了强可扩展性，且收敛性与网格尺寸和波数无关，相较于传统预条件子，在现代超级计算机上展现出更高的通信效率和更优的求解时间。

ABSTRACT

Fast multipole methods (FMM) were originally developed for accelerating $N$-body problems for particle-based methods. FMM is more than an $N$-body solver, however. Recent efforts to view the FMM as an elliptic Partial Differential Equation (PDE) solver have opened the possibility to use it as a preconditioner for a broader range of applications. FMM can solve Helmholtz problems with optimal $\mathcal{O}(N \log N)$ complexity, has compute-bound inner kernels, and highly asynchronous communication patterns. The combination of these features makes FMM an interesting candidate as a preconditioner for sparse solvers on architectures of the future. The use of FMM as a preconditioner allows us to use lower order multipole expansions than would be required as a solver because individual solves need not be accurate. This reduces the amount of computation and communication significantly and makes the time-to-solution competitive with state-of-the-art preconditioners. Furthermore, the high asynchronicity of FMM allows it to scale to much larger core counts than factorization-based and multilevel methods. We describe our tests in reproducible details with freely available codes.

研究动机与目标

• 为解决由高波数亥姆霍兹离散化产生的大规模、不定、复数线性系统，采用迭代方法求解的挑战。
• 开发一种可扩展的无矩阵预条件子，以克服多重网格法和代数多重网格法在不定亥姆霍兹问题上的局限性。
• 利用快速多极子方法的层次化结构和低秩非对角压缩特性，实现无需存储完整矩阵的高效预条件处理。
• 在大规模计算架构上实现与网格尺寸和波数无关的收敛速率，同时保持强可扩展性。
• 通过公开、可复现的代码，与最先进的预条件子相比，展示具有竞争力的求解时间。

提出的方法

• 通过近似亥姆霍兹算子逆的矩阵向量乘积，将FMM用作Krylov子空间求解器（特别是GMRES）的无矩阵预条件子。
• 将FMM与边界元法（BEM）结合，以精确施加边界条件，从而求解具有阻抗或狄利克雷边界条件的内部亥姆霍兹问题。
• FMM的层次化树状结构使得远场相互作用可实现低秩压缩，从而降低计算和通信开销。
• 预条件子采用低阶多极展开（例如$p=6$），因为仅需近似矩阵向量乘积，而非高精度求解。
• 算法在PETSc中实现，使用自定义FMM内核，并集成至PetIGA以支持等几何分析，底层FMM库采用ExaFMM。
• 在Shaheen II超级计算机上，通过PETSc的-log_summary选项，对最多1024个核心的强可扩展性进行了评估。

实验结果

研究问题

• RQ1 快速多极子方法能否被有效重用于不定亥姆霍兹方程的无矩阵预条件子？
• RQ2 FMM预条件子是否能在高波数亥姆霍兹问题中实现与网格尺寸和波数无关的收敛性？
• RQ3 与AMG、IC和多重网格等传统预条件子相比，FMM预条件子在大规模核心数下的可扩展性如何？
• RQ4 FMM预条件子能否减少通信量并提升亥姆霍兹问题迭代求解器的求解时间？
• RQ5 在三维亥姆霍兹问题中，FMM的计算复杂度与通信效率之间存在怎样的性能权衡？

主要发现

• FMM预条件子在测试的$\kappa$值和网格尺寸$h$下，均实现了与网格无关和与波数无关的收敛速率，表明其在不同问题参数下的鲁棒性。
• 在最多1024个核心上实现了强可扩展性，由于其高度异步的通信模式，FMM预条件子保持了高效的性能。
• 对于相同问题规模，三维亥姆霍兹FMM的运行时间约为三维泊松FMM的十倍，这是由于内核操作更复杂，但这一劣势被Krylov迭代中更快的收敛速度所抵消。
• FMM预条件子的求解时间与最先进的预条件子相比具有竞争力，尤其得益于通信量减少和无矩阵运算。
• 该方法为无矩阵方法，避免了存储大型稀疏矩阵，显著降低了内存带宽和存储需求。
• 所有代码（包括增强版PetIGA的FMM分支和ExaFMM库）均已公开，便于复现和进一步扩展。