We present an accelerated pipeline, based on high-performance computing techniques and normalizing flows, for joint Bayesian parameter estimation and model selection and demonstrate its efficiency in gravitational wave astrophysics. We integrate the Jim inference toolkit, a normalizing flow-enhanced Markov chain Monte Carlo (MCMC) sampler, with the learned harmonic mean estimator. Our Bayesian evidence estimates run on 1 GPU are consistent with traditional nested sampling techniques run on 16 CPU cores, while reducing the computation time by factors of 5× and 15× for 4-dimensional and 11-dimensional gravitational wave inference problems, respectively. Our code is available in well-tested and thoroughly documented open-source packages, ensuring accessibility and reproducibility for the wider research community.
@article{arxiv.2410.21076,
title = {Accelerated Bayesian parameter estimation and model selection for gravitational waves with normalizing flows},
author = {Alicja Polanska and Thibeau Wouters and Peter T. H. Pang and Kaze K. W. Wong and Jason D. McEwen},
journal= {arXiv preprint arXiv:2410.21076},
year = {2024}
}
Comments
accepted to NeurIPS 2024 workshop on Machine Learning and the Physical Sciences