[논문 리뷰] An adaptive aggregation based domain decomposition multilevel method for the lattice wilson dirac operator: multilevel results

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

Matrix Theory and Algorithms참고 문헌 14인용 수 12

한 줄 요약

이 논문은 격자 QCD의 격자 윌슨 디랙 연산자에 대한 적응형 집합 기반 다중수준 영역분할 가속화 다중격자 방법을 제시한다. 두 수준의 DD-α AMG 솔버를 적응형 보간과 곱셉 슈바르츠 스무딩을 사용해 진정한 다중수준 프레임워크로 확장함으로써, 표준 켈리브 방법 대비 700배의 속도 향상과 두 수준 버전 대비 1.7배의 성능 향상을 달성하였으며, 수준 간에 강한 스케일링 성능과 γ₅ 대칭성 유지가 유지되었다.

ABSTRACT

In lattice QCD computations a substantial amount of work is spent in solving linear systems arising in Wilson's discretization of the Dirac equations. We show first numerical results of the extension of the two-level DD-αAMG method to a true multilevel method based on our parallel MPI-C implementation. Using additional levels pays off, allowing to cut down the core minutes spent on one system solve by a factor of approximately 700 compared to standard Krylov subspace methods and yielding another speed-up of a factor of 1.7 over the two-level approach.

연구 동기 및 목표

• 격자 윌슨 디랙 연산자를 해결하기 위해 두 수준의 적응형 집합 기반 영역분할 다중격자(DD-α AMG) 방법을 진정한 다중수준 프레임워크로 확장하는 것.
• 윌슨 페르미온 이산화에서 발생하는 대규모 선형 시스템의 해를 빠르게 구하기 위해 계산 효율성을 향상시키는 것.
• 대칭을 유지하는 갈레르킨 조밀화를 통해 모든 굵은 수준에서 디랙 연산자의 γ₅ 대칭성을 유지하는 것.
• MPI와 적응형 설정 루틴을 사용하여 대규모 병렬 아키텍처에서 확장 가능한 성능을 달성하는 것.
• 다중격자 계층에서 추가 수준을 도입할 경우 두 수준 방법을 초월해 측정 가능한 성능 향상이 발생하는지 조사하는 것.

제안 방법

• 모든 수준에서 곱셉 슈바르츠(SAP) 스무딩을 사용하는 V-사이클 다중격자 전략을 적용하여 양호한 데이터 국소성과 강한 스케일링 성능 확보.
• 스핀-1,2와 스피너-3,4 자유도를 별도로 그룹화하는 적응형 집합을 사용하여 보간 연산자 $ P_\nu $ 를 구성함으로써 γ₅ 대칭성 유지.
• 굵은 수준 연산자는 갈레르킨 투영을 통해 형성됨: $ D_{\nu+1} = P_\nu^\top D_\nu P_\nu $, 이는 대칭성 유지와 일관된 오차 전파 보장.
• 역반복과 부트스트랩 기법을 조합한 새로운 설정 루틴을 통해 굵은 수준 보정 부분공간 구축을 위한 테스트 벡터를 효율적으로 생성.
• 다중격자 방법은 유연한 켈리브 부분공간 솔버(FGMRES 등) 내에서 프리컨디셔너로 사용되며, 굵은 수준의 시스템은 정확하게 해결하지 않고 설정 비용 절감.
• 다중수준 계층은 굵은 수준의 시스템이 직접 해법이 가능한 크기로 작아질 때까지 재귀적으로 적용되어 전체 다중격자 재귀 구현 가능.

실험 결과

연구 질문

• RQ1 두 수준 DD-α AMG 방법을 진정한 다중수준 프레임워크로 확장할 경우, 격자 윌슨 디랙 연산자를 해결하는 데 측정 가능한 성능 향상이 발생하는가?
• RQ2 적응형 집합, SAP 스무딩, 대칭성 유지 갈레르킨 조밀화를 조합할 경우, 여러 수준에서 안정적인 수렴 성능 유지를 할 수 있는가?
• RQ3 다중격자 계층에서 추가 수준을 포함할 경우, 두 수준 버전 대비 해법 시간과 확장성에 어떤 영향을 미치는가?
• RQ4 γ₅ 대칭성 유지가 다중수준 방법의 효과성과 안정성에 어떤 영향을 미치는가?
• RQ5 추가 수준의 계산 비용을 감안할 때, 다중수준 설정을 충분히 효율적으로 만들 수 있는가?

주요 결과

• 세 수준 DD-α AMG 방법은 128개 프로세스 구성에서 단일 해법에 대해 표준 켈리브 부분공간 방법 대비 약 700배의 속도 향상을 달성하였다.
• 다중수준 방법은 두 수준 DD-α AMG 방법 대비 1.7배 성능 향상을 보였으며, 추가 수준이 측정 가능한 성능 향상 기여를 한다는 것을 입증하였다.
• 모든 수준에서 γ₅ 대칭성이 유지되었으며, 이는 격자 QCD 시뮬레이션에서 물리적 일관성에 매우 중요하다.
• 세 수준 방법의 설정 시간은 두 수준 버전보다 略로 높지만, 해법 시간 감소가 이를 충분히 상쇄하여 단일 우변 벡터의 경우에도 유리하다.
• 수치 실험 결과 다중수준 방법은 격자 크기가 증가함에 따라 유리하게 스케일링되며, 임계 쿼크 질량 근처에서도 안정성이 향상됨을 보였다.
• 비정확한 정규화를 사용한 비정확한 정사영 방법에 비해, 속도와 대칭성 유지 측면에서 다중수준 방법이 뛰어나지만, 후자의 경우 굵은 수준의 복잡도 감소로 설정이 더 빠르다는 점을 감안하면 유의미한 성능 차이가 존재한다.