A complete framework for cosmological emulation and inference with CosmoPower

TL;DR

This paper introduces CosmoPower, a flexible Python framework for creating and sharing machine learning emulators of Einstein-Boltzmann codes, enabling fast, accurate cosmological inference for high-precision cosmology.

Contribution

The paper develops standards and software for generating, packaging, and using emulators of cosmological codes, demonstrated with high-accuracy emulators for CAMB calculations.

Findings

Emulators achieve accuracy within cosmic variance limits.

Significant reduction in computation time for parameter inference.

Framework successfully recovers input cosmology in simulated analyses.

Abstract

We present a coherent, re-usable python framework which further builds on the cosmological emulator code CosmoPower. In the current era of high-precision cosmology, we require high-accuracy calculations of cosmological observables with Einstein-Boltzmann codes. For detailed statistical analyses, such codes often incur high costs in terms of computing power, making parameter space exploration costly, especially for beyond-$\Lambda$CDM analyses. Machine learning-enabled emulators of Einstein-Boltzmann codes have emerged as a solution to this problem and have become a common way to perform fast cosmological analyses. To enable generation, sharing and use of emulators for inference, we define standards for robustly describing, packaging and distributing them, and present software for easily performing these tasks in an automated and replicable manner. We provide examples and guidelines for generating your own sufficiently accurate emulators and wrappers for using them in popular cosmological inference codes. We demonstrate our framework by presenting a suite of high-accuracy emulators for the CAMB code's calculations of CMB $C_\ell$, $P(k)$, background evolution, and derived parameter quantities. We show that these emulators are accurate enough for both $\Lambda$CDM analysis and a set of single- and two-parameter extension models (including $N_{\rm eff}$, $\sum m_{\nu}$ and $w_0 w_a$ cosmologies) with stage-IV observatories, recovering the original high-accuracy Einstein-Boltzmann spectra to tolerances well within the cosmic variance uncertainties across the full range of parameters considered. We also use our emulators to recover cosmological parameters in a simulated cosmic-variance limited experiment, finding results well within $0.1 \sigma$ of the input cosmology, while requiring typically $\lesssim1/50$ of the evaluation time than for the full Einstein-Boltzmann computation.