Comparative study of 1D and 2D convolutional neural network models with attribution analysis for gravitational wave detection from compact binary coalescences
arXiv:2312.04855 · doi:10.1103/PhysRevD.109.043011
Abstract
Recent advancements in gravitational wave astronomy have seen the application of convolutional neural networks (CNNs) in signal detection from compact binary coalescences. This study presents a comparative analysis of two CNN architectures: one-dimensional (1D) and two-dimensional (2D) along with an ensemble model combining both. We trained these models to detect gravitational wave signals from binary black hole (BBH) mergers, neutron star-black hole (NSBH) mergers, and binary neutron star (BNS) mergers within real detector noise. Our investigation entailed a comprehensive evaluation of the detection performance of each model type across different signal classes. To understand the models' decision-making processes, we employed feature map visualization and attribution analysis. The findings revealed that while the 1D model showed superior performance in detecting BBH signals, the 2D model excelled in identifying NSBH and BNS signals. Notably, the ensemble model outperformed both individual models across all signal types, demonstrating enhanced detection capabilities. Additionally, input feature visualization indicated distinct areas of focus in the data for the 1D and 2D models, emphasizing the effectiveness of their combination.
arXiv comment12 pages, 9 figures
References in corpus (50)
- Observation of Gravitational Waves from a Binary Black Hole Merger
- GW170817: Observation of Gravitational Waves from a Binary Neutron Star Inspiral
- Advanced Virgo: a 2nd generation interferometric gravitational wave detector
- Multi-messenger Observations of a Binary Neutron Star Merger
- GWTC-1: A Gravitational-Wave Transient Catalog of Compact Binary Mergers Observed by LIGO and Virgo during the First and Second Observing Runs
- Advanced LIGO
- GWTC-2: Compact Binary Coalescences Observed by LIGO and Virgo During the First Half of the Third Observing Run
- GWTC-3: Compact Binary Coalescences Observed by LIGO and Virgo During the Second Part of the Third Observing Run
- GW190425: Observation of a Compact Binary Coalescence with Total Mass
- Observation of gravitational waves from two neutron star-black hole coalescences
- FINDCHIRP: an algorithm for detection of gravitational waves from inspiraling compact binaries
- An improved effective-one-body model of spinning, nonprecessing binary black holes for the era of gravitational-wave astrophysics with advanced detectors
- KAGRA: 2.5 Generation Interferometric Gravitational Wave Detector
- Deep Learning for Real-time Gravitational Wave Detection and Parameter Estimation: Results with Advanced LIGO Data
- Gravity Spy: Integrating Advanced LIGO Detector Characterization, Machine Learning, and Citizen Science
- Open data from the third observing run of LIGO, Virgo, KAGRA and GEO
- Deep Neural Networks to Enable Real-time Multimessenger Astrophysics
- Matching matched filtering with deep networks in gravitational-wave astronomy
- Multiresolution techniques for the detection of gravitational-wave bursts
- Bayesian parameter estimation using conditional variational autoencoders for gravitational-wave astronomy
- Convolutional neural networks: a magic bullet for gravitational-wave detection?
- Real-Time Detection of Gravitational Waves from Binary Neutron Stars using Artificial Neural Networks
- Neural Importance Sampling for Rapid and Reliable Gravitational-Wave Inference
- Gravitational wave signal recognition of O1 data by deep learning
- Detection of gravitational-wave signals from binary neutron star mergers using machine learning
- MLGWSC-1: The first Machine Learning Gravitational-Wave Search Mock Data Challenge
- Applying deep neural networks to the detection and space parameter estimation of compact binary coalescence with a network of gravitational wave detectors
- Detection and Parameter Estimation of Gravitational Waves from Binary Neutron-Star Mergers in Real LIGO Data using Deep Learning
- Deep Learning Ensemble for Real-time Gravitational Wave Detection of Spinning Binary Black Hole Mergers
- Deep learning for gravitational wave forecasting of neutron star mergers
- Training Strategies for Deep Learning Gravitational-Wave Searches
- Convolutional neural networks for the detection of the early inspiral of a gravitational-wave signal
- Detection of Gravitational Waves Using Bayesian Neural Networks
- Deep Residual Networks for Gravitational Wave Detection
- Complete phenomenological gravitational waveforms from spinning coalescing binaries
- Using Deep Learning to Localize Gravitational Wave Sources
- Improved deep learning techniques in gravitational-wave data analysis
- Early warning of coalescing neutron-star and neutron-star-black-hole binaries from nonstationary noise background using neural networks
- From One to Many: A Deep Learning Coincident Gravitational-Wave Search
- Deep Learning Detection and Classification of Gravitational Waves from Neutron Star-Black Hole Mergers
- Ensemble of Deep Convolutional Neural Networks for real-time gravitational wave signal recognition
- LSTM and CNN application for core-collapse supernova search in gravitational wave real data
- Deep Learning with Quantized Neural Networks for Gravitational Wave Forecasting of Eccentric Compact Binary Coalescence
- Searches for Compact Binary Coalescence Events using Neural Networks in LIGO/Virgo Second Observation Period
- Identification of Binary Neutron Star Mergers in Gravitational-Wave Data Using YOLO One-Shot Object Detection
- Rapid search for massive black hole binary coalescences using deep learning
- Unsupervised Learning Architecture for Classifying the Transient Noise of Interferometric Gravitational-wave Detectors
- Swift sky localization of gravitational waves using deep learning seeded importance sampling
- Localization of gravitational waves using machine learning
- Visualizing convolutional neural network for classifying gravitational waves from core-collapse supernovae