Interpreting a Machine Learning Model for Detecting Gravitational Waves
Not on the map. Why?
Abstract (arXiv)
We describe a case study of translational research, applying interpretability techniques developed for computer vision to machine learning models used to search for and find gravitational waves. The models we study are trained to detect black hole merger events in non-Gaussian and non-stationary advanced Laser Interferometer Gravitational-wave Observatory (LIGO) data. We produced visualizations of the response of machine learning models when they process advanced LIGO data that contains real gravitational wave signals, noise anomalies, and pure advanced LIGO noise. Our findings shed light on the responses of individual neurons in these machine learning models. Further analysis suggests that different parts of the network appear to specialize in local versus global features, and that this difference appears to be rooted in the branched architecture of the network as well as noise characteristics of the LIGO detectors. We believe efforts to whiten these "black box" models can suggest future avenues for research and help inform the design of interpretable machine learning models for gravitational wave astrophysics.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 20 of its 49 references that are in the map. First posted on arXiv 15 Feb 2022.
Builds on
20 of 49 references.
References · 20
- 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,690
- 2017Axiomatic Attribution for Deep Networkscited by 3,350
- 2013Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Mapscited by 2,348
- 2016WaveNet: A Generative Model for Raw Audiocited by 1,710
- 2017Towards A Rigorous Science of Interpretable Machine Learningcited by 1,400
- 2017SmoothGrad: removing noise by adding noisecited by 1,097
- 2019BERT Rediscovers the Classical NLP Pipelinecited by 1,046
- 2018Deep learning for time series classification: a reviewcited by 575
- 2020Benchmarking Deep Learning Interpretability in Time Series Predictionscited by 102
- 2017Deep Learning for Real-time Gravitational Wave Detection and Parameter Estimation: Results with Advanced LIGO Datacited by 33
- 2016Deep Neural Networks to Enable Real-time Multimessenger Astrophysicscited by 24
- 2020Accelerated, Scalable and Reproducible AI-driven Gravitational Wave Detectioncited by 20
8 more
- 2019Enabling real-time multi-messenger astrophysics discoveries with deep learningcited by 18
- 2021A FAIR and AI-ready Higgs boson decay datasetcited by 14
- 2020Convergence of Artificial Intelligence and High Performance Computing on NSF-supported Cyberinfrastructurecited by 7
- 2017Deep Learning for Real-time Gravitational Wave Detection and Parameter Estimation with LIGO Datacited by 5
- 2018Deep Learning at Scale for the Construction of Galaxy Catalogs in the Dark Energy Surveycited by 5
- 2022Inference-optimized AI and high performance computing for gravitational wave detection at scalecited by 5
- 2020Infrastructure for Artificial Intelligence, Quantum and High Performance Computingcited by 2
- 2021Advancing Computing's Foundation of US Industry & Societycited by 1
Built on by
Citing papers per year: 2022: 1; 2023: 0; 2024: 0; 2025: 1.
2 papers.
- 2022Applications of physics informed neural operatorscited by 33
- 2025Interpretable Machine Learning in Physics: A Reviewcited by 11
- 2025Interpretable Machine Learning in Physics: A Review30 Mar 2025
- 2022Applications of physics informed neural operators23 Mar 2022