Classification of Atrial Fibrillation using Random Forest Algorithm

The electrocardiogram is indicates the electrical activity of the heart and it can be used to detect cardiac arrhythmias. In the present work, we exhibited a methodology to classify Atrial Fibrillation (AF), Normal rhythm, and Other abnormal ECG rhythms using a machine learning algorithm by analyzing single-lead ECG signals of short duration. First, the events of ECG signals will be detected, after that morphological features and HRV features are extracted. Finally, these features are applied to the Random Forest classifier to perform classification. The Physionet challenge 2017 dataset with more than 8500 ECG recordings is used to train our model. The proposed methodology yields an F1 score of 0.86, 0.97, and 0.83 respectively in classifying AF, normal, other rhythms, and an accuracy of 0.91 after performing a 5-fold cross-validation