Cardiff University | Prifysgol Caerdydd ORCA
Online Research @ Cardiff 
WelshClear Cookie - decide language by browser settings

Gravitational-wave surrogate models powered by artificial neural networks

Khan, Sebastian ORCID: and Green, Rhys 2021. Gravitational-wave surrogate models powered by artificial neural networks. Physical Review D 103 (6) , 064015. 10.1103/PhysRevD.103.064015

[thumbnail of 10.1103PhysRevD.103.064015 - OA.pdf] PDF - Published Version
Download (619kB)


Inferring the properties of black holes and neutron stars is a key science goal of gravitational-wave (GW) astronomy. To extract as much information as possible from GW observations, we must develop methods to reduce the cost of Bayesian inference. In this paper, we use artificial neural networks (ANNs) and the parallelization power of graphics processing units (GPUs) to improve the surrogate modeling method, which can produce accelerated versions of existing models. As a first application of our method, the artificial neural networks surrogate model (ANN-Sur), we build a time-domain surrogate model of the spin-aligned binary black hole (BBH) waveform model SEOBNRv4. We achieve median mismatches of approximately 2 e−5 and mismatches no worse than approximately 2e−3. For a typical BBH waveform generated from 12 Hz with a total mass of 60M⊙, the original SEOBNRv4 model takes 1794 ms. Existing custom-made code optimizations (SEOBNRv4opt) reduced this to 83.7 ms, and the interpolation-based, frequency-domain surrogate SEOBNRv4ROM can generate this waveform in 3.5 ms. Our ANN-Sur model when run on a CPU takes 1.2 ms and when run on a graphics processing unit (GPU) takes just 0.5 ms. ANN-Sur can also generate large batches of waveforms simultaneously. We find that batches of up to 10 3 waveforms can be evaluated on a GPU in just 1.57 ms, corresponding to a time per waveform of 0.0016 ms. This method is a promising way to utilize the parallelization power of GPUs to drastically increase the computational efficiency of Bayesian parameter estimation.

Item Type: Article
Date Type: Publication
Status: Published
Schools: Advanced Research Computing @ Cardiff (ARCCA)
Physics and Astronomy
Publisher: American Physical Society
ISSN: 2470-0010
Date of First Compliant Deposit: 28 July 2021
Date of Acceptance: 26 January 2021
Last Modified: 21 Jun 2024 13:16

Citation Data

Cited 11 times in Scopus. View in Scopus. Powered By Scopus® Data

Actions (repository staff only)

Edit Item Edit Item


Downloads per month over past year

View more statistics