BLAST: Beyond Limber Angular power Spectra Toolkit. A fast and efficient algorithm for 3x2 pt analysis

기술 요약: Blast.jl – 3 × 2 pt 분석을 위한 빠르고 효율적인 알고리즘

문제 정의
차세대 광학 및 분광 관측 조사(예: Euclid, LSST, Roman, DESI)의 등장은 거대 구조(LSS)에 대한 고정밀 측정을 약속한다. 이러한 조사들은 은하 클러스터링, 약한 중력 렌즈 시어-시어 상관관계(shear-shear correlations), 그리고 은하-은하 렌즈 효과의 교차 상관관계를 결합하는 "3 × 2 pt" 통계량에 크게 의존한다. 이러한 통계량의 이론적 모델링은 3D 물질 파워 스펙트럼과 고도로 진동하는 구면 베셀 함수(spherical Bessel functions)의 곱을 3차원으로 적분함으로써 각도 파워 스펙트럼($C_\ell$)을 계산하는 것을 필요로 한다.

계산 비용을 줄이기 위한 표준 접근 방식은 림버 근사(Limber approximation)를 사용하는 것인데, 이는 서로 다른 시대의 모드들이 선형 방향(line-of-sight) 적분에 기여하지 않는다고 가정함으로써 3D 적분을 1D 적분으로 단순화한다. 그러나 이 근사는 특히 큰 각도 규모(낮은 $\ell$)에서 상당한 오차를 유발한다. 미래의 조사들이 가진 제약 능력을 완전히 활용하고 계통 오차를 제어하기 위해서는 정확한 비-림버(non-Limber) 계산이 필수적이다. 하지만 수백만 번의 평가가 필요한 마르코프 연쇄 몬테카를로(MCMC)와 같은 파라미터 추론 방법에서 전체 3D 적분을 수행하는 것은 계산적으로 불가능하다. 예를 들어, 최근 림버 근사를 사용한 분석에서는 수렴을 위해 0.5백만 CPU 시간이 소요되었다.

방법론
저자들은 척도 의존적 성장률(scale-dependent growth rate)을 가정하지 않고도 전체 비-림버 각도 파워 스펙트럼을 효율적으로 계산하도록 설계된 새로운 알고리즘인 Blast.jl을 제시한다. 핵심 방법론은 **체비쇼프 다항식(Chebyshev polynomials)**을 이용한 물질 파워 스펙트럼의 분해에 기초한다.

