저비용 시뮬레이션 전이학습으로 우주론 추론 비용 10분의 1로 — FastPM 사전학습 모델

연구진은 저정밀 FastPM(Fast Particle-Mesh) 시뮬레이션으로 사전학습한 모델(FastPM)을 파워 스펙트럼에서 Ωm과 σ8을 추론하도록 훈련한 뒤, 더 적은 수의 고정밀 CAMELS N체·유체역학 시뮬레이션으로 미세조정했다. 동일한 사후분포 정확도에 도달하는 데 필요한 고정밀 시뮬레이션은 CAMELS만으로 훈련할 때의 5~15%에 불과했고, 사전학습용 시뮬레이션 비용을 포함한 총 훈련 계산 비용은 CPU 시간 기준 최대 한 자릿수 차수(약 10배)까지 줄었다. FastPM에는 바리온 물리가 포함되지 않지만, 유체역학 시뮬레이션 한 번보다도 적은 비용의 FastPM 실행으로 사전학습해도 전체 물질 파워 스펙트럼에서 총 훈련 계산량이 약 5분의 1로 감소했다. 또한 더 정확한 FastPM 실행으로 사전학습한 경우, CAMELS만으로 훈련해 얻은 어떤 결과보다도 정확한 사후분포를 얻었다. 저자들은 이를 시뮬레이션 기반 추론(SBI)과 우주론 분야 기반모델(foundation model)의 가능성을 보여주는 결과로 평가했다.