The future of cosmological likelihood-based inference: accelerated high-dimensional parameter estimation and model comparison

Davide Piras, Alicja Polanska, A. Spurio Mancini, Matthew A. Price, Jason D. McEwen

The Open Journal of Astrophysics 2024 article cited by 19

importance 5/10 ai assisted Cosmology and Gravitation Theories cosmology stage: research

Bayesian parameter estimation and model comparison in cosmology becomes computationally slow in high-dimensional settings, limiting analyses.

Slow inference constrains how quickly cosmologists can extract results from large observational datasets.

A machine learning emulator (CosmoPower-JAX) mimics cosmological observables, combined with differentiable/probabilistic programming (JAX, NumPyro), gradient-based MCMC, and learned harmonic mean evidence estimation to accelerate full Bayesian analysis.

No concrete numbers; abstract claims analysis in 'a fraction of the time' of traditional approaches

Demonstrated a complete accelerated Bayesian analysis pipeline, but on simulated data per the abstract.

Venue The Open Journal of Astrophysics
Published 2024
Work type article
Authors Davide Piras, Alicja Polanska, A. Spurio Mancini, Matthew A. Price, Jason D. McEwen
Citations 19 (OpenAlex, at import time)
Model or method CosmoPower-JAX neural network emulator; JAX/NumPyro; harmonic
Organization not stated in source
Primary source https://doi.org/10.33232/001c.123368
Open access diamond
Keywords cosmology, Bayesian inference, machine learning emulation, MCMC, model selection
5 / 10

real but narrower advance

Accelerates Bayesian cosmological inference, cutting a simulated Stage-IV analysis from 8 months on 48 CPUs to 2 days on 12 GPUs; benefits cosmology analyses, demonstrated on simulated data.

Basis: stated. The source itself states the scale. Stated scale: computational cost reduced from 8 months on 48 CPU cores to 2 days on 12 GPUs. Method and limits.

# Paper Year Citations Importance
1 Magnetic control of tokamak plasmas through deep reinforcement learning
Jonas Degrave, Federico Felici, Jonas Buchli, Michael Neunert, Brendan...
2022 739
7
2 Solving the quantum many-body problem with artificial neural networks
Giuseppe Carleo, Matthias Troyer
2017 2,205
7
3 Learning high-accuracy error decoding for quantum processors
Johannes Bausch, Andrew Senior, Francisco J. H. Heras, Thomas Edlich,...
2024 90
7
4 Optimizing Quantum Error Correction Codes with Reinforcement Learning
Hendrik Poulsen Nautrup, Nicolas Delfosse, Vedran Dunjko, Hans J. Brie...
2019 158
7
5 Machine-learning-assisted correction of correlated qubit errors in a topological code
Paul Baireuther, Thomas E. O’Brien, Brian Tarasinski, C. W. J. Beenakk...
2018 141
7