On Improving the Performance of Glitch Classification for Gravitational Wave Detection by using Generative Adversarial Networks

Abstract

Spectrogram classification plays an important role in analyzing gravitational wave data. In this paper, we propose a framework to improve the classification performance by using Generative Adversarial Networks (GANs). As substantial efforts and expertise are required to annotate spectrograms, the number of training examples is very limited. However, it is well known that deep networks can perform well only when the sample size of the training set is sufficiently large. Furthermore, the imbalanced sample sizes in different classes can also hamper the performance. In order to tackle these problems, we propose a GAN-based data augmentation framework. While standard data augmentation methods for conventional images cannot be applied on spectrograms, we found that a variant of GANs, ProGAN, is capable of generating high-resolution spectrograms which are consistent with the quality of the high-resolution original images and provide a desirable diversity. We have validated our framework by classifying glitches in the {\it Gravity Spy} dataset with the GAN-generated spectrograms for training. We show that the proposed method can provide an alternative to transfer learning for the classification of spectrograms using deep networks, i.e. using a high-resolution GAN for data augmentation instead. Furthermore, fluctuations in classification performance with small sample sizes for training and evaluation can be greatly reduced. Using the trained network in our framework, we have also examined the spectrograms with label anomalies in {\it Gravity Spy}.

arXiv commentAccepted for publication in MNRAS, 16 pages, 14 figures, 5 tables

References in corpus (16)

• Observation of Gravitational Waves from a Binary Black Hole Merger
• Advanced Virgo: a 2nd generation interferometric gravitational wave detector
• Advanced LIGO
• Rotation-invariant convolutional neural networks for galaxy morphology prediction
• Gravity Spy: Integrating Advanced LIGO Detector Characterization, Machine Learning, and Citizen Science
• Multiresolution techniques for the detection of gravitational-wave bursts
• Deep Transfer Learning: A new deep learning glitch classification method for advanced LIGO
• Glitch Classification and Clustering for LIGO with Deep Transfer Learning
• Stochastic Nature of Gravitational Waves from Supernova Explosions with Standing Accretion Shock Instability
• Transfer learning for galaxy morphology from one survey to another
• Detection and Classification of Supernova Gravitational Waves Signals: A Deep Learning Approach
• Stellar Mass Black Hole Formation and Multi-messenger Signals from Three Dimensional Rotating Core-Collapse Supernova Simulations
• Deep Transfer Learning for Star Cluster Classification: I. Application to the PHANGS-HST Survey
• Deep Learning at Scale for the Construction of Galaxy Catalogs in the Dark Energy Survey
• Neutron star collapse and gravitational waves with a non-convex equation of state
• Generalised gravitational burst generation with Generative Adversarial Networks

Cited by in corpus (8)

• Data quality up to the third observing run of Advanced LIGO: Gravity Spy glitch classifications
• Gravity Spy: Lessons Learned and a Path Forward
• Generating transient noise artifacts in gravitational-wave detector data with generative adversarial networks
• cDVGAN: One Flexible Model for Multi-class Gravitational Wave Signal and Glitch Generation
• Modeling the Time Evolution of Compact Binary Systems with Machine Learning
• DVGAN: Stabilize Wasserstein GAN training for time-domain Gravitational Wave physics
• Extract non-Gaussian Features in Gravitational Wave Observation Data Using Self-Supervised Learning
• Unveiling gravitational waves from core-collapse supernovae with MUSE