Fast likelihood-free inference in the LSS Stage IV era

Abstract

Forthcoming large-scale structure (LSS) Stage IV surveys will provide us with unprecedented data to probe the nature of dark matter and dark energy. However, analysing these data with conventional Markov Chain Monte Carlo (MCMC) methods will be challenging, due to the increase in the number of nuisance parameters and the presence of intractable likelihoods. In light of this, we present the first application of Marginal Neural Ratio Estimation (MNRE) (a recent approach in simulation-based inference) to LSS photometric probes: weak lensing, galaxy clustering and the cross-correlation power spectra. In order to analyse the hundreds of spectra simultaneously, we find that a pre-compression of data using principal component analysis, as well as parameter-specific data summaries lead to highly accurate results. Using expected Stage IV experimental noise, we are able to recover the posterior distribution for the cosmological parameters with a speedup factor of $\sim 10-60$ compared to classical MCMC methods. To illustrate that the performance of MNRE is not impeded when posteriors are highly non-Gaussian, we test a scenario of two-body decaying dark matter, finding that Stage IV surveys can improve current bounds on the model by up to one order of magnitude. This result supports that MNRE is a powerful framework to constrain the standard cosmological model and its extensions with next-generation LSS surveys.

arXiv comment18+9 pages, 9 figures, 3 tables. Improved slightly text and added comparison with MultiNest. v2 matches the published JCAP version. Comments welcome!

References in corpus (59)

• Array Programming with NumPy
• emcee: The MCMC Hammer
• Efficient Computation of CMB anisotropies in closed FRW models
• MultiNest: an efficient and robust Bayesian inference tool for cosmology and particle physics
• The Cosmic Linear Anisotropy Solving System (CLASS) II: Approximation schemes
• Stable clustering, the halo model and nonlinear cosmological power spectra
• Approximate Bayesian computation scheme for parameter inference and model selection in dynamical systems
• Bayes in the sky: Bayesian inference and model selection in cosmology
• Dark Energy Survey Year 3 Results: Cosmological Constraints from Galaxy Clustering and Weak Lensing
• Conservative Constraints on Early Cosmology: an illustration of the Monte Python cosmological parameter inference code
• Cobaya: Code for Bayesian Analysis of hierarchical physical models
• CMB power spectrum parameter degeneracies in the era of precision cosmology
• PolyChord: nested sampling for cosmology
• The Cosmic Linear Anisotropy Solving System (CLASS) I: Overview
• PolyChord: next-generation nested sampling
• HMcode-2020: Improved modelling of non-linear cosmological power spectra with baryonic feedback
• Galaxy alignments: An overview
• Accurate halo-model matter power spectra with dark energy, massive neutrinos and modified gravitational forces
• Real-time gravitational-wave science with neural posterior estimation
• Fast likelihood-free cosmology with neural density estimators and active learning
• Dark Energy Survey Year 3 Results: Photometric Data Set for Cosmology
• COSMOPOWER: emulating cosmological power spectra for accelerated Bayesian inference from next-generation surveys
• Galaxy alignments: Observations and impact on cosmology
• Euclid preparation: IX. EuclidEmulator2 -- Power spectrum emulation with massive neutrinos and self-consistent dark energy perturbations
• Eigenmode Analysis of Galaxy Redshift Surveys I. Theory and Methods
• Galaxy alignments: Theory, modelling and simulations
• Massive optimal data compression and density estimation for scalable, likelihood-free inference in cosmology
• Physical Bayesian modelling of the non-linear matter distribution: new insights into the Nearby Universe
• Fast $\epsilon$-free Inference of Simulation Models with Bayesian Conditional Density Estimation
• Approximating Likelihood Ratios with Calibrated Discriminative Classifiers
• Likelihood-free inference with neural compression of DES SV weak lensing map statistics
• Consistency of effective field theory analyses of the BOSS power spectrum
• Simulation-Based Inference of Reionization Parameters From 3D Tomographic 21 cm Lightcone Images
• Bayesian emulator optimisation for cosmology: application to the Lyman-alpha forest
• Linear cosmological constraints on 2-body decaying dark matter scenarios and the $S_8$ tension
• Fisher Matrices and Confidence Ellipses: A Quick-Start Guide and Software
• Fast and Credible Likelihood-Free Cosmology with Truncated Marginal Neural Ratio Estimation
• How much information can be extracted from galaxy clustering at the field level?
• A Full $w$CDM Analysis of KiDS-1000 Weak Lensing Maps using Deep Learning
• CONNECT: A neural network based framework for emulating cosmological observables and cosmological parameter inference
• Neural Posterior Estimation with guaranteed exact coverage: the ringdown of GW150914
• Cosmic Shear: Inference from Forward Models
• CosmicNet II: Emulating extended cosmologies with efficient and accurate neural networks
• A Trust Crisis In Simulation-Based Inference? Your Posterior Approximations Can Be Unfaithful
• SICRET: Supernova Ia Cosmology with truncated marginal neural Ratio EsTimation
• Estimating the warm dark matter mass from strong lensing images with truncated marginal neural ratio estimation
• Constraining the X-ray heating and reionization using 21-cm power spectra with Marginal Neural Ratio Estimation
• EFTofLSS meets simulation-based inference: ${\sigma }_8$ from biased tracers
• One never walks alone: the effect of the perturber population on subhalo measurements in strong gravitational lenses
• Sampling-Based Accuracy Testing of Posterior Estimators for General Inference
• Multifield Cosmology with Artificial Intelligence
• A simulation-based inference pipeline for cosmic shear with the Kilo-Degree Survey
• The future of cosmological likelihood-based inference: accelerated high-dimensional parameter estimation and model comparison
• Probing the two-body decaying dark matter scenario with weak lensing and the cosmic microwave background
• SIDE-real: Supernova Ia Dust Extinction with truncated marginal neural ratio estimation applied to real data
• What to do when things get crowded? Scalable joint analysis of overlapping gravitational wave signals
• Hybrid SBI or How I Learned to Stop Worrying and Learn the Likelihood
• Detection is truncation: studying source populations with truncated marginal neural ratio estimation
• Dark Energy Survey Year 3 results: likelihood-free, simulation-based $w$CDM inference with neural compression of weak-lensing map statistics

Cited by in corpus (5)

• KiDS-SBI: Simulation-based inference analysis of KiDS-1000 cosmic shear
• Euclid preparation. Sensitivity to non-standard particle dark matter model
• A frequentist view on the two-body decaying dark matter model
• How to embed any likelihood into SBI: Application to Planck + Stage IV galaxy surveys and Dynamical Dark Energy
• Fast Sampling of Cosmological Initial Conditions with Gaussian Neural Posterior Estimation