Convolutional neural network classifier for the output of the time-domain F-statistic

Our recent work is a study on the feasibility of application of 1D and 2D convolutional neural networks in classification of candidate signal distributions produced by the time-domain $\mathcal{F}$-statistic search (all-sky implementation used in our group):

Among the astrophysical sources in the Advanced Laser Interferometer Gravitational-Wave Observatory (LIGO) and Advanced Virgo detectors’ frequency band are rotating non-axisymmetric neutron stars emitting long-lasting, almost-monochromatic gravitational waves. Searches for these continuous gravitational-wave signals are usually performed in long stretches of data in a matched-filter framework e.g. the $\mathcal{F}$-statistic method. In an all-sky search for a priori unknown sources, a large number of templates are matched against the data using a pre-defined grid of variables (the gravitational-wave frequency and its derivatives, sky coordinates), subsequently producing a collection of candidate signals, corresponding to the grid points at which the signal reaches a pre-defined signal-to-noise threshold. An astrophysical signature of the signal is encoded in the multi-dimensional vector distribution of the candidate signals. In the first work of this kind, we apply a deep learning approach to classify the distributions. We consider three basic classes: Gaussian noise, astrophysical gravitational-wave signal, and a constant-frequency detector artifact (‘stationary line’), the two latter injected into the Gaussian noise. 1D and 2D versions of a convolutional neural network classifier are implemented, trained and tested on a broad range of signal frequencies. We demonstrate that these implementations correctly classify the instances of data at various signal-to-noise ratios and signal frequencies, while also showing concept generalization i.e. satisfactory performance at previously unseen frequencies. In addition we discuss the deficiencies, computational requirements and possible applications of these implementations.

Confusion matrix for the three-label classification evaluated on the test set for the 1D CNN (a) and 2D CNN (b) after the training. Although the <b>cgw</b> and the <b>noise</b> were classified on a similar level, the <b>line</b> caused significant problem for the 2D model. The majority of <b>line</b> instances resembled <b>noise</b> class in the image representation.
Confusion matrix for the three-label classification evaluated on the test set for the 1D CNN (a) and 2D CNN (b) after the training. Although the cgw and the noise were classified on a similar level, the line caused significant problem for the 2D model. The majority of line instances resembled noise class in the image representation.

Update 11.05.2020: accepted by the MLST!

Updated: