Identification of binary neutron star mergers in gravitational-wave data using object-detection machine learning models

João Aveiro, Felipe F. Freitas, Márcio Ferreira, Antonio Onofre, Constança Providência, Gonçalo Gonçalves, and José A. Font
Phys. Rev. D 106, 084059 – Published 28 October 2022

Abstract

We demonstrate the application of the YOLOv5 model, a general purpose convolution-based object detection model, in the task of detecting binary neutron star coalescence events from gravitational-wave data of current generation interferometer detectors. We also present a thorough explanation of the synthetic data generation and preparation tasks based on approximant waveform models used for the model training, validation and testing steps. Using this approach, we achieve mean average precision [0.50] values of 0.945 for a single class validation dataset and as high as 0.978 for test datasets. Moreover, the trained model is successful in identifying the GW170817 event in the LIGO H1 detector data. The identification of this event is also possible for the LIGO L1 detector data with an additional preprocessing step, without the need of removing the large glitch in the final stages of the inspiral. The detection of the GW190425 event is less successful, which attests to performance degradation with the signal-to-noise ratio. Our study indicates that the YOLOv5 model is an interesting approach for first-stage detection alarm pipelines.

  • Figure
  • Figure
  • Figure
  • Figure
  • Figure
  • Figure
  • Figure
3 More
  • Received 1 July 2022
  • Revised 19 August 2022
  • Accepted 28 September 2022

DOI:https://doi.org/10.1103/PhysRevD.106.084059

© 2022 American Physical Society

Physics Subject Headings (PhySH)

Gravitation, Cosmology & Astrophysics

Authors & Affiliations

João Aveiro1,*, Felipe F. Freitas2,†, Márcio Ferreira1,‡, Antonio Onofre3,∥, Constança Providência1,§, Gonçalo Gonçalves4,¶, and José A. Font5,6,**

  • 1CFisUC, Department of Physics, University of Coimbra, 3004-516 Coimbra, Portugal
  • 2Departamento de Física da Universidade de Aveiro and Centre for Research and Development in Mathematics and Applications (CIDMA), Campus de Santiago, 3810-183 Aveiro, Portugal
  • 3Centro de Física das Universidades do Minho e do Porto (CF-UM-UP), Universidade do Minho, 4710-057 Braga, Portugal
  • 4LIP, Department of Physics, University of Coimbra, 3004-516 Coimbra, Portugal
  • 5Departamento de Astronomía y Astrofísica, Universitat de València, Dr. Moliner 50, 46100, Burjassot (València), Spain
  • 6Observatori Astronòmic, Universitat de València, Catedrático José Beltrán 2, 46980, Paterna (València), Spain

  • *joao@aveiro.me
  • felipefreitas@ua.pt
  • marcio.ferreira@uc.pt
  • antonio.onofre@cern.ch
  • §cp@uc.pt
  • goncalo.mota.goncalves@gmail.com
  • **j.antonio.font@uv.es

Article Text (Subscription Required)

Click to Expand

References (Subscription Required)

Click to Expand
Issue

Vol. 106, Iss. 8 — 15 October 2022

Reuse & Permissions
Access Options
Author publication services for translation and copyediting assistance advertisement

Authorization Required


×
×

Images

×

Sign up to receive regular email alerts from Physical Review D

Log In

Cancel
×

Search


Article Lookup

Paste a citation or DOI

Enter a citation
×