Gravitational-wave parameter inference using Deep Learning

We explore machine learning methods to detect gravitational waves (GW) from binary black hole (BBH) mergers using deep learning (DL) algorithms. The DL networks are trained with gravitational waveforms obtained from BBH mergers with component masses randomly sampled in the range from 5 to 100 solar masses and luminosity distances from 100 Mpc to, at least, 2000 Mpc. The GW signal waveforms are injected in public data from the O2 run of the Advanced LIGO and Advanced Virgo detectors, in time windows that do not coincide with those of known detected signals, and the data from each detector in the Advanced LIGO and Advanced Virgo network is combined into a unique RGB image. We show that a classifier network can be trained in order to detect the presence of GW signal with high accuracy. Furthermore, we train a regression network to perform parameter inference on BBH spectrogram data. Without significant optimization of our algorithms we manage to corroborate most of the BBH detections in the GWTC-1 and GWTC-2 catalogs, and obtain parameter inference results that are mostly consistent with published results by the LIGO-Virgo Collaboration in GWTC-1. In particular, our predictions for the chirp mass are compatible (up to 3σ) with the official values for 90% of events.We thank Nicolás Sanchis-Gual for fruitful discussions that allowed the setup of the team involved in this project. This work was supported by the Spanish Agencia Estatal de Investigación (PGC2018-095984-B-I00), by the Generalitat Valenciana (PROMETEO/2019/071), by the EU’s Horizon 2020 research and innovation (RISE) programme (H2020-MSCA-RISE-2017 Grant No. FunFiCO-777740) and by the Portuguese Foundation for Science and Technology (FCT), project CERN/FIS-PAR/0029/2019. APM and FFF are supported by the FCT project PTDC/FIS-PAR/31000/2017 and by CIDMA through FCT, references UIDB/04106/2020 and UIDP/04106/2020. APM is also supported by the projects CERN/FIS-PAR/0027/2019, CERN/FISPAR/0002/2017 and by national funds (OE), through FCT, I.P., in the scope of the framework contract foreseen in the numbers 4, 5 and 6 of the article 23, of the Decree-Law 57/2016, of August 29, changed by Law 57/2017, of July 19.