Physics-inspired spatiotemporal-graph AI ensemble for the detection of higher order wave mode signals of spinning binary black hole mergers
[1];
[2];
[3];
- Argonne National Laboratory (ANL), Argonne, IL (United States); University of Illinois at Urbana-Champaign, IL (United States)
- Argonne National Laboratory (ANL), Argonne, IL (United States); University of Illinois at Urbana-Champaign, IL (United States); University of Chicago, IL (United States)
- Argonne National Laboratory (ANL), Argonne, IL (United States)
- Tata Institute of Fundamental Research, Bangalore (India)
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 ($$\ell$$, |m|) = {(2,2), (2,1), (3,3), (3,2), (4,4)}, and mode mixing effects in the $$\ell$$ = 3, |m| = 2 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 h using 256 NVIDIA A100 GPUs in the Polaris supercomputer at the Argonne Leadership Computing Facility. This distributed training approach exhibited optimal classification performance, and strong scaling up to 512 NVIDIA 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 h. 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 NVIDIA A100 GPU.
- Research Organization:
- Argonne National Laboratory (ANL), Argonne, IL (United States)
- Sponsoring Organization:
- USDOE Laboratory Directed Research and Development (LDRD) Program; National Science Foundation (NSF); USDOE Office of Science (SC), Basic Energy Sciences (BES). Scientific User Facilities (SUF)
- Grant/Contract Number:
- AC02-06CH11357; OAC-1931561; OAC-2209892; OAC-2005572
- OSTI ID:
- 2369833
- Alternate ID(s):
- OSTI ID: 2350857; OSTI ID: 2475502
- Journal Information:
- Machine Learning: Science and Technology, Vol. 5, Issue 2; ISSN 2632-2153
- Publisher:
- IOP PublishingCopyright Statement
- Country of Publication:
- United States
- Language:
- English
Similar Records
Inference-Optimized AI and High Performance Computing for Gravitational Wave Detection at Scale
APACE: AlphaFold2 and advanced computing as a service for accelerated discovery in biophysics