Training Strategies for Deep Learning Gravitational-Wave Searches

by   Marlin B. Schäfer, et al.

Compact binary systems emit gravitational radiation which is potentially detectable by current Earth bound detectors. Extracting these signals from the instruments' background noise is a complex problem and the computational cost of most current searches depends on the complexity of the source model. Deep learning may be capable of finding signals where current algorithms hit computational limits. Here we restrict our analysis to signals from non-spinning binary black holes and systematically test different strategies by which training data is presented to the networks. To assess the impact of the training strategies, we re-analyze the first published networks and directly compare them to an equivalent matched-filter search. We find that the deep learning algorithms can generalize low signal-to-noise ratio (SNR) signals to high SNR ones but not vice versa. As such, it is not beneficial to provide high SNR signals during training, and fastest convergence is achieved when low SNR samples are provided early on. During testing we found that the networks are sometimes unable to recover any signals when a false alarm probability <10^-3 is required. We resolve this restriction by applying a modification we call unbounded Softmax replacement (USR) after training. With this alteration we find that the machine learning search retains ≥ 97.5% of the sensitivity of the matched-filter search down to a false-alarm rate of 1 per month.



There are no comments yet.


page 1

page 2

page 3

page 4


Detection of gravitational-wave signals from binary neutron star mergers using machine learning

As two neutron stars merge, they emit gravitational waves that can poten...

Deep learning for clustering of continuous gravitational wave candidates II: identification of low-SNR candidates

Broad searches for continuous gravitational wave signals rely on hierarc...

Improving significance of binary black hole mergers in Advanced LIGO data using deep learning : Confirmation of GW151216

We present a novel Machine Learning (ML) based strategy to search for co...

Deep learning for gravitational-wave data analysis: A resampling white-box approach

In this work, we apply Convolutional Neural Networks (CNNs) to detect gr...

From One to Many: A Deep Learning Coincident Gravitational-Wave Search

Gravitational waves from the coalescence of compact-binary sources are n...

Efficient Training of Deep Classifiers for Wireless Source Identification using Test SNR Estimates

We investigate the potential of training time reduction for deep learnin...

On Filter Generalization for Music Bandwidth Extension Using Deep Neural Networks

In this paper, we address a sub-topic of the broad domain of audio enhan...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.