2024年4月22日, 10:00-10:15

中国物理学会引力与相对论天体物理分会“2024年学术年会”
暨第六届伽利略-徐光启国际会议​​ | 湖南 · 衡阳

WaveFormer: transformer-based denoising method for gravitational-wave data

王赫 (He Wang)

hewang@ucas.ac.cn

中国科学院大学 · 国际理论物理中心(亚太地区)

中国科学院大学 · 引力波宇宙太极实验室(北京/杭州)

based on He Wang et al 2024 Mach. Learn.: Sci. Technol. 5 015046

    1400Ripples                             Air Compressor                                   Blip

    Extremely Loud                                  Helix                                          Koi Fish

Various types of Glitch

Background

  • The improvement of data quality is a very complex issue, with data from over 20,000 sensor channels determining the quality of the gravitational wave science data channel.

  • Reducing non-Gaussian short-duration pulse interference (Glitches) in gravitational wave data will help reduce the false alarm rate of gravitational wave signals.

  • Removing Glitches from gravitational wave detection data is a multi-classification problem.

    • Traditional machine learning algorithms​ ​Powell J, et al. CQG, 2015
    • Deep learning algorithms Zevin, M, et al. C

Ormiston R, et al. PRR, 2020

  • DeepClean: One-dimensional Convolutional Neural Network which takes a specified set of witness channels and subsequently outputs the predicted noise in strain.

IGWN​ data processing

Non-stationary

Non-Gaussianity

Background

Related Works

Model Structure

Precessing & Train

Effect on Noise

Effect on BBH signals

Related Works

  • Extraction and denoising GW signals using deep learning:
    • Both Wei et al. [PLB 2020] and Chatterjee et al. [PRD 2021] have shown that considering phase overlaps yields excellent results.
  • Detecting and denoising GW signals using deep learning:
    • Both Bacon et al. [2205.13513] and Murali et al. [PRD 2023] could recover the phase of original GW signal with certain cycles but failed to recover the complete evaluation in amplitude scale.

Chatterjee C, Wen L, et al. PRD 2021

Wei W and Huerta E A. PLB 2020

Bacon P. et al.  arXiv: 2205.13513

GW170823

Murali C & Lumley D. PRD 2023

Network Architecture

  • The WaveFormer, a billion-scale transformer-based model, excels in suppressing realistic noise and recovering injections or GW events, thereby significantly improving data quality.
  • In its application, it treats each overlapping time-domain data subsequence as an individual token, akin to tokenization in natural language processing (NLP).
["This", "is", "a", "sample"]

Data Preprocessing and Training Strategy

\frac{d-mean}{std} = \frac{h}{std}+\frac{n-mean}{std}

Strain

Whiten

Normalized

 ∼\(10^{−19}\)

 ∼\(10^{2}\)

 ∼\(10^{0}\)

32 s

32 s

merger

\(t_c\) (around GW150914)

\oplus

(Cal network SNR)

Band-pass: [20, 2048] Hz

Patching (tokenized) with size 0.125 s and overlap 50%

[1, 128, 256]

(Standard normalization)

dynamic masking

[1, 16512]

[1, 128, 256]

(PSD\(_i\) from noise)

Band-pass: [20, 2048] Hz

WaveFormer

MSE-Loss\(_i\)

\(std\)

[1, 128, 256]

Noise\(_i\):

Signal\(_i\):

Input\(_i\):

Label\(_i\):

Output\(_i\):

8.0625 s

8.0625 s

Given �=ℎ+�d=h+n, we can normalize d as follows:

  • Implementations:
    • PSD sampling from real noise.
    • input size: 8.0625 sec
    • fs = ​2048Hz
    • Band-pass: 20~2048Hz
    • Masked loss

Effect on Realistic Noise

  • Noise level percentile amplitude is significantly reduced, by approximately two orders.
  • Further ASD analysis shows that WaveFormer effectively eliminates both narrowband and broadband spectral information, substantially lowering frequency contributions.
  • Using the Gravity Spy database for glitches with SNR > 10 and confidence > 0.95, results show significant suppression of glitches in real advanced LIGO-Virgo noise.

(Bottom panels: results of glitches)

(Upper panels: results of pure noise)

Time-series and spectrogram example of blip.

Recovery of Binary Black Holes

  • Overlap and matched-filtering signal-to-noise are calculated to represent phase and amplitude recovery performance.
  • Among the intermediate frequency range (20–200 Hz) that covers rich BBH signal information, the ASD distribution of denoised waveform is evidently consistent with that of target signal.