1. 체비쇼프 분해: 알고리즘은 $[k_{min}, k_{max}]$ 구간에서 매끄러운 물질 파워 스펙트럼 $P(k, \chi_1, \chi_2)$를 체비쇼프 급수 전개를 사용하여 근사한다. 이를 통해 적분을 기하학적 성분과 우주론적 성분으로 분리할 수 있다.
2. 기하학적 항의 사전 계산: 알고리즘은 가장 계산 집약적인 부분, 즉 두 개의 구면 베셀 함수와 체비쇼프 다항식의 곱을 적분하는 과정이 우주론적 파라미터와 무관하게 되도록 구조를 재편한다. 이러한 항들($\tilde{T}^{AB}_{n; \ell}$)은 한 번 계산되어 저장된다.
3. 효율적 평가: 새로운 우주론 모델에 대해, 알고리즘은 파워 스펙트럼의 체비쇼프 계수만을 다시 계산하면 되며, 이 과정은 계산 비용이 매우 저렴하다. 나머지 $\chi$에 대한 외적 적분과 비율 $R = \chi_2/\chi_1$에 대한 적분은 표준 쿼드러처 규칙(Simpson's rule 및 Clenshaw-Curtis quadrature)을 사용하여 수행된다.
4. 좌표 변환: $\chi_1 \approx \chi_2$인 영역이 적분에 가장 크게 기여하므로, 해당 영역을 더 잘 샘플링하기 위해 $(\chi_1, \chi_2)$에서 $(\chi, R)$로 변수 변환을 수행한다.
5. 비선형 보정: 이 방법은 선형 및 비선형 파워 스펙트럼을 별도로 취급한다. 림버 근사가 유효한 작은 규모에서 중요한 비선형 기여는 림버 기반 접근법을 사용하여 보정하는 반면, 전체 비-림버 적분은 선형 성분에 적용된다.

주요 기여

• 알고리즘 혁신: Blast.jl은 FFTLog 기반 방식(FKEM 및 matter 등)의 로그 격자 간격 제약을 피하고, Levin 방식의 반복적인 미분 방정식 풀이를 피하는 체비쇼프 다항식 기반의 분해 접근법을 도입한다.
• 사전 계산 전략: 우주론과 무관한 기하학적 적분을 격리함으로써, 파라미터 추론 과정 중 반복되는 평가 시간을 획기적으로 단축한다.
• 유연성: 경쟁 방법들과 달리, Blast.jl은 척도 독립적 성장률을 가정하지 않으므로 질량이 큰 중성미자(massive neutrinos)나 수정 중력(modified gravity)이 존재하는 우주론에도 적용 가능하다.
• 오픈 소스: 코드는 GitHub에 공개되어 있으며, 사용법을 상세히 설명하는 저장소가 함께 제공된다.

결과
알고리즘은 비-림버 적분 방법을 벤치마킹하기 위해 LSST 암흑 에너지 과학 협력단(LSST Dark Energy Science Collaboration)이 조직한 "N5K"(Non-local No-Nonsense Non-Limber Numerical Knock-out) 챌린지를 통해 평가되었다. 테스트는 64개 스레드를 갖춘 고성능 컴퓨팅 노드(Narval)에서 수행되었으며, Blast.jl을 챌린지 우승자인 FKEM (CosmoLike), matter, Levin과 비교하였다.

• 속도: 기준 정확도 요구 사항($\ell < 200$에서 $\Delta\chi^2 < 0.2$)에서, Blast.jl은 64개 코어를 사용하여 0.00512초의 실행 시간을 달성했다. 이는 이전의 가장 빠른 방법이었던 FKEM(0.05871초)보다 약 10배 빠르며, 32개 코어로 최적화했을 때는 약 15배 빠르다(0.00360초).
• 확장성(Scaling): Blast.jl은 토모그래피 적색편이 빈(tomographic redshift bins)의 수에 대해 우수한 확장성을 보여준다. 다른 방법들은 빈의 수에 따라 실행 시간이 증가하는 반면, Blast.jl의 실행 시간은 빈의 수(2개에서 15개 사이)에 관계없이 사실상 독립적으로 유지된다.
• 정확도: 이 방법은 챌린지의 정확도 요구 사항을 충족한다. 저자들은 체비쇼프 다항식의 수와 샘플링 지점을 조정함으로써 실행 시간에 큰 영향을 주지 않고도 정확도를 조절할 수 있다고 언급했다.
• 정밀도 대 속도의 트레이드오프: 부동 소수점 정밀도를 Float64에서 Float32로 낮추면 정확도에 미미한 영향($\Delta\chi^2$가 0.1054에서 0.1064로 증가)을 주면서 실행 시간을 2배 개선할 수 있었다.

의의
본 논문은 Blast.jl이 현재 가용한 가장 빠른 비-림버 적분 알고리즘이라고 주장하며, 기준 시나리오에서 최첨단 방법들을 10배 이상의 차이로 능가한다고 밝히고 있다. 이 알고리즘의 주요 의의는 오류가 발생하기 쉬운 림버 근사에 의존하지 않고도, 차세대 조사를 위한 3 × 2 pt 통계량을 사용하는 우주론적 분석을 계산적으로 가능하게 만든다는 점에 있다.

저자들은 이러한 속도가 미래의 대규모 데이터셋을 활용한 MCMC 기반 파라미터 추론을 가능하게 하는 데 결정적이라고 강조한다. 또한, 알고리즘 설계가 적색편이 공간 왜곡(redshift space distortions), 확대 편향(magnification bias), 상대론적 보정(relativistic corrections)의 포함 및 CMB 렌즈 효과와의 통합과 같은 향후 확장을 허용한다는 점을 명시한다. 또한 자동 미분(automatic differentiation)을 통합할 계획임을 밝혔는데, 이는 경사 하강법 기반의 추론(gradient-based inference) 방법을 사용하여 우주론 분석 파이프라인을 더욱 가속화할 수 있게 할 것이다.