Capse.jl: efficient and auto-differentiable CMB power spectra emulation

Abstract

We present Capse.jl, a novel neural network-based emulator designed for rapid and accurate prediction of Cosmic Microwave Background (CMB) temperature, polarization, and lensing angular power spectra. The emulator computes predictions in just a few microseconds with emulation errors below $0.1\sigma$ for all the scales relevant for the upcoming CMB-S4 survey. Capse.jl can also be trained in an hour's time on a 8-cores CPU. We test Capse.jl on Planck 2018, ACT DR4, and 2018 SPT-3G data and demonstrate its capability to derive cosmological constraints comparable to those obtained by traditional methods, but with a computational efficiency that is three to six orders of magnitude higher. We take advantage of the differentiability of our emulators to use gradient-based methods, such as Pathfinder and Hamiltonian Monte Carlo (HMC), which speed up the convergence and increase sampling efficiency. Together, these features make Capse.jl a powerful tool for studying the CMB and its implications for cosmology. When using the fastest combination of our likelihoods, emulators, and analysis algorithm, we are able to perform a Plancky TT + TE + EE analysis in less than a second. To ensure full reproducibility, we provide open access to the codes and data required to reproduce all the results of this work.

arXiv comment16 pages, 6 figures. Accepted for publication in The Open Journal of Astrophysics. Capse.jl is available $\text{href}$

References in corpus (51)

• Adam: A Method for Stochastic Optimization
• Efficient Computation of CMB anisotropies in closed FRW models
• MCMC using Hamiltonian dynamics
• The Cosmic Linear Anisotropy Solving System (CLASS) II: Approximation schemes
• The Baryon Oscillation Spectroscopic Survey of SDSS-III
• The No-U-Turn Sampler: Adaptively Setting Path Lengths in Hamiltonian Monte Carlo
• The Simons Observatory: Science goals and forecasts
• Dark Energy Survey Year 1 Results: Cosmological Constraints from Galaxy Clustering and Weak Lensing
• Planck 2018 results. V. CMB power spectra and likelihoods
• Wide-Field InfrarRed Survey Telescope-Astrophysics Focused Telescope Assets WFIRST-AFTA 2015 Report
• Statistics of Cosmic Microwave Background Polarization
• Power Spectrum Analysis of Three-Dimensional Redshift Surveys
• Determination of Inflationary Observables by Cosmic Microwave Background Anisotropy Experiments
• The DESI Experiment, a whitepaper for Snowmass 2013
• The clustering of galaxies in the completed SDSS-III Baryon Oscillation Spectroscopic Survey: Anisotropic galaxy clustering in Fourier-space
• HMcode-2020: Improved modelling of non-linear cosmological power spectra with baryonic feedback
• Massive Lossless Data Compression and Multiple Parameter Estimation from Galaxy Spectra
• Measurements of the E-Mode Polarization and Temperature-E-Mode Correlation of the CMB from SPT-3G 2018 Data
• Galaxy And Mass Assembly (GAMA): improved cosmic growth measurements using multiple tracers of large-scale structure
• A Measurement of the Quadrupole Power Spectrum in the Clustering of the 2dF QSO Survey
• The BACCO Simulation Project: Exploiting the full power of large-scale structure for cosmology
• Euclid preparation: II. The EuclidEmulator -- A tool to compute the cosmology dependence of the nonlinear matter power spectrum
• COSMOPOWER: emulating cosmological power spectra for accelerated Bayesian inference from next-generation surveys
• Euclid preparation: IX. EuclidEmulator2 -- Power spectrum emulation with massive neutrinos and self-consistent dark energy perturbations
• Fast cosmological parameter estimation using neural networks
• Massive data compression for parameter-dependent covariance matrices
• SPHEREx: NASA's Near-Infrared Spectrophotmetric All-Sky Survey
• JAX-COSMO: An End-to-End Differentiable and GPU Accelerated Cosmology Library
• CONNECT: A neural network based framework for emulating cosmological observables and cosmological parameter inference
• Data compression in cosmology: A compressed likelihood for Planck data
• Fast Cosmological Parameter Estimation from Microwave Background Temperature and Polarization Power Spectra
• Accelerating Large-Scale-Structure data analyses by emulating Boltzmann solvers and Lagrangian Perturbation Theory
• Mitigating the optical depth degeneracy using the kinematic Sunyaev-Zel'dovich effect with CMB-S4
• CosmicNet II: Emulating extended cosmologies with efficient and accurate neural networks
• Kernel-Based Emulator for the 3D Matter Power Spectrum from CLASS
• Bispectrum-window convolution via Hankel transform
• Euclid-era cosmology for everyone: Neural net assisted MCMC sampling for the joint 3x2 likelihood
• $\text{matryoshka}$: Halo Model Emulator for the Galaxy Power Spectrum
• COMET: Clustering Observables Modelled by Emulated perturbation Theory
• Parameter Inference for Weak Lensing using Gaussian Processes and MOPED
• Fast and robust Bayesian Inference using Gaussian Processes with GPry
• LINNA: Likelihood Inference Neural Network Accelerator
• $\text{matryoshka}$ II: Accelerating Effective Field Theory Analyses of the Galaxy Power Spectrum
• Pathfinder: Parallel quasi-Newton variational inference
• Microcanonical Hamiltonian Monte Carlo
• Uncertainty-aware and Data-efficient Cosmological Emulation using Gaussian Processes and PCA
• Window function convolution with deep neural network models
• Field-Level Inference with Microcanonical Langevin Monte Carlo
• Fast emulation of two-point angular statistics for photometric galaxy surveys
• High-accuracy emulators for observables in $\Lambda$CDM, $N_{\mathrm{eff}}$, $\Sigma m_{\nu }$, and $w$ cosmologies
• Fluctuation without dissipation: Microcanonical Langevin Monte Carlo

Cited by in corpus (15)

• The CosmoVerse White Paper: Addressing observational tensions in cosmology with systematics and fundamental physics
• candl: Cosmic Microwave Background Analysis with a Differentiable Likelihood
• DISCO-DJ I: a differentiable Einstein-Boltzmann solver for cosmology
• Fast and effortless computation of profile likelihoods using CONNECT
• LINX: A Fast, Differentiable, and Extensible Big Bang Nucleosynthesis Package
• Selecting samples of galaxies with fewer Fingers-of-God
• Reducing nuisance prior sensitivity via non-linear reparameterization, with application to EFT analyses of large-scale structure
• Calculating Bayesian evidence for inflationary models using CONNECT
• DeepSSM: an emulator of gravitational wave spectra from sound waves during cosmological first-order phase transitions
• Constraining the primordial power spectrum using a differentiable likelihood
• Cosmological parameters from the joint analysis of Density Split and Second Order Statistics: an Emulator based on the Halo Occupation Distribution
• Design and optimization of neural networks for multifidelity cosmological emulation
• Constraining cosmological parameters using density split lensing and the conditional stellar mass function
• On the accuracy of posterior recovery with neural network emulators
• ABCMB: A Python+JAX Package for the Cosmic Microwave Background Power Spectrum