arrow
Return

Statistically-informed deep learning for gravitational wave parameter estimation

delete2021-11-30
delete27
delete
OA
AI
H
Hongyu Shen
E
E. A. Huerta *
E
E. O’Shea
P
P. Kumar
Z
Zhizhen Zhao *
DOI:10.1088/2632-2153/ac3843delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
We introduce deep learning models to estimate the masses of the binary components of black hole mergers, (m(1), m(2)), and three astrophysical properties of the post-merger compact remnant, namely, the final spin, a(f), and the frequency and damping time of the ringdown oscillations of the fundamental l = m = 2 bar mode, (omega(R), omega(1)). Our neural networks combine a modified WaveNet architecture with contrastive learning and normalizing flow. We validate these models against a Gaussian conjugate prior family whose posterior distribution is described by a closed analytical expression. Upon confirming that our models produce statistically consistent results, we used them to estimate the astrophysical parameters (m(1), m(2), a(f), omega(R), omega(I)) of five binary black holes: GW150914, GW170104, GW170814, GW190521 and GW190630. We use PyCBC Inference to directly compare traditional Bayesian methodologies for parameter estimation with our deep learning based posterior distributions. Our results show that our neural network models predict posterior distributions that encode physical correlations, and that our data-driven median results and 90% confidence intervals are similar to those produced with gravitational wave Bayesian analyses. This methodology requires a single V100 NVIDI A GPU to produce median values and posterior distributions within two milliseconds for each event. This neural network, and a tutorial for its use, are available at the Data and Learning Hub for Science.
Keywords:
gravitational wave
deep learning
parameter estimation
posterior inference

Journal

M
Machine Learning-Science and Technology
IF:
4.6
Papers:
1.1K
Citations:
3.4K

Organization

U
united states department of energy (doe)
Scholars:
11.3W
Papers: 9.6W
Citations: 246
University of Illinois System cover
University of Illinois System
Scholars:
6.8W
Papers: 6.2W
Citations: 644