Gravitational Wave-Signal Recognition Model Based on Fourier Transform and Convolutional Neural Network

Abstract

The recent detection of gravitational waves is a remarkable milestone in the history of astrophysics. With the further development of gravitational wave detection technology, traditional filter-matching methods no longer meet the needs of signal recognition. Thus, it is imperative that we develop new methods. In this study, we apply a gravitational wave signal recognition model based on Fourier transformation and a convolutional neural network (CNN). The gravitational wave time-domain signal is transformed into a 2D frequency-domain signal graph for feature recognition using a CNN model. Experimental results reveal that the frequency-domain signal graph provides a better feature description of the gravitational wave signal than that provided by the time-domain signal. Our method takes advantage of the CNN’s convolution computation to improve the accuracy of signal recognition. The impact of the training set size and image filtering on the performance of the developed model is also evaluated. Additionally, the Resnet101 model, developed on the Baidu EasyDL platform, is adopted as a comparative model. Our average recognition accuracy performs approximately 4% better than the Resnet101 model. Based on the excellent performance of convolutional neural network in the field of image recognition, this paper studies the characteristics of gravitational wave signals and obtains a more appropriate recognition model after training and tuning, in order to achieve the purpose of automatic recognition of whether the signal data contain real gravitational wave signals.