[논문 리뷰] Adaptive Aggregation Based Domain Decomposition Multigrid for the Lattice Wilson Dirac Operator

Andreas Frommer, Karsten Kahl외 3명|arXiv (Cornell University)|2013. 03. 06.

Advanced Numerical Methods in Computational Mathematics참고 문헌 30인용 수 11

한 줄 요약

이 논문은 격자 QCD에서 격자 윌슨 디랙 연산자를 해결하기 위한 새로운 적응형 집합 기반 도메인 분할 다중 격자 방법인 DD-αAMG를 제안한다. 도메인 분할을 스무딩 기법으로 사용하고 적응형 대수적 다중 격자 계층 구조를 구성함으로써, 비정확한 탈출 기법 대비 최대 1.43배의 성능 향상을 달성하였으며, 8,192개 코어에서 표준 AMG 대비 2–3배 빠른 해법 시간을 기록하였다. 또한 설정 시간을 줄이고 재귀적 다중 격자 순환을 가능하게 하였다.

ABSTRACT

In lattice QCD computations a substantial amount of work is spent in solving discretized versions of the Dirac equation. Conventional Krylov solvers show critical slowing down for large system sizes and physically interesting parameter regions. We present a domain decomposition adaptive algebraic multigrid method used as a precondtioner to solve the "clover improved" Wilson discretization of the Dirac equation. This approach combines and improves two approaches, namely domain decomposition and adaptive algebraic multigrid, that have been used seperately in lattice QCD before. We show in extensive numerical test conducted with a parallel production code implementation that considerable speed-up over conventional Krylov subspace methods, domain decomposition methods and other hierarchical approaches for realistic system sizes can be achieved.

연구 동기 및 목표

• 윌슨 디랙 연산자를 위한 강력하고 확장 가능한 전처리자 개발을 통해 격자 QCD 시뮬레이션에서의 임계 느려짐을 해결한다.
• 기존의 켈러리브 솔버와 기존 전처리자(예: 홀짝, 탈출)가 여전히 임계 느려짐에 시달리는 문제를 극복한다.
• 적응형 대수적 다중 격자에 도메인 분할을 스무딩 기법으로 통합하여 병렬 확장성 향상과 글로벌 통신 감소를 달성한다.
• 비정확한 탈출 기법과 달리, 군집 격자 해법의 정확도 요구 수준을 낮춤으로써 설정 시간을 줄이고 재귀적 다중 격자 순환을 가능하게 한다.
• 슈퍼컴퓨터에서 사용 가능한 생산 수준의 병렬 구현을 통해 실제 대규모 격자 구조에서 뛰어난 성능을 입증한다.

제안 방법

• 더 나은 병렬성과 통신 감소를 위해 기존의 켈러리브 기반 스무딩 기법을 대체하는 도메인 분할(DD)을 스무딩 기법으로 사용하는 다중 격자 프레임워크를 제안한다.
• 테스트 벡터 기반 적응형 보간법을 사용하면서 Γ₅ 대칭성을 유지함으로써 집합 기반 대수적 다중 격자(AMG) 프레임워크를 격자 QCD에 적응시킨다.
• 낮은 정확도로 군집 격자 연산자를 계산하는 새로운 적응형 설정 전략을 도입하여 설정 속도 향상과 재귀적 다중 격자 순환 가능성을 확보한다.
• 군집 시스템을 근사적으로 해법하는 이중 수준 방법을 사용하여 메서드를 전체 다중 격자 순환으로 확장할 수 있도록 한다.
• Juropa와 같은 슈퍼컴퓨터에서 MPI와 하이브리드 OpenMP+MPI를 사용한 병렬 생산 수준 코드베이스에 구현한다.
• 비정확한 탈출 기반 프레임워크를 기준으로 삼지만, 군집 격자 해법 정확도를 낮추고 재귀화를 가능하게 함으로써 이를 진정된 다중 격자 방법으로 재구성한다.

실험 결과

연구 질문

• RQ1 도메인 분할이 격자 윌슨 디랙 연산자를 위한 대수적 다중 격자에서 스무딩 기법으로 효과적으로 사용될 수 있는가? 이는 확장성 향상과 통신 감소에 기여하는가?
• RQ2 제안된 적응형 설정 전략은 기존의 AMG 및 비정확한 탈출 기법에 비해 설정 비용과 수렴 속도 측면에서 어떻게 비교되는가?
• RQ3 기존의 켈러리브 솔버 및 기존 전처리자에 비해 새로운 방법이 임계 느려짐을 얼마나 줄이는가?
• RQ4 군집 격자 해법의 정확도를 낮춰도 수렴 성능이 떨어지지 않으며, 재귀적 다중 격자 순환을 가능하게 하는가?
• RQ5 수천 개의 CPU 코어를 사용하는 대규모 실질적 격자 구조에서 이 방법이 어떤 성능 향상을 달성하는가?

주요 결과

• 불량한 조건을 가진 게이지 구조를 가진 128×64³ 격자에서 DD-αAMG는 비정확한 탈출 기법 대비 1.43배 빠른 성능을 기록했으며, 설정 및 해법 시간이 각각 40% 이상 감소하였다.
• 8,192개 코어에서 DD-αAMG는 표준 AMG 대비 2–3배 더 빠른 해법 시간을 기록했으며, 설정 비용이 유사한 상황에서 해법 시간은 1.82초로 줄어들었다. 이는 AMG의 5.51초 대비 향상된 성능이다.
• 8,192개 코어에서 설정 시간은 27.7초로 단축되었으며, 이는 기본 AMG 설정 설정 기준 89.9초 대비 약 69% 감소한 것이다. 해법 과정에서 GMRES 반복 수는 10–11회로 유지되었다.
• 도메인 분할 스무딩으로 인해 글로벌 통신이 감소하여, 특히 고코어 수에서 강력한 스케일링 성능을 보였다.
• 기존의 BiCGStab를 통한 고정밀 고유벡터 근사가 필요한 전통적 AMG 방법에 비해 적응형 설정 단계가 크게 빨라졌다.
• 군집 격자 해법의 정확도를 낮추었음에도 불구하고 DD-αAMG는 강력한 수렴 성능을 유지하였으며, 재귀적 다중 격자 순환의 실현 가능성을 입증하였다.