astroABC: An Approximate Bayesian Computation Sequential Monte Carlo sampler for cosmological parameter estimation

Abstract

Given the complexity of modern cosmological parameter inference where we are faced with non-Gaussian data and noise, correlated systematics and multi-probe correlated data sets, the Approximate Bayesian Computation (ABC) method is a promising alternative to traditional Markov Chain Monte Carlo approaches in the case where the Likelihood is intractable or unknown. The ABC method is called "Likelihood free" as it avoids explicit evaluation of the Likelihood by using a forward model simulation of the data which can include systematics. We introduce astroABC, an open source ABC Sequential Monte Carlo (SMC) sampler for parameter estimation. A key challenge in astrophysics is the efficient use of large multi-probe datasets to constrain high dimensional, possibly correlated parameter spaces. With this in mind astroABC allows for massive parallelization using MPI, a framework that handles spawning of jobs across multiple nodes. A key new feature of astroABC is the ability to create MPI groups with different communicators, one for the sampler and several others for the forward model simulation, which speeds up sampling time considerably. For smaller jobs the Python multiprocessing option is also available. Other key features include: a Sequential Monte Carlo sampler, a method for iteratively adapting tolerance levels, local covariance estimate using scikit-learn's KDTree, modules for specifying optimal covariance matrix for a component-wise or multivariate normal perturbation kernel, output and restart files are backed up every iteration, user defined metric and simulation methods, a module for specifying heterogeneous parameter priors including non-standard prior PDFs, a module for specifying a constant, linear, log or exponential tolerance level, well-documented examples and sample scripts. This code is hosted online at https://github.com/EliseJ/astroABC

19 pages, 3 figures. Comments welcome

References in corpus (14)

• Cosmological parameters from CMB and other data: a Monte-Carlo approach
• Approximate Bayesian computation scheme for parameter inference and model selection in dynamical systems
• Observational Probes of Cosmic Acceleration
• Bayes in the sky: Bayesian inference and model selection in cosmology
• Adaptive approximate Bayesian computation
• CosmoSIS: modular cosmological parameter estimation
• The Effect of Covariance Estimator Error on Cosmological Parameter Constraints
• Approximate Bayesian Computation for Astronomical Model Analysis: A Case Study in Galaxy Demographics and Morphological Transformation at High Redshift
• Likelihood-Free Cosmological Inference with Type Ia Supernovae: Approximate Bayesian Computation for a Complete Treatment of Uncertainty
• cosmoabc: Likelihood-free inference via Population Monte Carlo Approximate Bayesian Computation
• A new model to predict weak-lensing peak counts II. Parameter constraint strategies
• Combining Probes of Large-Scale Structure with CosmoLike
• Integrated approach to cosmology: Combining CMB, large-scale structure and weak lensing
• On estimating cosmology-dependent covariance matrices

Cited by in corpus (28)

• Gaia DR2 reveals a star formation burst in the disc 2-3 Gyr ago
• Bayesian optimisation for likelihood-free cosmological inference
• The Canada-France Imaging Survey: Reconstructing the Milky Way Star Formation History from its White Dwarf Population
• Fast and Credible Likelihood-Free Cosmology with Truncated Marginal Neural Ratio Estimation
• A New Method to Measure the Post-Reionization Ionizing Background from the Joint Distribution of Lyman-$\alpha$ and Lyman-$\beta$ Forest Transmission
• Towards a model-independent reconstruction approach for late-time Hubble data
• Simulation-based inference of dynamical galaxy cluster masses with 3D convolutional neural networks
• Dark energy by natural evolution: Constraining dark energy using Approximate Bayesian Computation
• Accelerating Approximate Bayesian Computation with Quantile Regression: Application to Cosmological Redshift Distributions
• Kernel Selection for Gaussian Process in Cosmology: with Approximate Bayesian Computation Rejection and Nested Sampling
• Prospects for Recovering Galaxy Intrinsic Shapes from Projected Quantities
• 21st Century Statistical and Computational Challenges in Astrophysics
• Primordial power spectrum and cosmology from black-box galaxy surveys
• Inferring the photometric and size evolution of galaxies from image simulations
• Approximating Bayes in the 21st Century
• A simulation-based inference pipeline for cosmic shear with the Kilo-Degree Survey
• Accurate X-ray Timing in the Presence of Systematic Biases With Simulation-Based Inference
• CoLFI: Cosmological Likelihood-free Inference with Neural Density Estimators
• Machine Learning Accelerated Likelihood-Free Event Reconstruction in Dark Matter Direct Detection
• Hubble constant by natural selection: Evolution chips in the Hubble tension
• Modeling competitive evolution of multiple languages
• A tomographic spherical mass map emulator of the KiDS-1000 survey using conditional generative adversarial networks
• Joint inference of the Milky Way star formation history and IMF from Gaia all-sky $G<13$ data
• Deep Learning and genetic algorithms for cosmological Bayesian inference speed-up
• Probing the Consistency of Cosmological Contours for Supernova Cosmology
• Sequentially guided MCMC proposals for synthetic likelihoods and correlated synthetic likelihoods
• Deep inference of simulated strong lenses in ground-based surveys
• A new method to retrieve the star formation history from white dwarf luminosity functions -- an application to the Gaia catalogue of nearby stars