2D Transfer Learning for ECG Classification using Continuous Wavelet Transform

Wei Zhang
DOI: https://doi.org/10.1101/2024.07.11.24310258
2024-07-11
Abstract:Advanced deep neural networks, when trained on extensive datasets, can outperform cardiologists in diagnosing cardiac arrhythmias. However, the availability of large-scale training data is often impractical. This study explores the use of transfer learning to identify and classify three ECG patterns. It applies knowledge gained from 2D image classification tasks to the domain of 1D time-series ECG signal classification. The research leverages various deep learning models to classify continuous wavelet transform (2D representations) of ECG signals. The effectiveness of these transferred deep learning models in classifying ECG time-series data is then evaluated.
What problem does this paper attempt to address?
This paper discusses how to use 2D transfer learning to solve the problem of electrocardiogram (ECG) classification. In the absence of a large amount of training data, advanced deep neural networks may outperform cardiac experts in diagnosing cardiac rhythm abnormalities. The study uses continuous wavelet transform to transform 1D time-series ECG signals into 2D images, and then applies pre-trained deep learning models for classification. In this way, the researchers evaluated the effectiveness of these deep learning models in identifying ECG time-series data. Specifically, the paper uses seven different deep learning architectures, ranging from 41 layers to 708 layers, and selects scalograms (wavelet transform images) that preserve more detailed features instead of other transformation methods. The paper also emphasizes the transfer learning potential of pre-trained ImageNet models in time-series signal classification, especially for ECG signals. In the experiments, the study used a dataset containing 162 ECG records with different cardiac conditions. The results showed that the performance of the deep networks improves as the depth of the architecture increases, but it decreases after a certain point. The best performance was achieved by ResNet-50, with an accuracy of 94.12%. This indicates that selecting a suitable pre-trained architecture for time-series analysis is crucial for medical image classification tasks, especially for ECG signal classification. In conclusion, the paper aims to address the problem of data scarcity in ECG classification through transfer learning, proposing an effective method and emphasizing the importance of selecting appropriate deep learning models. This research not only contributes to ECG analysis but also has potential applications in the detection of other cardiovascular diseases.