[논문 리뷰] PolyChord: nested sampling for cosmology

Will Handley, M. P. Hobson외 1명|arXiv (Cornell University)|2015. 02. 06.

Scientific Research and Discoveries인용 수 8

한 줄 요약

PolyChord는 하드 우도 제약 조건 내에서 단면 표집을 사용하는 고차원 베이지안 추론을 위한 내장 표집 알고리즘으로, 베이지안 증거 계산과 사후 분포 표본 추출을 효율적으로 수행한다. 다중Nested(MultiNest)보다 고차원 공간에서 파arameter 속도 계층 구조를 활용하고 거의 선형적 병렬 처리를 가능하게 하여, 가우시안 우도에서 O(D³) 스케일링을 보이며 MultiNest의 지수적 스케일링을 초월한다. 이는 플랑크 2015 분석에서 37개 매개변수의 천체물리학적 추론을 가능하게 하였다.

ABSTRACT

PolyChord is a novel nested sampling algorithm tailored for high dimensional parameter spaces. In addition, it can fully exploit a hierarchy of parameter speeds such as is found in CosmoMC and CAMB. It utilises slice sampling at each iteration to sample within the hard likelihood constraint of nested sampling. It can identify and evolve separate modes of a posterior semi-independently and is parallelised using openMPI. PolyChord is available for download at: http://ccpforge.cse.rl.ac.uk/gf/project/polychord/

연구 동기 및 목표

• 현대 천체물리학적 매개변수 공간에서 다중Nested(MultiNest)와 같은 기존 내장 표집 알고리즘의 한계를 해결하기 위해.
• 천체물리학적 우도에서 흔한 매개변수 속도 계층 구조를 활용하여, 베이지안 증거 계산과 사후 표본 추출을 효율적으로 수행하기 위해.
• 정확도와 수렴성을 유지하면서 고차원, 다모달 사후 분포에서 확장 가능하고 병렬 처리가 가능한 알고리즘 개발을 위해.
• 반복 기반 표집의 대안으로, 하드 우도 제약 조건 내에서 단면 표집을 사용하여 내장 표집에서 거부 표집을 대체하기 위해.

제안 방법

• PolyChord는 내장 표집 알고리즘에서 정의된 하드 우도 제약 조건 내에서 매 반복마다 단면 표집을 수행하여, 거부 표집을 대체한다.
• 매개변수를 느린 속도에서 빠른 속도로 변환하기 위해 콜레스키 분해를 사용하여 천체물리학 모델의 매개변수 속도 계층을 자연스럽게 활용한다.
• 알고리즘은 별도의 사후 모드를 식별하고 부분적으로 독립적으로 진화시켜, 다모달 분포에서 혼합 성능을 향상시킨다.
• openMPI를 사용하여 병렬 처리되며, 프로세스 수가 라이브 포인트 수 이하일 경우 거의 선형적인 성능 향상이 이루어진다.
• 표본 추출 공간으로의 변환을 통해 각 매개변수는 오직 더 빠른 매개변수와만 혼합되며, 이는 혼합 성능을 향상시키는 장기 반복 체인을 가능하게 한다.
• CosmoMC와 CAMB와의 통합을 통해 표준 메트로폴리스-해스팅스 MCMC를 내장 표집으로 대체함으로써, CosmoChord 프레임워크를 구성한다.

실험 결과

연구 질문

• RQ1 내장 표집 알고리즘이 현대 천체물리학에서 흔한 고차원 매개변수 공간을 효율적으로 처리할 수 있는가?
• RQ2 내장 표집 알고리즘이 매개변수 속도 계층을 어떻게 활용하여 표본 추출 효율성을 향상시킬 수 있는가?
• RQ3 하드 우도 제약 조건 내에서 단면 표집이 고차원, 다모달 사후 분포에서 거부 표집을 능가할 수 있는가?
• RQ4 이 새로운 접근법을 사용할 경우 고차원 문제에서 증거 계산과 수렴 성능의 스케일링 행동은 어떠한가?
• RQ5 openMPI를 통한 병렬 처리가 정확도를 희생시키지 않고 성능 향상에 얼마나 기여할 수 있는가?

주요 결과

• PolyChord는 D차원 가우시안 우도에서 우도 평가에 대해 O(D³) 스케일링을 달성하며, 고차원에서 다중Nested의 지수적 스케일링과 대비된다.
• 동일한 가우시안 테스트 케이스에서 PolyChord는 분석적 값과 0.4 로그 단위 이내의 증거 추정치를 생성하여, 중요도 내장 표집을 사용한 다중Nested와 동등한 정확도를 확보한다.
• 알고리즘은 플랑크 2015 논문에서 인플레이션 제약 조건을 분석할 때 37개 매개변수(4개 느림, 19개 중간, 14개 빠름)의 천체물리학적 추론을 성공적으로 수행하여 실제 적용 가능성을 입증하였다.
• PolyChord의 병렬 처리는 프로세스 수가 라이브 포인트 수 이하일 경우 거의 선형적 성능 향상을 보이며, 대규모 컴퓨팅 클러스터의 효율적 활용을 가능하게 한다.
• 단면 표집과 콜레스키 기반 매개변수 순서 조정을 통해 다모달 사후 분포의 효과적 탐색과 표준 거부 표집 대비 향상된 혼합 성능이 가능하다.
• CosmoMC 내의 CosmoChord 구현은 MCMC를 내장 표집으로 성공적으로 대체하였으며, 정확도를 유지하면서도 모델 비교를 위한 견고한 증거 계산을 가능하게 하였다.