수십억 은하의 비등방성 3점 상관함수 초고속 계산

초록

Galactos는 Intel Xeon Phi 기반 슈퍼컴퓨터에서 O(N²) 알고리즘과 부하 균형 k‑d 트리를 활용해 2 억 개 이상의 은하에 대한 비등방성 3점 상관함수(3PCF)를 실시간에 가깝게 계산한다. SIMD 벡터화와 캐시 재사용을 극대화해 단일 노드에서 피크 성능의 39%를 달성했으며, 전체 Cori 시스템(9636노드)에서는 5.06 PF(지속) 성능을 기록한다.

상세 분석

이 논문은 우주론의 핵심 과제인 암흑 에너지와 중력 이론 검증을 위해 은하 군집의 비등방성 3점 상관함수(3PCF)를 대규모 데이터에 적용할 수 있는 혁신적인 계산 프레임워크를 제시한다. 기존 3PCF 알고리즘은 삼중 조합을 모두 탐색해야 하므로 O(N³) 복잡도를 갖아 수백만 개 은하 수준에서도 실용적이지 못했다. Galactos는 두 단계의 핵심 아이디어로 이를 O(N²)로 낮춘다. 첫째, 각 은하를 ‘primary’로 삼아 반경 Rmax(200 Mpc/h) 이내의 ‘secondary’ 은하들을 k‑d 트리를 이용해 효율적으로 찾는다. 트리는 부하 균형을 위해 동적으로 파티셔닝되며, 각 노드가 담당하는 영역이 균등하도록 설계돼 대규모 분산 환경에서도 작업 불균형을 최소화한다. 둘째, 삼각형의 각 변 방향을 구면조화함수(Yℓm)로 전개하고, 두 변의 조화함수 곱을 이용해 비등방성 정보를 보존한다. 구면조화 전개는 회전 불변성을 유지하면서도 각 변과 시선(z축) 사이의 각도를 자연스럽게 포함한다. 구현 측면에서는 Intel Xeon Phi(KNL) 아키텍처의 512‑bit SIMD 레지스터를 활용해 거리 계산, 구면조화 계수 축적, 그리고 다중 스레드 동기화를 벡터화하였다. L1/L2 캐시 재사용을 극대화하기 위해 데이터 레이아웃을 구조체‑오브‑배열(SOA) 형태로 변환하고, 메모리 접근 패턴을 연속적으로 유지하도록 루프 차원을 재배열했다. 결과적으로 단일 노드에서 피크 FLOP 수의 39%를 달성했으며, 전체 시스템에서는 9.8 PF(피크)·5.06 PF(지속) 성능을 기록, 2 억 개 은하에 대한 3PCF를 수시간 내에 완성한다. 이러한 성능은 기존 2PCF 전용 최적화 코드와 비교해 동일한 하드웨어에서 5배 이상 빠른 속도를 제공한다.