Automated Classification of Cardiac Arrhythmia using Short-Duration ECG Signals and Machine Learning

Abstract

Accurate detection of cardiac arrhythmias is crucial for preventing premature deaths. The current study employs a dual-stage Discrete Wavelet Transform (DWT) and a median filter to eliminate noise from ECG signals. Subsequently, ECG signals are segmented, and QRS regions are extracted for further preprocessing. The study considers five cardiac arrhythmias: normal beats, Premature Ventricular Contractions (PVC), Premature Atrial Contractions (PAC), Right Bundle Branch Block (R-BBB), and Left Bundle Branch Block (L-BBB) for classification. Nine distinct temporal features are extracted from the segmented QRS complex. These features are then applied to six different classifiers for arrhythmia classification. The classifiers' performance is evaluated using the MIT-BIH Arrhythmia Database (MIT-BIH AD). Support Vector Machine (SVM) and Ensemble Tree classifiers demonstrate superior performance in classifying the five different classes. Particularly, the Support Vector Machine classifier achieves high sensitivity (97.44%), specificity (99.36%), positive predictive value (97.44%), and accuracy (98.97%) with a Gaussian kernel. This comprehensive approach, integrating preprocessing, and feature extraction, holds promise for improving automatic cardiac arrhythmia classification in clinical trials.

Keywords: Cardiac arrhythmia; ECG; L-BBB; Machine learning; PAC; PVC; R-BBB.