(Upper panels: Signal amplitude recovery performance

(Bottom panels: Signal phase recovery performance​)

Bacon P. et al.  arXiv: 2205.13513

  • These results show that our denoising algorithm outperformed others by capturing the characteristic chirping morphology of BBH evolution, and can denoise signals in realistic detection scenarios without affecting signal characteristics such as phase and amplitude.
  • For the event GW191204_171526, classified as either an NSBH or a low-mass BBH candidate in GWTC-3, the overlap with IMRPhenomXPHM achieved 0.93 and 0.95 on H1 and L1, respectively, which are marked improvements over those achieved by BayesWave and cWB (with overlaps between 0.820.86).

GW191204_171526

Recovery of Binary Black Holes

Search Strategy Overview

  • Firstly, we obtain the denoised output by utilizing Waveformer. Then, triggers are defined and identified by three steps including,

    • Find Peaks. Locate triggers on a single detector by finding its maximum all local-maximum (0.2s away from neighboring maximum/local-maximum).

  • An search algorithm for GW require that: [cite: 2010.07244]

    1. the same signal is seen in the detectors; (the same signal is seen by time-shifting in single detector)

    2. the same waveform must be present both detectors;

    3. and the signal’s time of arrival must be consistent with the GW travel time between the observatories.

Search Strategy Overview

  • Firstly, we obtain the denoised output by utilizing Waveformer. Then, triggers are defined and identified by three steps including,

    • Find Peaks. Locate triggers on a single detector by finding its maximum all local-maximum (0.2s away from neighboring maximum/local-maximum).

    • By constraining triggers that exist on both two detectors, we get VALID triggers. (consist 3~4 segments)

Search Strategy Overview

  • Firstly, we obtain the denoised output by utilizing Waveformer. Then, triggers are defined and identified by three steps including,

    • Find Peaks. Locate triggers on a single detector by finding its maximum all local-maximum (0.2s away from neighboring maximum/local-maximum).

    • By constraining triggers that exist on both two detectors, we get VALID triggers. (consist 3~4 segments)

    • Calculate the correlation of the to-be-evaluated trigger across channels or within a single channel, between its noisy and corresponding denoised segments, as well as between denoised segments themselves.

L^2(\text{Corr}^{\text{ab}}(n))
\text{Corr}^{{{H}\bar{H}}}(n)
\text{Corr}^{{{L}\bar{L}}}(n)
\text{Corr}^{\text{ab}}(n) = \max^{i\in[-2,2],i\in\mathbb{Z}}_{t\in[i\Delta t-\epsilon,i\Delta t+\epsilon]} \langle \bar{h}^a_{(n)}(t)|\bar{h}^b_{(n+i)}(t)\rangle\,, a,b\in(H,L,\bar{H}, \bar{L})
\bar{t}_{a}(n) =\text{argmax}_t \,h^a_{(n)}(t)
\text{Valid}_{\bar{t}_{a}(n)}(n, n+1) = \begin{cases} 1 & \text{ if } |\bar{t}_{a}(n) - \bar{t}_{a}(n+1)| < 0.1 \text{ ms}\\ 0 & \text{ if } \text{otherwise} \end{cases}
\text{Corr}^{{\bar{H}\bar{H}}}(n),\text{Corr}^{{\bar{L}\bar{L}}}(n),\text{Corr}^{{\bar{H}\bar{L}}}(n),\text{Corr}^{{H\bar{H}}}(n),\text{Corr}^{{L\bar{L}}}(n),\text{Corr}^{{H\bar{L}}}(n),\text{Corr}^{{L\bar{H}}}(n)

noisy input segments

denoised output segments

\(\bar{H}\)

\(\bar{L}\)

\({H}\)

\({L}\)

\rho_\text{ranking}

Inverse FAR calculation

  • Firstly, we obtain the denoised output by utilizing Waveformer. Then, triggers are defined and identified by three steps including,

    • Find Peaks. Locate triggers on a single detector by finding its maximum all local-maximum (0.2s away from neighboring maximum/local-maximum).

    • By constraining triggers that exist on both two detectors, we get VALID triggers. (consist 3~4 segments)

    • Calculate the correlation of the to-be-evaluated trigger across channels or within a single channel, between its noisy and corresponding denoised segments, as well as between denoised segments themselves.

  • Through time shift, background analysis is done on other triggers around the target trigger. (time-shift interval 0.1 sec)
  • Finally, by counting the number of false alarm trigger pairs, we obtain the IFAR value of the target trigger, which represents the reported or candidate BBH event in this experiment.

(PyCBC) Davies​, et al. PRD 2020

Ours

Significance Estimates

  • Assessed denoising workflow performance by comparing with GWTC-1, GWTC-2, GWTC2.1, and GWTC-3 catalogs and associated data releases.

  • Noted significant divergence in IFAR distribution between our results and those from GWTC and OGC catalogs.

  • Achieved significant IFAR improvement across all 75 reported BBH events, indicating effective suppression of loud terrestrial noise.

    • Example: For low SNR (\(10.8_{-0.4}^{+0.3}\)) event GW200208_130117, obtained an IFAR of 8916 years, surpassing maximum IFAR of <4000 years in other catalogs.

  • Variability in IFAR improvement linked to the original data's noise nature, including its non-Gaussian, non-stationary characteristics, and different signal recognition strategies by pipelines.

  • IFAR performance significantly depends on the reduction of non-Gaussian noise near each event.

    • Events with substantial IFAR improvement had misleading non-Gaussian noise effectively eliminated.

    • Events where IFAR underperforms retained non-Gaussian characteristics, possibly due to WaveFormer's inherent systematic errors.

Summary

  • Developed an AI-based workflow with WaveFormer, combining convolutional neural network and transformer for effective GW noise suppression and hierarchical feature extraction across a wide frequency range.
  • Achieved significant noise suppression and signal recovery performance improvements, including state-of-the-art results on real observational data and BBH events, leading to dramatic data quality improvement and significant IFAR enhancement on 75 reported BBH events.

Text

Text

GW151226

GW151012

LVK.  arXiv:1602.03839

He Wang, et al. MLST. 5, 1 (2024): 015046.

  • Challenges in Model Interpretability
    • The black-box nature of AI models complicates interpretability, challenging the comparison of AI-generated detection statistics with traditional matched filtering chi-square distributions.
    • Convincing the scientific community of the pipeline's validity and the statistical significance of new discoveries remains difficult despite the model's ability to identify potential gravitational wave signals.
       
  • Future Directions
    • Construct a comprehensive GW signal search pipeline for BBH/BNS/NSBH events.
    • Explore the use of ensemble learning and other statistical methods to enhance the interpretability of the AI detection pipeline and address issues related to its validity.