candl: Cosmic Microwave Background Analysis with a Differentiable Likelihood
arXiv:2401.13433 · doi:10.1051/0004-6361/202449432
Abstract
We present candl, an automatically differentiable python likelihood for analysing cosmic microwave background (CMB) power spectrum measurements. candl is powered by JAX, which makes it fast and easy to calculate derivatives of the likelihood. This facilitates, for example, robust Fisher matrices without finite-difference methods. We show the benefits of candl through a series of example calculations, covering forecasting, robustness tests, and gradient-based Markov chain Monte Carlo sampling. These also include optimising the band power bin width to minimise parameter errors of a realistic mock data set. Moreover, we calculate the correlation of parameter constraints from correlated and partially overlapping subsets of the SPT-3G 2018 data release. In a traditional analysis framework, these tasks are slow and require careful fine-tuning to obtain stable results. As such, a fully differentiable pipeline allows for a higher level of scrutiny; we argue that this is the paradigm shift required to leverage incoming data from ground-based experiments, which will significantly improve the cosmological parameter constraints from the Planck mission. candl comes with the latest primary and lensing power spectrum data from the South Pole Telescope and Atacama Cosmology Telescope collaborations and will be used as part of the upcoming SPT-3G and data releases. Along with the core code, we release a series of auxiliary tools, which simplify common analysis tasks and interface the likelihood with other cosmological software. candl is pip-installable and publicly available at https://github.com/Lbalkenhol/candl .
arXiv comment8 pages, 3 figures
References in corpus (49)
- SciPy 1.0--Fundamental Algorithms for Scientific Computing in Python
- The NumPy array: a structure for efficient numerical computation
- Efficient Computation of CMB anisotropies in closed FRW models
- The Cosmic Linear Anisotropy Solving System (CLASS) II: Approximation schemes
- The Simons Observatory: Science goals and forecasts
- Planck 2018 results. V. CMB power spectra and likelihoods
- Planck 2013 results. XXII. Constraints on inflation
- Conservative Constraints on Early Cosmology: an illustration of the Monte Python cosmological parameter inference code
- Planck 2015 results. XI. CMB power spectra, likelihoods, and robustness of parameters
- Cobaya: Code for Bayesian Analysis of hierarchical physical models
- The 10 Meter South Pole Telescope
- The Atacama Cosmology Telescope: DR4 Maps and Cosmological Parameters
- PolyChord: nested sampling for cosmology
- PolyChord: next-generation nested sampling
- The Atacama Cosmology Telescope: DR6 Gravitational Lensing Map and Cosmological Parameters
- The Atacama Cosmology Telescope: A Measurement of the DR6 CMB Lensing Power Spectrum and its Implications for Structure Growth
- Massive Lossless Data Compression and Multiple Parameter Estimation from Galaxy Spectra
- The Atacama Cosmology Telescope: A Measurement of the Cosmic Microwave Background Power Spectra at 98 and 150 GHz
- Fast and reliable MCMC for cosmological parameter estimation
- Measurements of the E-Mode Polarization and Temperature-E-Mode Correlation of the CMB from SPT-3G 2018 Data
- Likelihood Analysis of CMB Temperature and Polarization Power Spectra
- Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro
- The Atacama Cosmology Telescope: Two-Season ACTPol Spectra and Parameters
- The Atacama Cosmology Telescope: likelihood for small-scale CMB data
- COSMOPOWER: emulating cosmological power spectra for accelerated Bayesian inference from next-generation surveys
- CMB Polarization can constrain cosmology better than CMB temperature
- Escaping local minima with derivative-free methods: a numerical investigation
- A Measurement of Gravitational Lensing of the Cosmic Microwave Background Using SPT-3G 2018 Data
- Massive data compression for parameter-dependent covariance matrices
- On the role of interaction in sequential Monte Carlo algorithms
- Efficient Cosmological Parameter Estimation with Hamiltonian Monte Carlo
- 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
- Bayesian inflationary reconstructions from Planck 2018 data
- Reconstructing the primordial power spectrum from the CMB
- Likelihood methods for the combined analysis of CMB temperature and polarisation power spectra
- CosmoPower-JAX: high-dimensional Bayesian inference with differentiable cosmological emulators
- Cosmological parameter estimation with free-form primordial power spectrum
- Capse.jl: efficient and auto-differentiable CMB power spectra emulation
- Likelihood methods for CMB experiments
- Extreme data compression for the CMB
- Microcanonical Hamiltonian Monte Carlo
- LimberJack.jl: auto-differentiable methods for angular power spectra analyses
- Uncertainty-aware and Data-efficient Cosmological Emulation using Gaussian Processes and PCA
- Generalisations of Fisher Matrices
- Making maps of cosmological parameters
- Analytic Calculation of Covariance between Cosmological Parameters from Correlated Data Sets, with an Application to SPTpol
- The Cosmic Microwave Background and Its Polarization
- An Auto-Differentiable Likelihood Pipeline for the Cross-Correlation of CMB and Large-Scale Structure due to the Kinetic Sunyaev-Zeldovich Effect
Cited by in corpus (10)
- The CosmoVerse White Paper: Addressing observational tensions in cosmology with systematics and fundamental physics
- SPT-3G D1: CMB temperature and polarization power spectra and cosmology from 2019 and 2020 observations of the SPT-3G Main field
- LINX: A Fast, Differentiable, and Extensible Big Bang Nucleosynthesis Package
- Reducing nuisance prior sensitivity via non-linear reparameterization, with application to EFT analyses of large-scale structure
- jaxspec : a fast and robust Python library for X-ray spectral fitting
- Constraining the primordial power spectrum using a differentiable likelihood
- KiDS-Legacy: Constraining dark energy, neutrino mass, and curvature
- Beyond thermal approximations: Precise cosmological bounds on Axion-Like Particles
- Genetic algorithm demystified for cosmological parameter estimation
- ABCMB: A Python+JAX Package for the Cosmic Microwave Background Power Spectrum