Original scientific paper
https://doi.org/10.21857/yq32ohx1v9
Combining Time-Frequency Signal Analysis and Machine Learning with an Example in Gravitational-Wave Detection
Jonatan Lerga
orcid.org/0000-0002-4058-8449
; Faculty of Engineering, University of Rijeka, Rijeka, Croatia
Nikola Lopac
orcid.org/0000-0002-0616-1265
; Faculty of Maritime Studies, University of Rijeka, Rijeka Croatia
David Bačnar
orcid.org/0000-0001-5160-3571
; Faculty of Engineering, University of Rijeka, Rijeka, Croatia
Franko Hržić
orcid.org/0000-0003-1513-0337
; Faculty of Engineering, University of Rijeka, Rijeka, Croatia
Abstract
This paper presents a method for classifying noisy, non-stationary signals in the time-frequency domain using artificial intelligence. The preprocessed time-series signals are transformed into time-frequency representations (TFrs) from Cohen’s class resulting in the TFr images, which are used as input to the machine learning algorithms. We have used three state-of-the-art deep-learning 2d convolutional neural network (Cnn) architectures (ResNet-101, Xception, and EfficientNet). The method was demonstrated on the challenging task of detecting gravitational-wave (gw) signals in intensive real-life, non-stationary, non-gaussian, and non-white noise. The results show excellent classification performance of the proposed approach in terms of classification accuracy, area under the receiver operating characteristic curve (roC auC), recall, precision, F1 score, and area under the precision-recall curve (PR AUC). The novel method outperforms the baseline machine learning model trained on the time-series data in terms of all considered metrics. The study indicates that the proposed technique can also be extended to various other applications dealing with non-stationary data in intensive noise.
Keywords
non-stationary signals; time-frequency representations; artificial intelligence; machine learning; convolutional neural networks; gravitational waves
Hrčak ID:
307575
URI
Publication date:
23.3.2023.
Visits: 753 *