Enhanced Arrhythmia Detection Using Filtered Data, CNN, Graph Convolutional Networks, and SVM on MIT-BIH and PTB Databases

Mallikarjunamallu K.
DOI: https://doi.org/10.52783/jes.6078
2024-01-25
Abstract:Arrhythmia classification and detection are essential for the early diagnosis of heart diseases, but accurately identifying arrhythmias is challenging due to the inherent noise in electrocardiogram (ECG) data. This study presents a novel method for arrhythmia detection that follows a systematic approach. First, ECG data from the MIT-BIH Arrhythmia Database and the PTB Diagnostic Database are preprocessed using three distinct filters: wavelet transform (WT), finite impulse response (FIR), and an innovative infinite impulse response (IIR) filter to remove noise. The filtered data are then processed through a one-dimensional convolutional neural network (1D-CNN) and graph convolutional networks (GCNs) for feature extraction. These features are initially classified using a support vector machine (SVM) into normal and abnormal categories. For further detection, the classified data undergo additional processing through Short-Time Fourier Transform (STFT) for time-frequency analysis, followed by Principal Component Analysis (PCA) for dimensionality reduction. Finally, a Bidirectional Long Short-Term Memory (BiLSTM) network is employed to detect specific arrhythmias within the classified signals. The proposed method is evaluated using the MIT-BIH and PTB datasets, demonstrating that the combination of STFT, PCA, and BiLSTM significantly enhances the detection of specific arrhythmias. The inclusion of GCNs and the innovative IIR filter further improves classification accuracy. This approach's effectiveness is validated through comparisons with existing arrhythmia identification methods, highlighting its advancements in both classification and detailed arrhythmia detection.
What problem does this paper attempt to address?