用神经网络加速黑洞并合波形模拟,精度高且支持梯度计算。
Fast, accurate, and differentiable: a neural-network surrogate for NRSur7dq4 precessing binary black hole waveforms

- 分量分解+独立MLP建模,实现快速准确的波形拟合。
- 1毫秒内完成单个波形计算,95%误差低于0.001。
- 首次实现可微分、GPU加速的全链路引力波参数推断。
我们提出一个神经网络代理模型,用于模拟预旋双黑洞并合的NRSur7dq4引力波波形。该代理将波形分解为多个构成量,对每个量训练独立的多层感知机(MLP)。在覆盖完整参数空间(质量比1≤q≤4,自旋|χ_{A,B}|≤0.8)的10,000个波形上验证了模型性能。对于60至300 $M_igodot$ 的典型总质量,天空平均频率域失配中位数在 $8.0 \times 10^{-5}$ 至 $1.7 \times 10^{-4}$ 之间,95百分位低于 $10^{-3}$。在NVIDIA L40S GPU上,单个波形端到端计算耗时约1毫秒,较LALSimulation C实现快约10倍,批量大小64时吞吐量提升约140倍。整个NRSur7dq4神经网络波形到似然函数的流程基于JAX实现,完全可微。这是首个结合数值相对论保真度与全可微、GPU加速推断流程的预旋波形神经网络代理,支持自动微分下的梯度推断方法,包括费舍尔信息矩阵、GPU加速嵌套采样、梯度驱动马尔可夫链蒙特卡洛和重要性采样。
原文摘要 · Abstract (English)
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 \leq q \leq 4$, $|χ_{A,B}| \leq 0.8$). For representative total masses between 60 and 300 $M_\odot$, median sky-averaged frequency-domain mismatches range from $8.0 \times 10^{-5}$ to $1.7 \times 10^{-4}$, with 95th percentiles below $10^{-3}$. On an NVIDIA L40S GPU the JAX surrogate evaluates a single waveform in about 1 ms end-to-end, roughly 10 times faster than the LALSimulation C implementation of NRSur7dq4, and sustains about 140 times 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 NRSur7dq4 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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。