Flexible Gravitational-Wave Parameter Estimation with Transformers
arXiv:2512.02968 · doi:10.1103/n3ld-b4wv
Abstract
Gravitational-wave data analysis relies on accurate and efficient methods to extract physical information from noisy detector signals, yet the increasing rate and complexity of observations represent a growing challenge. Deep learning provides a powerful alternative to traditional inference, but existing neural models typically lack the flexibility to handle variations in data analysis settings. Such variations accommodate imperfect observations or are required for specialized tests, and could include changes in detector configurations, overall frequency ranges, or localized cuts. We introduce a flexible transformer-based architecture paired with a training strategy that enables adaptation to diverse analysis settings at inference time. Applied to parameter estimation, we demonstrate that a single flexible model, called Dingo-T1, can (i) analyze 48 gravitational-wave events from the third LIGO-Virgo-KAGRA Observing Run under a wide range of analysis configurations, (ii) enable systematic studies of how detector and frequency configurations impact inferred posteriors, and (iii) perform inspiral-merger-ringdown consistency tests probing general relativity. Dingo-T1 also improves median sample efficiency on real events from a baseline of 1.4% to 4.2%. Our approach thus demonstrates flexible and scalable inference with a principled framework for handling missing or incomplete data, key capabilities for current and next-generation observatories.
5+15 pages, 3+8 figures
References in corpus (36)
- Array Programming with NumPy
- The Astropy Project: Sustaining and Growing a Community-oriented Open-source Project and the Latest Major Release (v5.0) of the Core Package
- Advanced Virgo: a 2nd generation interferometric gravitational wave detector
- Multi-messenger Observations of a Binary Neutron Star Merger
- Advanced LIGO
- GW170817: Measurements of Neutron Star Radii and Equation of State
- GWTC-3: Compact Binary Coalescences Observed by LIGO and Virgo During the Second Part of the Third Observing Run
- Bilby: A user-friendly Bayesian inference library for gravitational-wave astronomy
- Tests of General Relativity with Binary Black Holes from the second LIGO-Virgo Gravitational-Wave Transient Catalog
- Computationally efficient models for the dominant and sub-dominant harmonic modes of precessing binary black holes
- Bayesian inference for compact binary coalescences with BILBY: Validation and application to the first LIGO--Virgo gravitational-wave transient catalogue
- BayesWave: Bayesian Inference for Gravitational Wave Bursts and Instrument Glitches
- A guide to LIGO-Virgo detector noise and extraction of transient gravitational-wave signals
- BayesLine: Bayesian Inference for Spectral Estimation of Gravitational Wave Detector Noise
- Real-time gravitational-wave science with neural posterior estimation
- Constraints on the cosmic expansion history from GWTC-3
- Bayesian parameter estimation using conditional variational autoencoders for gravitational-wave astronomy
- Gravitational-wave parameter estimation with autoregressive neural network flows
- The BayesWave analysis pipeline in the era of gravitational wave observations
- Convolutional neural networks: a magic bullet for gravitational-wave detection?
- Learning Bayesian posteriors with neural networks for gravitational-wave inference
- Testing general relativity using gravitational wave signals from the inspiral, merger and ringdown of binary black holes
- Tests of General Relativity with GWTC-3
- Neural Importance Sampling for Rapid and Reliable Gravitational-Wave Inference
- Image-based deep learning for classification of noise transients in gravitational wave detectors
- Bilby-MCMC: An MCMC sampler for gravitational-wave inference
- SWIGLAL: Python and Octave interfaces to the LALSuite gravitational-wave data analysis libraries
- Accelerating gravitational wave parameter estimation with multi-band template interpolation
- Accelerating parameter estimation of gravitational waves from compact binary coalescence using adaptive frequency resolutions
- Time-Frequency Analysis of Gravitational Wave Data
- Real-time gravitational-wave inference for binary neutron stars using machine learning
- New Gravitational Wave Discoveries Enabled by Machine Learning
- The anti-aligned spin of GW191109: glitch mitigation and its implications
- 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
- Transformer Networks for Continuous Gravitational-wave Searches