Physics-inspired spatiotemporal-graph AI ensemble for the detection of higher order wave mode signals of spinning binary black hole mergers
arXiv:2306.15728 · doi:10.1088/2632-2153/ad4c37
Abstract
We present a new class of AI models for the detection of quasi-circular, spinning, non-precessing binary black hole mergers whose waveforms include the higher order gravitational wave modes , and mode mixing effects in the harmonics. These AI models combine hybrid dilated convolution neural networks to accurately model both short- and long-range temporal sequential information of gravitational waves; and graph neural networks to capture spatial correlations among gravitational wave observatories to consistently describe and identify the presence of a signal in a three detector network encompassing the Advanced LIGO and Virgo detectors. We first trained these spatiotemporal-graph AI models using synthetic noise, using 1.2 million modeled waveforms to densely sample this signal manifold, within 1.7 hours using 256 A100 GPUs in the Polaris supercomputer at the ALCF. Our distributed training approach had optimal performance, and strong scaling up to 512 A100 GPUs. With these AI ensembles we processed data from a three detector network, and found that an ensemble of 4 AI models achieves state-of-the-art performance for signal detection, and reports two misclassifications for every decade of searched data. We distributed AI inference over 128 GPUs in the Polaris supercomputer and 128 nodes in the Theta supercomputer, and completed the processing of a decade of gravitational wave data from a three detector network within 3.5 hours. Finally, we fine-tuned these AI ensembles to process the entire month of February 2020, which is part of the O3b LIGO/Virgo observation run, and found 6 gravitational waves, concurrently identified in Advanced LIGO and Advanced Virgo data, and zero false positives. This analysis was completed in one hour using one A100 GPU.
14 pages, 6 figures, and 3 tables
References in corpus (44)
- Neural Message Passing for Quantum Chemistry
- Multi-messenger Observations of a Binary Neutron Star Merger
- WaveNet: A Generative Model for Raw Audio
- Spectral Networks and Locally Connected Networks on Graphs
- Science Case for the Einstein Telescope
- Frequency-domain gravitational waves from non-precessing black-hole binaries. II. A phenomenological model for the advanced detector era
- The PyCBC search for gravitational waves from compact binary coalescence
- Computationally efficient models for the dominant and sub-dominant harmonic modes of precessing binary black holes
- Deep Learning for Real-time Gravitational Wave Detection and Parameter Estimation: Results with Advanced LIGO Data
- Open data from the third observing run of LIGO, Virgo, KAGRA and GEO
- Deep Neural Networks to Enable Real-time Multimessenger Astrophysics
- The LIGO Open Science Center
- Matching matched filtering with deep networks in gravitational-wave astronomy
- Enhancing Gravitational-Wave Science with Machine Learning
- Real-Time Detection of Gravitational Waves from Binary Neutron Stars using Artificial Neural Networks
- Gravitational Wave Denoising of Binary Black Hole Mergers with Deep Learning
- Image-based deep learning for classification of noise transients in gravitational wave detectors
- Gravitational wave signal recognition of O1 data by deep learning
- Enabling real-time multi-messenger astrophysics discoveries with deep learning
- How effective is machine learning to detect long transient gravitational waves from neutron stars in a real search?
- Detection of gravitational-wave signals from binary neutron star mergers using machine learning
- Accelerated, Scalable and Reproducible AI-driven Gravitational Wave Detection
- Convergence of Artificial Intelligence and High Performance Computing on NSF-supported Cyberinfrastructure
- Applying deep neural networks to the detection and space parameter estimation of compact binary coalescence with a network of gravitational wave detectors
- MLGWSC-1: The first Machine Learning Gravitational-Wave Search Mock Data Challenge
- Deep learning for clustering of continuous gravitational wave candidates
- 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
- Some Optimizations on Detecting Gravitational Wave Using Convolutional Neural Network
- Training Strategies for Deep Learning Gravitational-Wave Searches
- Detection of Gravitational Waves Using Bayesian Neural Networks
- Fusing numerical relativity and deep learning to detect higher-order multipole waveforms from eccentric binary black hole mergers
- Early warning of coalescing neutron-star and neutron-star-black-hole binaries from nonstationary noise background using neural networks
- Deep Learning Detection and Classification of Gravitational Waves from Neutron Star-Black Hole Mergers
- From One to Many: A Deep Learning Coincident Gravitational-Wave Search
- Inference-optimized AI and high performance computing for gravitational wave detection at scale
- Quasinormal modes of dS and AdS Black Holes: Feedforward neural network method
- Deep Learning with Quantized Neural Networks for Gravitational Wave Forecasting of Eccentric Compact Binary Coalescence
- Physics-inspired deep learning to characterize the signal manifold of quasi-circular, spinning, non-precessing binary black hole mergers
- Deep Learning Model on Gravitational Waveforms in Merging and Ringdown Phases of Binary Black Hole Coalescences
- Black-Hole evaporation from the perspective of neural networks
- Deep Learning for Real-time Gravitational Wave Detection and Parameter Estimation with LIGO Data
- 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
- The first AI simulation of a black hole