ECG classification using Deep CNN and Gramian Angular Field

Youssef Elmir,Yassine Himeur,Abbes Amira
2023-07-25
Abstract:This paper study provides a novel contribution to the field of signal processing and DL for ECG signal analysis by introducing a new feature representation method for ECG signals. The proposed method is based on transforming time frequency 1D vectors into 2D images using Gramian Angular Field transform. Moving on, the classification of the transformed ECG signals is performed using Convolutional Neural Networks (CNN). The obtained results show a classification accuracy of 97.47% and 98.65% for anomaly detection. Accordingly, in addition to improving the classification performance compared to the state-of-the-art, the feature representation helps identify and visualize temporal patterns in the ECG signal, such as changes in heart rate, rhythm, and morphology, which may not be apparent in the original signal. This has significant implications in the diagnosis and treatment of cardiovascular diseases and detection of anomalies.
Signal Processing,Information Retrieval,Machine Learning
What problem does this paper attempt to address?
The main objective of this paper is to improve the classification performance of electrocardiogram (ECG) signals by introducing a new feature representation method. Specifically, the study proposes a method based on Gramian Angular Field (GAF) transformation, which converts one-dimensional time-frequency vectors into two-dimensional images, and then uses convolutional neural networks (CNN) to classify these images. This approach not only enhances classification accuracy but also helps identify and visualize temporal patterns in ECG signals, such as changes in heart rate, rhythm, and morphology, which are significant for the diagnosis and treatment of cardiovascular diseases. The main contributions include: 1. **Feature Representation**: Using GAF to convert ECG signals into two-dimensional images. 2. **Classification Model**: Employing various CNN architectures (such as CNN, VGG-16, ResNet50, and EfficientNet) for classification. 3. **Experimental Results**: Achieving a classification accuracy of 97.47% on the MIT-BIH Arrhythmia dataset and 98.65% on the PTB Diagnostic ECG dataset. 4. **Visualization Analysis**: Making it easier to identify abnormal patterns in ECG signals through GAF images. These results indicate that the combination of GAF and CNN methods performs excellently in ECG signal classification tasks and exhibits good robustness and generalization ability.