Rapid parameter estimation for pulsar-timing-array datasets with variational inference and normalizing flows

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Vallisneri, Michele, Crisostomi, Marco, Johnson, Aaron D., Meyers, Patrick M.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910447900295168
author Vallisneri, Michele
Crisostomi, Marco
Johnson, Aaron D.
Meyers, Patrick M.
author_facet Vallisneri, Michele
Crisostomi, Marco
Johnson, Aaron D.
Meyers, Patrick M.
contents In the gravitational-wave analysis of pulsar-timing-array datasets, parameter estimation is usually performed using Markov Chain Monte Carlo methods to explore posterior probability densities. We introduce an alternative procedure that relies instead on stochastic gradient-descent Bayesian variational inference, whereby we obtain the weights of a neural-network approximation of the posterior by minimizing the Kullback-Leibler divergence of the approximation from the exact posterior. This technique is distinct from simulation-based inference with normalizing flows, since we train the network for a single dataset, rather than the population of all possible datasets, and we require the computation of the data likelihood and its gradient. Unlike Markov Chain methods, our technique can transparently exploit highly parallel computing platforms. This makes it extremely fast on modern graphical processing units, where it can analyze the NANOGrav 15-yr dataset in few tens of minutes, depending on the probabilistic model, as opposed to hours or days with the analysis codes used until now. We expect that this speed will unlock new kinds of astrophysical and cosmological studies of pulsar-timing-array datasets. Furthermore, variational inference would be viable in other contexts of gravitational-wave data analysis as long as differentiable and parallelizable likelihoods are available.
format Preprint
id arxiv_https___arxiv_org_abs_2405_08857
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Rapid parameter estimation for pulsar-timing-array datasets with variational inference and normalizing flows
Vallisneri, Michele
Crisostomi, Marco
Johnson, Aaron D.
Meyers, Patrick M.
General Relativity and Quantum Cosmology
High Energy Astrophysical Phenomena
Instrumentation and Methods for Astrophysics
In the gravitational-wave analysis of pulsar-timing-array datasets, parameter estimation is usually performed using Markov Chain Monte Carlo methods to explore posterior probability densities. We introduce an alternative procedure that relies instead on stochastic gradient-descent Bayesian variational inference, whereby we obtain the weights of a neural-network approximation of the posterior by minimizing the Kullback-Leibler divergence of the approximation from the exact posterior. This technique is distinct from simulation-based inference with normalizing flows, since we train the network for a single dataset, rather than the population of all possible datasets, and we require the computation of the data likelihood and its gradient. Unlike Markov Chain methods, our technique can transparently exploit highly parallel computing platforms. This makes it extremely fast on modern graphical processing units, where it can analyze the NANOGrav 15-yr dataset in few tens of minutes, depending on the probabilistic model, as opposed to hours or days with the analysis codes used until now. We expect that this speed will unlock new kinds of astrophysical and cosmological studies of pulsar-timing-array datasets. Furthermore, variational inference would be viable in other contexts of gravitational-wave data analysis as long as differentiable and parallelizable likelihoods are available.
title Rapid parameter estimation for pulsar-timing-array datasets with variational inference and normalizing flows
topic General Relativity and Quantum Cosmology
High Energy Astrophysical Phenomena
Instrumentation and Methods for Astrophysics
url https://arxiv.org/abs/2405.08857