[论文解读] Accelerated parameter estimation in Bilby with relative binning

K. Murali Krishna, A. Vijaykumar+6 more|arXiv (Cornell University)|Dec 10, 2023

Pulsars and Gravitational Waves Research被引用 7

一句话总结

本文提出在开源贝叶斯推断软件Bilby中实现相对分箱（relative binning），以加速引力波参数估计。通过将波形比值近似为频率分箱上的分段线性函数，该方法降低了似然评估成本，使在单核CPU上仅用14小时即可完成GW170817事件的参数恢复，同时保持高精度并通过了统计检验。

ABSTRACT

We describe an implementation of the relative binning technique to speed up parameter estimation of gravitational-wave signals. We first give a pedagogical overview of relative binning, discussing also the expressions for the likelihood marginalized over phase and distance. Then, we describe the details of the code in exttt{Bilby}, an open-source software package commonly used for parameter estimation of gravitational-wave sources. Our code is able to reproduce the parameters of GW170817 in 14 hours on a single-core CPU, performs well on simulated signals, and passes the percentile-percentile (p-p) tests. We also illustrate that relative binning is an ideal technique to estimate the parameters of signals in next-generation gravitational wave detectors.

研究动机与目标

• 解决长时长短引力波信号（尤其是双中子星系统）贝叶斯参数估计的高计算成本问题。
• 在不牺牲精度的前提下，减少参数估计中似然评估所需的时间。
• 为下一代引力波探测器（具有更长带内信号和更高灵敏度）实现更快的推断速度。
• 在Bilby软件包中实现相对分箱，以支持高效、可扩展且可复现的参数估计。
• 确保与统计验证方法（如百分位-百分位（p-p）检验）兼容，以确认结果的可靠性。

提出的方法

• 在频率分箱上，使用分段线性函数近似参数空间中引力波形的比值，从而减少所需波形评估的次数。
• 预先计算分箱边界处波形比值的基函数系数，实现在采样过程中通过插值快速获取结果，而无需重新评估完整波形。
• 采用修改后的似然函数，仅依赖插值得到的波形比值和预计算的汇总统计量，最大限度减少实时计算量。
• 将该方法集成到Bilby中，利用其现有的贝叶斯推断与采样基础设施。
• 实现时间边际化的两种替代方案：一种通过波形比值插值，另一种通过修改汇总统计量；但后者被证明计算效率较低。
• 将该方法应用于真实和模拟信号，通过p-p检验验证结果，并与标准似然方法进行比较。

实验结果

研究问题

• RQ1 相对分箱是否能显著降低引力波参数估计的计算成本，同时不降低精度？
• RQ2 在真实引力波事件（如GW170817）上，相对分箱在运行时间和参数恢复方面表现如何？
• RQ3 在使用相对分箱时，时间边际化是否可行且有益，还是反而引入了计算开销？
• RQ4 相对分箱能否有效集成到Bilby软件包中，以支持更广泛的社区使用？
• RQ5 该方法在信号持续时间增加和探测器灵敏度提高的情况下（特别是针对下一代探测器）是否具有良好可扩展性？

主要发现

• 在Bilby中实现的相对分箱方法，仅用单核CPU在14小时内即完成GW170817事件的参数恢复，相比标准方法有显著加速。
• 该方法通过了百分位-百分位（p-p）检验，证实后验分布具有统计有效性且校准良好。
• 该方法在模拟信号上保持了高精度，表明其在不同信号参数和噪声实现下均具有鲁棒性。
• 通过插值实现的时间边际化虽可行，但计算成本高于标准相对分箱，整体效率较低。
• 该方法在信号持续时间增加时表现出良好的可扩展性，特别适用于下一代探测器（如Cosmic Explorer和Einstein Telescope）所观测的更长带内信号。
• 该实现已集成至Bilby，支持可复现、开源的参数估计，且运行时开销极低。