Fast, accurate, and differentiable: a neural-network surrogate for NRSur7dq4 precessing binary black hole waveforms

Michael Pürrer, Ashwin Girish, Lucy Thomas, Scott Field, Vijay Varma

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 4 functions out of this paper's own repositories and ran 4 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

Repositories linked to this paper

• mpuerrer/jax-nrsur7dq4-nn

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

We present a neural network surrogate model that emulates the NRSur7dq4 gravitational waveform model for precessing binary black hole mergers. The surrogate decomposes the waveform into constituent quantities and trains an independent multilayer perceptron (MLP) for each. We validate the surrogate against NRSur7dq4 on 10 000 waveforms spanning its full parameter space (1 ≤ q ≤ 4, |χA,B| ≤ 0.8). For representative total masses between 60 and 300 M⊙, median sky-averaged frequency-domain mismatches range from 8.0 × 10 -5 to 1.7 × 10 -4 , with 95th percentiles below 10 -3 . On an NVIDIA L40S GPU the JAX surrogate evaluates a single waveform in ∼ 1 ms end-to-end (∼ 10× faster than the lalsimulation C implementation of NRSur7dq4) and sustains ∼ 140× the lalsimulation throughput at batch size 64, making it well suited for both low-latency parameter-estimation samplers and large-scale waveform generation. The full NR-Sur7dq4 NN waveform-to-likelihood pipeline is implemented in JAX and is differentiable. This is the first neural-network surrogate of a precessing numerical-relativity waveform model to combine validated NR-faithful accuracy with a fully differentiable, GPU-accelerated inference pipeline, enabling gradient-based inference approaches via automatic differentiation including Fisher information matrices, GPU-accelerated nested sampling, gradient-based MCMC and importance sampling.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2607.24960")
get_code_for_paper("2607.24960")
have("2607.24960")

Connect an agent — have() is free.