[논문 리뷰] A Matrix-free Preconditioner for the Helmholtz Equation based on the Fast Multipole Method

Huda Ibeid, Rio Yokota외 1명|arXiv (Cornell University)|2016. 08. 08.

Electromagnetic Scattering and Analysis참고 문헌 27인용 수 4

한 줄 요약

이 논문은 빠른 다중체 방법(Fast Multipole Method, FMM)을 활용하여 헬름홀츠 방정식에 대한 행렬 미존재(preconditioner)를 제안한다. FMM의 $Ó(N/log N)$ 복잡도와 계산 집약적 커널을 이용해 대규모 희소 선형 시스템의 스케일러블하고 효율적인 반복적 해법을 가능하게 한다. 이 방법은 메esh 및 파장수에 관계없이 수렴성을 확보하며, 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.

연구 동기 및 목표

• 고파장수 헬름홀츠 이산화에서 발생하는 대규모 비정칙 복소계수 선형 시스템을 반복적 방법으로 해결하는 데 도전하는 데 목적이 있다.
• 비정칙 헬름홀츠 문제에서 다중격자 및 대수적 다중격자 방법의 한계를 극복할 수 있는 스케일러블하고 행렬 미존재의 예조건화 방법을 개발하는 데 목적이 있다.
• 빠른 다중체 방법의 계층적 구조와 낮은 랭크 외접 상호작용 압축을 활용하여 전체 행렬을 저장하지 않고도 효율적인 예조건화를 달성하는 데 목적이 있다.
• 메쉬 및 파장수에 독립적인 수렴 속도를 확보하면서 동시에 대규모 아키텍처에서 강력한 스케일러비리티를 유지하는 데 목적이 있다.
• 공개된, 재현 가능한 코드를 사용하여 최신 예조건화 방법과 비교했을 때 경쟁 가능한 해법 시간을 보여주는 데 목적이 있다.

제안 방법

• Krylov 부분공간 해법기, 특히 GMRES에 대해 FMM을 행렬 미존재 예조건화 방법으로 사용하여 헬름홀츠 연산자의 역행렬 작용을 근사한다.
• 경계요소법(Boundary Element Method, BEM)과 FMM을 결합하여 경계 조건을 정확하게 구현함으로써, 임피던스 또는 딜레르흐 경계 조건을 가진 내부 헬름홀츠 문제를 해결할 수 있도록 한다.
• FMM의 계층적 트리 구조는 원거리 상호작용의 낮은 랭크 압축을 가능하게 하여 계산 비용과 통신 비용을 모두 감소시킨다.
• 정확한 행렬-벡터 곱 계산이 필요할 뿐 고정밀도 해법이 필요하지 않기 때문에 저순위 다중체 전개(p=6 등)를 사용한다.
• PETSc에 맞춤형 FMM 커널을 구현하고 PetIGA에 통합하여 이소지오메트릭 분석을 수행하며, ExaFMM를 기반 FMM 라이브러리로 사용한다.
• 1024코어까지의 슈퍼컴퓨터인 Shaheen II를 사용하여 강력한 스케일러비리티를 평가하였으며, 성능 측정은 PETSc의 -log_summary 옵션을 통해 수행되었다.

실험 결과

연구 질문

• RQ1 빠른 다중체 방법은 비정칙 헬름홀츠 방정식에 대해 효과적으로 행렬 미존재 예조건화 방법으로 재사용될 수 있는가?
• RQ2 FMM 예조건화 방법은 고파장수 헬름홀츠 문제에서 메쉬 및 파장수에 독립적인 수렴성을 확보하는가?
• RQ3 기존의 예조건화 방법들인 AMG, IC 및 다중격자 방법과 비교해 FMM 예조건화 방법은 대규모 코어 수에서 어떻게 스케일링되는가?
• RQ4 FMM 예조건화 방법은 헬름홀츠 문제의 반복적 해법에서 통신 부담을 줄이고 해법 시간을 향상시킬 수 있는가?
• RQ5 3차원 헬름홀츠 문제에서 FMM의 계산 복잡도와 통신 효율성 사이의 성능 트레이드오프는 어떠한가?

주요 결과

• 테스트된 $\kappa$ 값과 메쉬 크기 $h$에 대해 FMM 예조건화 방법은 메쉬 독립적이고 파장수 독립적인 수렴 속도를 확보하여 문제 파rameter에 관계없이 강건함을 입증한다.
• 1024코어까지의 강력한 스케일러비리티가 입증되었으며, FMM 예조건화 방법은 매우 이면적 통신 패턴 덕분에 효율적인 성능 유지를 유지한다.
• 동일한 문제 크기에서 3D 헬름홀츠 FMM은 3D 포isson FMM보다 약 10배 정도 느리지만, Krylov 반복에서의 더 빠른 수렴으로 인해 이 손해를 상쇄한다.
• FMM 예조건화 방법의 해법 시간은 최신 예조건화 방법과 경쟁 가능하며, 특히 통신 감소와 행렬 미존재 연산 덕분에 유리하다.
• 이 방법은 행렬 미존재이며 큰 희소 행렬을 저장하지 않아 메모리 대역폭과 저장 요구량을 크게 감소시킨다.
• 모든 코드, 특히 FMM 강화 버전의 PetIGA와 ExaFMM 라이브러리가 공개되어 재현성과 확장성을 확보하고 있다.