Detection of Gravitational Waves Using Bayesian Neural Networks

Abstract

We propose a new model of Bayesian Neural Networks to not only detect the events of compact binary coalescence in the observational data of gravitational waves (GW) but also identify the full length of the event duration including the inspiral stage. This is achieved by incorporating the Bayesian approach into the CLDNN classifier, which integrates together the Convolutional Neural Network (CNN) and the Long Short-Term Memory Recurrent Neural Network (LSTM). Our model successfully detect all seven BBH events in the LIGO Livingston O2 data, with the periods of their GW waveforms correctly labeled. The ability of a Bayesian approach for uncertainty estimation enables a newly defined `awareness' state for recognizing the possible presence of signals of unknown types, which is otherwise rejected in a non-Bayesian model. Such data chunks labeled with the awareness state can then be further investigated rather than overlooked. Performance tests with 40,960 training samples against 512 chunks of 8-second real noise mixed with mock signals of various optimal signal-to-noise ratio $0\le {\rho }_{\text{opt}}\le 18$ show that our model recognizes 90% of the events when ${\rho }_{\text{opt}}>7$ (100% when ${\rho }_{\text{opt}}>8.5$) and successfully labels more than 95% of the waveform periods when ${\rho }_{\text{opt}}>8$. The latency between the arrival of peak signal and generating an alert with the associated waveform period labeled is only about 20 seconds for an unoptimized code on a moderate GPU-equipped personal computer. This makes our model possible for nearly real-time detection and for forecasting the coalescence events when assisted with deeper training on a larger dataset using the state-of-art HPCs.

arXiv comment17 pages, 13 figures

References in corpus (65)

• Adam: A Method for Stochastic Optimization
• Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
• Auto-Encoding Variational Bayes
• Observation of Gravitational Waves from a Binary Black Hole Merger
• GW170817: Observation of Gravitational Waves from a Binary Neutron Star Inspiral
• Intriguing properties of neural networks
• Advanced Virgo: a 2nd generation interferometric gravitational wave detector
• WaveNet: A Generative Model for Raw Audio
• Gravitational Waves and Gamma-rays from a Binary Neutron Star Merger: GW170817 and GRB 170817A
• 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
• GW151226: Observation of Gravitational Waves from a 22-Solar-Mass Binary Black Hole Coalescence
• What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?
• Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning
• GW170104: Observation of a 50-Solar-Mass Binary Black Hole Coalescence at Redshift 0.2
• GW170814: A Three-Detector Observation of Gravitational Waves from a Binary Black Hole Coalescence
• GW190814: Gravitational Waves from the Coalescence of a 23 M${}_{\odot }$ Black Hole with a 2.6 M${}_{\odot }$ Compact Object
• On Calibration of Modern Neural Networks
• GW190425: Observation of a Compact Binary Coalescence with Total Mass $\sim 3.4M_{\odot }$
• An Ordinary Short Gamma-Ray Burst with Extraordinary Implications: Fermi-GBM Detection of GRB 170817A
• GW190521: A Binary Black Hole Merger with a Total Mass of $150M_{\odot }$
• Weight Uncertainty in Neural Networks
• GW170608: Observation of a 19-solar-mass Binary Black Hole Coalescence
• Binary Black Hole Mergers in the first Advanced LIGO Observing Run
• INTEGRAL Detection of the First Prompt Gamma-Ray Signal Coincident with the Gravitational Wave Event GW170817
• Physics, Astrophysics and Cosmology with Gravitational Waves
• GW190412: Observation of a Binary-Black-Hole Coalescence with Asymmetric Masses
• The PyCBC search for gravitational waves from compact binary coalescence
• 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
• What is the Most Promising Electromagnetic Counterpart of a Neutron Star Binary Merger?
• Matched filtering of gravitational waves from inspiraling compact binaries: Computational cost and template placement
• Analysis Framework for the Prompt Discovery of Compact Binary Mergers in Gravitational-wave Data
• High-accuracy comparison of numerical relativity simulations with post-Newtonian expansions
• 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
• Coherent method for detection of gravitational wave bursts
• First measurement of the Hubble constant from a dark standard siren using the Dark Energy Survey galaxies and the LIGO/Virgo binary-black-hole merger GW170814
• Deep Neural Networks to Enable Real-time Multimessenger Astrophysics
• The LIGO Open Science Center
• Toward Early-Warning Detection of Gravitational Waves from Compact Binary Coalescence
• Matching matched filtering with deep networks in gravitational-wave astronomy
• TensorFlow Distributions
• Bayesian parameter estimation using conditional variational autoencoders for gravitational-wave astronomy
• Low-latency analysis pipeline for compact binary coalescences in the advanced gravitational wave detector era
• Flipout: Efficient Pseudo-Independent Weight Perturbations on Mini-Batches
• Deep Transfer Learning: A new deep learning glitch classification method for advanced LIGO
• Glitch Classification and Clustering for LIGO with Deep Transfer Learning
• Constraints on cosmic strings using data from the first Advanced LIGO observing run
• Convolutional neural networks: a magic bullet for gravitational-wave detection?
• Bayesian Recurrent Neural Networks
• Mitigation of the instrumental noise transient in gravitational-wave data surrounding GW170817
• Real-Time Detection of Gravitational Waves from Binary Neutron Stars using Artificial Neural Networks
• Low-Latency Gravitational Wave Alerts for Multi-Messenger Astronomy During the Second Advanced LIGO and Virgo Observing Run
• Gravitational Wave Denoising of Binary Black Hole Mergers with Deep Learning
• Noise Reduction in Gravitational-wave Data via Deep Learning
• High-energy electromagnetic offline follow-up of LIGO-Virgo gravitational-wave binary coalescence candidate events
• Optimizing searches for electromagnetic counterparts of gravitational wave triggers
• Applying deep neural networks to the detection and space parameter estimation of compact binary coalescence with a network of gravitational wave detectors
• Classifying the unknown: discovering novel gravitational-wave detector glitches using similarity learning
• Binary Neutron Stars Gravitational Wave Detection Based on Wavelet Packet Analysis And Convolutional Neural Networks
• Maximising the detection probability of kilonovae associated with gravitational wave observations
• Classification of gravitational-wave glitches via dictionary learning
• Astrophysical Prior Information and Gravitational-wave Parameter Estimation
• Time Series Prediction : Predicting Stock Price

Cited by in corpus (27)

• Improving the open cluster census. II. An all-sky cluster catalogue with Gaia DR3
• Accelerated, Scalable and Reproducible AI-driven Gravitational Wave Detection
• Detection and Parameter Estimation of Gravitational Waves from Binary Neutron-Star Mergers in Real LIGO Data using Deep Learning
• Deep Residual Networks for Gravitational Wave Detection
• Translating neutron star observations to nuclear symmetry energy via artificial neural networks
• Deep Learning Detection and Classification of Gravitational Waves from Neutron Star-Black Hole Mergers
• Machine learning forecasts of the cosmic distance duality relation with strongly lensed gravitational wave events
• Generalized Approach to Matched Filtering using Neural Networks
• Inference-optimized AI and high performance computing for gravitational wave detection at scale
• Identification of Binary Neutron Star Mergers in Gravitational-Wave Data Using YOLO One-Shot Object Detection
• Variational Inference as an alternative to MCMC for parameter estimation and model selection
• New Gravitational Wave Discoveries Enabled by Machine Learning
• Decoding Neutron Star Observations: Revealing Composition through Bayesian Neural Networks
• Accelerating Recurrent Neural Networks for Gravitational Wave Experiments
• LSST Cadence Strategy Evaluations for AGN Time-series Data in Wide-Fast-Deep Field
• Interpretable AI forecasting for numerical relativity waveforms of quasi-circular, spinning, non-precessing binary black hole mergers
• The Gaia-ESO Survey: Preparing the ground for 4MOST & WEAVE galactic surveys. Chemical evolution of lithium with machine learning
• Advances in Machine and Deep Learning for Modeling and Real-time Detection of Multi-Messenger Sources
• Comparative study of 1D and 2D convolutional neural network models with attribution analysis for gravitational wave detection from compact binary coalescences
• Inferring the Langevin Equation with Uncertainty via Bayesian Neural Networks
• AI and extreme scale computing to learn and infer the physics of higher order gravitational wave modes of quasi-circular, spinning, non-precessing binary black hole mergers
• Novel Deep Learning Approach to Detecting Binary Black Hole Mergers
• Physics-inspired spatiotemporal-graph AI ensemble for the detection of higher order wave mode signals of spinning binary black hole mergers
• Boosting the Efficiency of Parametric Detection with Hierarchical Neural Networks
• Genetic-algorithm-optimized neural networks for gravitational wave classification
• Extract non-Gaussian Features in Gravitational Wave Observation Data Using Self-Supervised Learning
• Bayesian classification of astronomical spectra with class uncertainties