GWSkyNet-Multi II: an updated machine learning model for rapid classification of gravitational-wave events

TL;DR

GWSkyNet-Multi II is an improved machine learning model that rapidly classifies gravitational-wave events during O4, aiding follow-up decisions by providing robust, interpretable predictions with uncertainty estimates.

Contribution

The paper introduces GWSkyNet-Multi II, a simplified, more robust version of the original model that offers probabilistic classifications and uncertainties for gravitational-wave event categories.

Findings

93% consistency with LVK classifications for significant events

Provides normalized probability scores and uncertainties

Simplified architecture with more interpretable inputs

Abstract

Multi-messenger observations of gravitational waves and electromagnetic emission from compact object mergers offer unique insights into the structure of neutron stars, the formation of heavy elements, and the expansion rate of the Universe. With the LIGO-Virgo-KAGRA (LVK) gravitational-wave detectors currently in their fourth observing run (O4), it is an exciting time for detecting these mergers. However, assessing whether to follow up a candidate gravitational-wave event given limited telescope time and resources is challenging; the candidate can be a false alert due to detector glitches, or may not have any detectable electromagnetic counterpart even if it is real. GWSkyNet-Multi is a machine learning model developed to facilitate follow-up decisions by providing real-time classification of candidate events, using localization information released in LVK rapid public alerts. Here we introduce GWSkyNet-Multi II, an updated model targeted towards providing more robust and informative predictions during O4 and beyond. Specifically, the model now provides normalized probability scores and associated uncertainties for each of the four corresponding source categories released by the LVK: glitch, binary black hole, neutron star-black hole, and binary neutron star. Informed by explainability studies of the original model, the updated model architecture is also significantly simplified, including replacing input images with intuitive summary values that are more interpretable. For significant event alerts issued during O4a and O4b, GWSkyNet-Multi II produces a prediction that is consistent with the updated LVK classification for 93% of events. The updated model can be used by the community to help make time-critical follow-up decisions.