Binary Black Hole Parameter Estimation with Hybrid CNN-Transformer Neural Networks

Abstract

The detection of gravitational waves has revolutionized our ability to explore fundamental aspects of the Universe. Traditionally, modeled gravitational-wave signals have been identified using template-based matched filtering, followed by coincidence analysis across multiple detectors in the signal-to-noise ratio time series. Recent advances in Machine Learning and Deep Learning have sparked growing interest in their application to both signal detection and parameter estimation. In this study, a hybrid Deep Learning strategy is proposed that leverages the effectiveness of Transformer encoders alongside well-established Convolutional Neural Network architectures in an attempt to estimate the intrinsic and extrinsic parameters of non-precessing binary black hole systems. The primary focus of this work is point estimation, producing single best-fit values for each parameter rather than full posterior distributions. This method is evaluated on both simulated signals embedded in Gaussian noise and real gravitational-wave events, and it demonstrates strong predictive performance and robustness across key astrophysical parameters.

Accepted manuscript. 12 pages, 10 figures

References in corpus (17)

• Observation of Gravitational Waves from a Binary Black Hole Merger
• Advanced Virgo: a 2nd generation interferometric gravitational wave detector
• An improved effective-one-body model of spinning, nonprecessing binary black holes for the era of gravitational-wave astrophysics with advanced detectors
• Deep Learning for Real-time Gravitational Wave Detection and Parameter Estimation: Results with Advanced LIGO Data
• Deep Neural Networks to Enable Real-time Multimessenger Astrophysics
• Real-time gravitational-wave science with neural posterior estimation
• Matching matched filtering with deep networks in gravitational-wave astronomy
• Gravitational-wave parameter estimation with autoregressive neural network flows
• Real-Time Detection of Gravitational Waves from Binary Neutron Stars using Artificial Neural Networks
• 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
• Real-time gravitational-wave inference for binary neutron stars using machine learning
• Using Deep Learning to Localize Gravitational Wave Sources
• Deep Learning Detection and Classification of Gravitational Waves from Neutron Star-Black Hole Mergers
• Rapid localization of gravitational wave sources from compact binary coalescences using deep learning
• Pre-merger sky localization of gravitational waves from binary neutron star mergers using deep learning
• Deep Learning-based Search for Microlensing Signature from Binary Black Hole Events in GWTC-1 and -2