Complete parameter inference for GW150914 using deep learning
Not on the map. Why?
Abstract (arXiv)
The LIGO and Virgo gravitational-wave observatories have detected many exciting events over the past five years. As the rate of detections grows with detector sensitivity, this poses a growing computational challenge for data analysis. With this in mind, in this work we apply deep learning techniques to perform fast likelihood-free Bayesian inference for gravitational waves. We train a neural-network conditional density estimator to model posterior probability distributions over the full 15-dimensional space of binary black hole system parameters, given detector strain data from multiple detectors. We use the method of normalizing flows---specifically, a neural spline normalizing flow---which allows for rapid sampling and density estimation. Training the network is likelihood-free, requiring samples from the data generative process, but no likelihood evaluations. Through training, the network learns a global set of posteriors: it can generate thousands of independent posterior samples per second for any strain data consistent with the prior and detector noise characteristics used for training. By training with the detector noise power spectral density estimated at the time of GW150914, and conditioning on the event strain data, we use the neural network to generate accurate posterior samples consistent with analyses using conventional sampling techniques.
Key facts. Cited by 25 papers in the map (data through 7 Oct 2026). First posted on arXiv 7 Aug 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 8; 2022: 5; 2023: 4; 2024: 3; 2025: 3; 2026: 2.
25 papers.
Sort:
By area:
- 2023Flow Matching for Scalable Simulation-Based Inferencecited by 97
- 2021Real-time gravitational-wave science with neural posterior estimationcited by 76
- 2022Rediscovering orbital mechanics with machine learningcited by 71
- 2023Learning Robust Statistics for Simulation-based Inference under Model Misspecificationcited by 48
- 2022Investigating the Impact of Model Misspecification in Neural Simulation-based Inferencecited by 44
- 2022Neural Importance Sampling for Rapid and Reliable Gravitational-Wave Inferencecited by 28
- 2024Real-time gravitational-wave inference for binary neutron stars using machine learningcited by 21
- 2021A deep generative model for probabilistic energy forecasting in power systems: normalizing flowscited by 18
- 2021Group equivariant neural posterior estimationcited by 16
- 2021Advances in Machine and Deep Learning for Modeling and Real-time Detection of Multi-Messenger Sourcescited by 10
- 2021Arbitrary Marginal Neural Ratio Estimation for Simulation-based Inferencecited by 5
- 2021Interpretable AI forecasting for numerical relativity waveforms of quasi-circular, spinning, non-precessing binary black hole mergerscited by 5
13 more
- 2022Inference-optimized AI and high performance computing for gravitational wave detection at scalecited by 5
- 2022Adapting to noise distribution shifts in flow-based gravitational-wave inferencecited by 5
- 2023Comparison of Affine and Rational Quadratic Spline Coupling and Autoregressive Flows through Robust Statistical Testscited by 5
- 2021Swift sky localization of gravitational waves using deep learning seeded importance samplingcited by 4
- 2024Bayesian evidence estimation from posterior samples with normalizing flowscited by 4
- 2021AI 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 mergerscited by 3
- 2023Hyperparameter optimization of hp-greedy reduced basis for gravitational wave surrogatescited by 1
- 2024Accelerated Bayesian parameter estimation and model selection for gravitational waves with normalizing flowscited by 1
- 2025Flexible Gravitational-Wave Parameter Estimation with Transformerscited by 1
- 2025Enhancing the reliability of machine learning for gravitational wave parameter estimation with attention-based modelscited by 0
- 2025Recent Advances in Simulation-based Inference for Gravitational Wave Data Analysiscited by 0
- 2026When Attention Collapses: Residual Evidence Modeling for Compositional Inferencecited by 0
- 2026Binary Black Hole Parameter Estimation with Hybrid CNN-Transformer Neural Networkscited by 0
- 2026Binary Black Hole Parameter Estimation with Hybrid CNN-Transformer Neural Networks11 Jun 2026
- 2026When Attention Collapses: Residual Evidence Modeling for Compositional Inference4 May 2026
- 2025Flexible Gravitational-Wave Parameter Estimation with Transformers2 Dec 2025
- 2025Recent Advances in Simulation-based Inference for Gravitational Wave Data Analysis15 Jul 2025
- 2025Enhancing the reliability of machine learning for gravitational wave parameter estimation with attention-based models17 Jan 2025
- 2024Accelerated Bayesian parameter estimation and model selection for gravitational waves with normalizing flows28 Oct 2024
- 2024Real-time gravitational-wave inference for binary neutron stars using machine learning12 Jul 2024
- 2024Bayesian evidence estimation from posterior samples with normalizing flows18 Apr 2024
- 2023Hyperparameter optimization of hp-greedy reduced basis for gravitational wave surrogates23 Oct 2023
- 2023Flow Matching for Scalable Simulation-Based Inference26 May 2023
- 2023Learning Robust Statistics for Simulation-based Inference under Model Misspecification25 May 2023
- 2023Comparison of Affine and Rational Quadratic Spline Coupling and Autoregressive Flows through Robust Statistical Tests23 Feb 2023
13 more
- 2022Adapting to noise distribution shifts in flow-based gravitational-wave inference16 Nov 2022
- 2022Neural Importance Sampling for Rapid and Reliable Gravitational-Wave Inference11 Oct 2022
- 2022Investigating the Impact of Model Misspecification in Neural Simulation-based Inference5 Sep 2022
- 2022Rediscovering orbital mechanics with machine learning4 Feb 2022
- 2022Inference-optimized AI and high performance computing for gravitational wave detection at scale26 Jan 2022
- 2021AI 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 mergers13 Dec 2021
- 2021Group equivariant neural posterior estimation25 Nov 2021
- 2021Swift sky localization of gravitational waves using deep learning seeded importance sampling1 Nov 2021
- 2021Interpretable AI forecasting for numerical relativity waveforms of quasi-circular, spinning, non-precessing binary black hole mergers13 Oct 2021
- 2021Arbitrary Marginal Neural Ratio Estimation for Simulation-based Inference1 Oct 2021
- 2021Real-time gravitational-wave science with neural posterior estimation23 Jun 2021
- 2021A deep generative model for probabilistic energy forecasting in power systems: normalizing flows17 Jun 2021
- 2021Advances in Machine and Deep Learning for Modeling and Real-time Detection of Multi-Messenger Sources13 May 2021
- 2023Flow Matching for Scalable Simulation-Based Inferencecited by 97
- 2023Flow Matching for Scalable Simulation-Based Inference26 May 2023
- 2021Real-time gravitational-wave science with neural posterior estimationcited by 76
- 2022Rediscovering orbital mechanics with machine learningcited by 71
- 2023Learning Robust Statistics for Simulation-based Inference under Model Misspecificationcited by 48
- 2022Investigating the Impact of Model Misspecification in Neural Simulation-based Inferencecited by 44
- 2022Neural Importance Sampling for Rapid and Reliable Gravitational-Wave Inferencecited by 28
- 2024Real-time gravitational-wave inference for binary neutron stars using machine learningcited by 21
- 2021A deep generative model for probabilistic energy forecasting in power systems: normalizing flowscited by 18
- 2021Group equivariant neural posterior estimationcited by 16
- 2021Advances in Machine and Deep Learning for Modeling and Real-time Detection of Multi-Messenger Sourcescited by 10
- 2021Arbitrary Marginal Neural Ratio Estimation for Simulation-based Inferencecited by 5
- 2021Interpretable AI forecasting for numerical relativity waveforms of quasi-circular, spinning, non-precessing binary black hole mergerscited by 5
- 2022Inference-optimized AI and high performance computing for gravitational wave detection at scalecited by 5
12 most cited of 24 All 24 (JSON)
- 2026Binary Black Hole Parameter Estimation with Hybrid CNN-Transformer Neural Networks11 Jun 2026
- 2026When Attention Collapses: Residual Evidence Modeling for Compositional Inference4 May 2026
- 2025Flexible Gravitational-Wave Parameter Estimation with Transformers2 Dec 2025
- 2025Recent Advances in Simulation-based Inference for Gravitational Wave Data Analysis15 Jul 2025
- 2025Enhancing the reliability of machine learning for gravitational wave parameter estimation with attention-based models17 Jan 2025
- 2024Accelerated Bayesian parameter estimation and model selection for gravitational waves with normalizing flows28 Oct 2024
- 2024Real-time gravitational-wave inference for binary neutron stars using machine learning12 Jul 2024
- 2024Bayesian evidence estimation from posterior samples with normalizing flows18 Apr 2024
- 2023Hyperparameter optimization of hp-greedy reduced basis for gravitational wave surrogates23 Oct 2023
- 2023Learning Robust Statistics for Simulation-based Inference under Model Misspecification25 May 2023
- 2023Comparison of Affine and Rational Quadratic Spline Coupling and Autoregressive Flows through Robust Statistical Tests23 Feb 2023
- 2022Adapting to noise distribution shifts in flow-based gravitational-wave inference16 Nov 2022
12 newest of 24 All 24 (JSON)