COSMOPOWER: emulating cosmological power spectra for accelerated Bayesian inference from next-generation surveys

We present COSMOPOWER, a suite of neural cosmological power spectrum emulators providing orders-of-magnitude acceleration for parameter estimation from two-point statistics analyses of Large-Scale Structure (LSS) and Cosmic Microwave Background (CMB) surveys. The emulators replace the computation of matter and CMB power spectra from Boltzmann codes; thus, they do not need to be re-trained for different choices of astrophysical nuisance parameters or redshift distributions. The matter power spectrum emulation error is less than 0.4 per cent in the wavenumber range k ∈ [10−5, 10] Mpc−1 for redshift z ∈ [0, 5]. COSMOPOWER emulates CMB temperature, polarization, and lensing potential power spectra in the 5-σ region of parameter space around the Planck best-fitting values with an error 10 per cent of the expected shot noise for the forthcoming Simons Observatory. COSMOPOWER is showcased on a joint cosmic shear and galaxy clustering analysis from the Kilo-Degree Survey, as well as on a Stage IV Euclid-like simulated cosmic shear analysis. For the CMB case, COSMOPOWER is tested on a Planck 2018 CMB temperature and polarization analysis. The emulators always recover the fiducial cosmological constraints with differences in the posteriors smaller than sampling noise, while providing a speed-up factor up to O(104) to the complete inference pipeline. This acceleration allows posterior distributions to be recovered in just a few seconds, as we demonstrate in the Planck likelihood case. COSMOPOWER is written entirely in PYTHON, can be interfaced with all commonly used cosmological samplers, and is publicly available at: https://github.com/alessiospuriomancini/cosmopower.