[论文解读] Kernel Aggregated Fast Multipole Method: Efficient summation of Laplace and Stokes kernel functions

Wen Yan, Robert N. Blackwell|arXiv (Cornell University)|Oct 28, 2020

Fluid Dynamics Simulations and Interactions参考文献 27被引用 7

一句话总结

本文提出核聚合快速多极方法（KAFMM），即KIFMM的扩展，通过在不同八叉树遍历阶段使用针对核函数的等效点，加速拉普拉斯和斯托克斯核求和。该方法在支持开放边界、周期性边界、无滑移壁面及多种核函数（包括斯托克斯点、RPY张量和正则化斯托克斯点）的前提下，实现了O(N)复杂度，且周期性条件下的性能开销极低。相关开源STKFMM库实现了高效计算。

ABSTRACT

Many different simulation methods for Stokes flow problems involve a common computationally intense task -- the summation of a kernel function over $O(N^2)$ pairs of points. One popular technique is the Kernel Independent Fast Multipole Method (KIFMM), which constructs a spatial adaptive octree for all points and places a small number of equivalent multipole and local equivalent points around each octree box, and completes the kernel sum with $O(N)$ cost, using these equivalent points. Simpler kernels can be used between these equivalent points to improve the efficiency of KIFMM. Here we present further extensions and applications to this idea, to enable efficient summations and flexible boundary conditions for various kernels. We call our method the Kernel Aggregated Fast Multipole Method (KAFMM), because it uses different kernel functions at different stages of octree traversal. We have implemented our method as an open-source software library STKFMM based on the high performance library PVFMM, with support for Laplace kernels, the Stokeslet, regularized Stokeslet, Rotne-Prager-Yamakawa (RPY) tensor, and the Stokes double-layer and traction operators. Open and periodic boundary conditions are supported for all kernels, and the no-slip wall boundary condition is supported for the Stokeslet and RPY tensor. The package is designed to be ready-to-use as well as being readily extensible to additional kernels.

研究动机与目标

• 解决模拟斯托克斯流及相关问题中核求和带来的O(N²)计算瓶颈。
• 通过引入分阶段核聚合，扩展核无关快速多极法（KIFMM），提升多种核函数下的计算效率。
• 在统一、可扩展的框架中实现复杂边界条件（开放、周期性、无滑移壁面）的支持。
• 提供高性能、开源的STKFMM库，支持多种物理核函数，且性能开销极低。

提出的方法

• 提出KAFMM，其在向上（多极）和向下（局部）八叉树遍历阶段使用不同的核函数，实现在不同阶段的核特异性近似。
• 在每个八叉树盒子中使用等效源点（M点和L点）表示远场相互作用，替代直接成对求和。
• 通过在不同遍历层级应用不同核函数（如拉普拉斯、斯托克斯点、RPY）实现核聚合，优化精度与速度。
• 支持多种核函数，包括拉普拉斯、斯托克斯点、正则化斯托克斯点、RPY张量及牵引算子，均保持一致的O(N)复杂度。
• 通过将M→L算子整合到向下遍历阶段，实现周期性边界条件，将开销降低至求值时间的约5%。
• 基于PVFMM进行最小化修改，确保与现有并行化及SIMD优化（AVX2，未来支持AVX-512）兼容。

实验结果

研究问题

• RQ1 在不同八叉树遍历阶段使用核特异性近似，能否显著降低拉普拉斯与斯托克斯核求和的计算成本？
• RQ2 如何高效地将周期性边界条件集成到核无关快速多极法中，且性能损耗极低？
• RQ3 单一FMM框架在保持一致O(N)扩展性前提下，能在多大程度上支持如斯托克斯点、RPY张量和正则化斯托克斯点等多样化物理核函数？
• RQ4 在核无关FMM框架中，支持无滑移壁面边界条件的性能开销有多大？
• RQ5 KAFMM实现的并行扩展性如何？在不同核函数与边界条件下表现如何？

主要发现

• KAFMM方法对核求和实现了O(N)计算复杂度，使大规模斯托克斯与拉普拉斯问题的可扩展模拟成为可能。
• 周期性边界条件相比开放边界条件仅增加约5%的计算开销，显著降低了以往实现的成本。
• STKFMM库支持12种不同的核函数，包括斯托克斯点、RPY张量及牵引算子，在开放、单重、双重及三重周期性条件下均保持一致性能。
• 成功实现了斯托克斯点与RPY张量的无滑移壁面边界条件，扩展了该方法在受限流体动力学中的适用性。
• 强弱扩展基准测试显示，在48核节点上并行效率达约70%，与底层PVFMM库性能相当。
• 该方法具有高度可扩展性，软件设计清晰，可轻松集成新核函数而无需修改核心算法。