Using deep learning to localize gravitational wave sources

Chayan Chatterjee, Linqing Wen, Kevin Vinsen, Manoj Kovalam Lakshmi Narasimha, Amitava Datta

Research output: Contribution to journalArticle

1 Citation (Scopus)

Abstract

Deep learning algorithms, in particular neural networks, have been steadily gaining popularity among the gravitational wave community for the last few years. The reliability and accuracy of deep learning approaches in gravitational wave detection, parameter estimation, and glitch classification have already been proved and verified by several groups in recent years. In this paper, we report on the construction of the deep artificial neural network (ANN) to localize simulated gravitational wave signals in the sky with high accuracy. We have modeled the sky as a sphere and have considered cases in which the sphere is divided into 18, 50, 128, 1024, 2048, and 4096 sectors. The sky direction of the gravitational wave source is estimated by classifying the signal into one of these sectors based on its right ascension and declination values for each of these cases. To do this, we have injected simulated binary black hole gravitational wave signals of component masses sampled uniformly between
30


M

and
80


M

into Gaussian noise and used the whitened strain values to obtain the input features for training our ANN. We input features such as the delays in arrival times, phase differences, and amplitude ratios at each of the three detectors Hanford, Livingston, and Virgo, from the raw time-domain strain values as well as from analytical versions of these signals, obtained through Hilbert transformation. We show that our model is able to classify gravitational wave samples, not used in the training process, into their correct sectors with very high accuracy (greater than 90%) for coarse angular resolution using 18, 50, and 128 sectors. We also test our localization on test samples with injection parameters of the published LIGO binary black hole merger events GW150914, GW170818, and GW170823 for 1024, 2048, and 4096 sectors and compare the result with that from BAYESTAR and parameter estimation. In addition, we report that the time taken by our model to localize one gravitational wave signal is around 0.018 s on 14 Intel Xeon CPU cores.
Original languageEnglish
Article number103025
Number of pages17
JournalPhysical Review D - Particles, Fields, Gravitation and Cosmology
Volume100
Issue number10
Publication statusPublished - 26 Nov 2019

Fingerprint Dive into the research topics of 'Using deep learning to localize gravitational wave sources'. Together they form a unique fingerprint.

  • Projects

    ARC Centre of Excellence for Gravitational Wave Discovery

    Blair, D., Wen, L., Zhao, C., Ju, L., Coward, D., Bailes, M., McClelland, D. E., Levin, Y., Scott, S., Ottaway, D., Melatos, A. & Veitch, P.

    Australian Research Council

    1/01/1731/12/23

    Project: Research

    Cite this