Some optimizations on detecting gravitational wave using convolutional neural network

Xiang Ru Li, Wo Liang Yu, Xi Long Fan, G. Jogesh Babu

Research output: Contribution to journalArticlepeer-review

24 Scopus citations


This work investigates the problem of detecting gravitational wave (GW) events based on simulated damped sinusoid signals contaminated with white Gaussian noise. It is treated as a classification problem with one class for the interesting events. The proposed scheme consists of the following two successive steps: decomposing the data using a wavelet packet, representing the GW signal and noise using the derived decomposition coeficients; and determining the existence of any GW event using a convolutional neural network (CNN) with a logistic regression output layer. The characteristic of this work is its comprehensive investigations on CNN structure, detection window width, data resolution, wavelet packet decomposition and detection window overlap scheme. Extensive simulation experiments show excellent performances for reliable detection of signals with a range of GW model parameters and signal-to-noise ratios. While we use a simple waveform model in this study, we expect the method to be particularly valuable when the potential GW shapes are too complex to be characterized with a template bank.

Original languageEnglish (US)
Article number54501
JournalFrontiers of Physics
Issue number5
StatePublished - Oct 1 2020

All Science Journal Classification (ASJC) codes

  • Physics and Astronomy (miscellaneous)


Dive into the research topics of 'Some optimizations on detecting gravitational wave using convolutional neural network'. Together they form a unique fingerprint.

Cite this