[논문 리뷰] Rotation method for accelerating multiple-spherical Bessel function integrals against a numerical source function

Zachary Slepian, Yin Li외 2명|arXiv (Cornell University)|2019. 11. 29.

Radio Astronomy Observations and Technology참고 문헌 4인용 수 8

한 줄 요약

이 논문은 다중구면 베셀 함수(sBF) 적분을 가속화하는 데 목적이 있는 '회전 방법'을 소개한다. 이는 삼각함수의 곱-합 항등식을 활용해 적분을 재구성함으로써 1D FFT를 적용할 수 있도록 하여 $N_{\rm g}\log N_{\rm g}$의 복잡도를 달성한다. 이 방법은 난이도가 높은 $N_{\rm g}^{p+1}$ 접근 방식에 비해 최대 1000–3000배 빠른 성능을 보이며, DESI 및 LSST와 같은 대규모 설문조사에서 통계상 상관관계 함수 및 공분산 행렬 계산의 효율성을 크게 향상시킨다.

ABSTRACT

A common problem in cosmology is to integrate the product of two or more spherical Bessel functions (sBFs) with different configuration-space arguments against the power spectrum or its square, weighted by powers of wavenumber. Naively computing them scales as $N_{ m g}^{p+1}$ with $p$ the number of configuration space arguments and $N_{ m g}$ the grid size, and they cannot be done with Fast Fourier Transforms (FFTs). Here we show that by rewriting the sBFs as sums of products of sine and cosine and then using the product to sum identities, these integrals can then be performed using 1-D FFTs with $N_{ m g} \log N_{ m g}$ scaling. This "rotation" method has the potential to accelerate significantly a number of calculations in cosmology, such as perturbation theory predictions of loop integrals, higher order correlation functions, and analytic templates for correlation function covariance matrices. We implement this approach numerically both in a free-standing, publicly-available extsc{Python} code and within the larger, publicly-available package exttt{mcfit}. The rotation method evaluated with direct integrations already offers a factor of 6-10$ imes$ speed-up over the naive approach in our test cases. Using FFTs, which the rotation method enables, then further improves this to a speed-up of $\sim$$1000-3000 imes$ over the naive approach. The rotation method should be useful in light of upcoming large datasets such as DESI or LSST. In analysing these datasets recomputation of these integrals a substantial number of times, for instance to update perturbation theory predictions or covariance matrices as the input linear power spectrum is changed, will be one piece in a Monte Carlo Markov Chain cosmological parameter search: thus the overall savings from our method should be significant.

연구 동기 및 목표

• 우주론에서 수치적 원천 함수와의 다중구면 베셀 함수 적분을 평가할 때 발생하는 계산적 병목 현상을 해결하기 위해.
• 구성공간 변수 $p$개에 대해 난이도가 높은 $N_{\rm g}^{p+1}$의 스케일링을 극복하기 위해.
• 기본적인 FFT 가속화와 호환되지 않는 sBF 적분에 대해 빠른 푸리에 변환(FFTs)을 효율적으로 적용할 수 있도록 하기 위해.
• 대규모 구조 설문조사에서 상관관계 함수 및 공분산 행렬을 계산하기 위한 수치적으로 안정적이고 고정밀도의 방법을 제공하기 위해.
• 반복적인 적분 재계산이 필요한 몬테카를로 마르코프 체인(MCMC) 매개변수 추정과 같은 계산적으로 부담스러운 작업을 지원하기 위해.

제안 방법

• 삼각함수 항등식을 사용하여 구면 베셀 함수를 사인과余현의 합으로 재작성한다.
• 곱-합 삼각함수 항등식을 적용하여 다중 구성공간 변수에 대한 적분을 단일 차이와 합에 대한 적분으로 변환한다.
• 문제를 구성공간 평면에서 $(a,b)$에서 $(a-b, a+b)$로 45도의 회전으로 재구성함으로써 1D FFT 적용이 가능하도록 한다.
• 결과적으로 발생하는 1D FFT를 실시간으로 수행하여 큰 룩업 테이블을 피하고 메모리 내 계산을 가능하게 한다.
• 회전 변환을 활용해 고차원 sBF 적분을 $N_{\rm g}\log N_{\rm g}$ 복잡도를 가진 1D FFT의 연속으로 변환한다.
• 공개된 파이썬 패키지와 mcfit 프레임워크 내에 구현하여 광범위한 사용성을 확보한다.

실험 결과

연구 질문

• RQ1 수치적 원천 함수를 가진 다중구면 베셀 함수 적분은 난이도가 높은 $N_{\rm g}^{p+1}$ 스케일링을 초월해 가속화될 수 있는가?
• RQ2 그들 고유의 다차원적 구조에도 불구하고 sBF 적분을 1D FFT에 적합한 형태로 재구성할 수 있는가?
• RQ3 직접 수치적 적분 및 기존의 룩업 기반 접근 방식에 비해 회전 방법이 달성할 수 있는 성능 향상은 어느 정도인가?
• RQ4 적색편이 설문조사의 공분산 행렬 추정과 같은 고정밀도 천체물리학적 응용에서 이 방법은 어떤 성능을 보이는가?
• RQ5 수치적 정밀도 손실 없이 이 방법을 더 높은 sBF 지수 $\ell$로 확장할 수 있는가?

주요 결과

• 회전 방법은 FFT 기반 평가를 가능하게 함으로써 다중 sBF 적분의 계산 스케일을 $N_{\rm g}^{p+1}$에서 $N_{\rm g}\log N_{\rm g}$로 감소시킨다.
• 시험 케이스에서 직접 통합에 비해 6–10배의 성능 향상을 기록하였으며, FFT와 결합할 경우 최대 1000–3000배의 성능 향상을 달성한다.
• 난이도가 높은 방법과의 수치적 일치도가 충분하여 DESI 분석과 같은 응용에서 체계적 오차가 무시할 수 있을 정도이다.
• 큰 룩업 테이블을 피함으로써 효율적인 메모리 내 계산이 가능해지고 I/O 병목 현상이 감소한다.
• 이 방법은 이미 $\ell = 0$ 및 $\ell = 2$ 다중극성분에 대해 효과적이며, 2PCF 공분산 행렬의 핵심 구성요소를 커버한다.
• 고차원의 $\ell$에서는 적분함수의 큰 다이나믹 레인지로 인해 도전 과제가 발생하지만, 현재의 설문조사 규모 응용에는 여전히 타당하다.