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

Abstract

We use artificial intelligence (AI) to learn and infer the physics of higher order gravitational wave modes of quasi-circular, spinning, non precessing binary black hole mergers. We trained AI models using 14 million waveforms, produced with the surrogate model NRHybSur3dq8, that include modes up to $\ell \le 4$ and $(5,5)$, except for $(4,0)$ and $(4,1)$, that describe binaries with mass-ratios $q\le 8$, individual spins $s_{\{1,2\}}^z\in [-0.8,0.8]$, and inclination angle $\theta \in [0,\pi ]$.Our probabilistic AI surrogates can accurately constrain the mass-ratio, individual spins, effective spin, and inclination angle of numerical relativity waveforms that describe such signal manifold. We compared the predictions of our AI models with Gaussian process regression, random forest, k-nearest neighbors, and linear regression, and with traditional Bayesian inference methods through the PyCBC Inference toolkit, finding that AI outperforms all these approaches in terms of accuracy, and are between three to four orders of magnitude faster than traditional Bayesian inference methods. Our AI surrogates were trained within 3.4 hours using distributed training on 1,536 NVIDIA V100 GPUs in the Summit supercomputer.

arXiv comment22 pages, 12 figures

References in corpus (38)

• Deep Learning for Real-time Gravitational Wave Detection and Parameter Estimation: Results with Advanced LIGO Data
• PyCBC Inference: A Python-based parameter estimation toolkit for compact binary coalescence signals
• Surrogate model of hybridized numerical relativity binary black hole waveforms
• Deep Neural Networks to Enable Real-time Multimessenger Astrophysics
• Matching matched filtering with deep networks in gravitational-wave astronomy
• Bayesian parameter estimation using conditional variational autoencoders for gravitational-wave astronomy
• Enhancing Gravitational-Wave Science with Machine Learning
• Gravitational-wave parameter estimation with autoregressive neural network flows
• Learning Bayesian posteriors with neural networks for gravitational-wave inference
• 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
• Reduced-order modeling with artificial neurons for gravitational-wave inference
• Deep-Learning continuous gravitational waves
• Gravitational wave signal recognition of O1 data by deep learning
• Enabling real-time multi-messenger astrophysics discoveries with deep learning
• Noise Reduction in Gravitational-wave Data via Deep Learning
• How effective is machine learning to detect long transient gravitational waves from neutron stars in a real search?
• Accelerated, Scalable and Reproducible AI-driven Gravitational Wave Detection
• Detection of gravitational-wave signals from binary neutron star mergers using machine learning
• 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
• Gravitational-wave surrogate models powered by artificial neural networks: The ANN-Sur for waveform generation
• Deep learning for clustering of continuous gravitational wave candidates
• Deep-Learning Continuous Gravitational Waves: Multiple detectors and realistic noise
• Deep learning for gravitational wave forecasting of neutron star mergers
• Training Strategies for Deep Learning Gravitational-Wave Searches
• Some Optimizations on Detecting Gravitational Wave Using Convolutional Neural Network
• 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
• From One to Many: A Deep Learning Coincident Gravitational-Wave Search
• Inference-optimized AI and high performance computing for gravitational wave detection at scale
• 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
• Artificial neural network subgrid models of 2-D compressible magnetohydrodynamic turbulence
• Interpretable AI forecasting for numerical relativity waveforms 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
• Advances in Machine and Deep Learning for Modeling and Real-time Detection of Multi-Messenger Sources

Cited by in corpus (3)

• MLGWSC-1: The first Machine Learning Gravitational-Wave Search Mock Data Challenge
• FAIR AI Models in High Energy Physics
• Physics-inspired spatiotemporal-graph AI ensemble for the detection of higher order wave mode signals of spinning binary black hole mergers