English

Accelerated Bayesian parameter estimation and model selection for gravitational waves with normalizing flows

Instrumentation and Methods for Astrophysics 2024-11-01 v2 High Energy Astrophysical Phenomena Machine Learning General Relativity and Quantum Cosmology

Abstract

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 11 GPU are consistent with traditional nested sampling techniques run on 1616 CPU cores, while reducing the computation time by factors of 5×5\times and 15×15\times for 44-dimensional and 1111-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.

Keywords

Cite

@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