Differentiable and hardware-accelerated waveforms for gravitational wave data analysis

Abstract

We propose the use of automatic differentiation through the programming framework for accelerating a variety of analysis tasks throughout gravitational wave (GW) science. Firstly, we demonstrate that complete waveforms which cover the inspiral, merger, and ringdown of binary black holes (i.e., IMRPhenomD) can be written in and demonstrate that the serial evaluation speed of the waveform (and its derivative) is similar to the implementation in . Moreover, allows for graphics processing unit–accelerated waveform calls which can be over an order of magnitude faster than serial evaluation on a CPU. We then focus on three applications where efficient and differentiable waveforms are essential. Firstly, we demonstrate how gradient descent can be used to optimize the ∼200 coefficients that are used to calibrate the waveform model. In particular, we demonstrate that the typical with numerical relativity waveforms can be improved by more than 50%. Secondly, we show that Fisher forecasting calculations can be sped up by ∼3–5× (on a CPU) with no loss in accuracy. This increased speed makes Fisher forecasting for a population of events substantially simpler. Finally, we show that gradient-based samplers like Hamiltonian Monte Carlo lead to significantly reduced autocorrelation values when compared to traditional Monte Carlo methods. Since differentiable waveforms have substantial advantages for a variety of tasks throughout GW science, we propose that waveform developers use to build new waveforms moving forward. Our waveform code, , can be found on GitHub website and will continue to be updated with new waveforms as they are implemented. Published by the American Physical Society 2024