Short-term atrial fibrillation detection using electrocardiograms: A comparison of machine learning approaches

Masud Shah Jahan; Marjan Mansourvar; Sadasivan Puthusserypady; Uffe Kock Wiil; Abdolrahman Peimankar

doi:10.1016/j.ijmedinf.2022.104790

Short-term atrial fibrillation detection using electrocardiograms: A comparison of machine learning approaches

Int J Med Inform. 2022 Jul:163:104790. doi: 10.1016/j.ijmedinf.2022.104790. Epub 2022 May 7.

Authors

Masud Shah Jahan¹, Marjan Mansourvar², Sadasivan Puthusserypady³, Uffe Kock Wiil¹, Abdolrahman Peimankar⁴

Affiliations

¹ Centre of Health Informatics and Technology, The Mærsk Mc-Kinney Møller Institute, University of Southern Denmark, 5230 Odense, Denmark.
² Centre of Health Informatics and Technology, The Mærsk Mc-Kinney Møller Institute, University of Southern Denmark, 5230 Odense, Denmark; Department of Mathematics and Computer Science, University of Southern Denmark, 5230 Odense, Denmark.
³ Department of Health Technology, Technical University of Denmark, 2800 Kgs. Lyngby, Denmark.
⁴ Centre of Health Informatics and Technology, The Mærsk Mc-Kinney Møller Institute, University of Southern Denmark, 5230 Odense, Denmark. Electronic address: abpe@mmmi.sdu.dk.

PMID: 35552189
DOI: 10.1016/j.ijmedinf.2022.104790

Abstract

Background: Atrial fibrillation (AF) is one of the most prevalent cardiac arrhythmias, which challenges the healthcare systems globally.Timely detection of AF can potentially reduce the mortality and morbidity rates as well as alleviate the economic burden caused by this.Digital solutions are shown to enhance the diagnosis of cardiac abnormalities.

Objectives: By the latest advancements in the field of medical informatics and tele-health monitoring, huge amount of electro-physiological signals, such as electrocardiograms (ECG), can be easily collected.One of the most common ways for physicians/cardiologists to analyse these signals is through visual inspection.However, it is not always easy and in most cases cumbersome to analyse these big amounts of ECG data.Therefore, it is of great interest to develop models that are capable of analyzing these data and help physicians making better decisions.This paper proposes and compares well-known machine learning (ML) algorithms to diagnose short episodes of AF. This also paves the way for real-time detection of AF in clinical settings.

Methods: Different ML algorithms such as Support Vector Machine (SVM), Decision Tree (DT), Random Forest (RF), Stacking Classifier (SC), Extreme Gradient Boosting (XGBoost), and Adaptive Boosting (AdaBoost) were applied to detect AF. These models were trained using extracted statistical features from ECG signals.

Results: The proposed ML models were trained on a dataset with 23 ECG records of length approximately 10 h each using leave one group out cross validation (LOGO-CV) technique and achieved the best sensitivity (Se), specificity (Sp), positive predictive value (PPV), false positive rate (FPR), and F₁-score of 85.67%, 81.25%, 90.85%, 18.75% and 88.18%, respectively, to classify AF from normal sinus rhythms (NSR) in short ECG segments of 20 heartbeats.Additionally, the models were examined on three unseen datasets, namely the Long Term AF dataset, MIT-BIH Arrhythmia dataset, and MIT-BIH Normal Sinus Rhythm dataset, to assess their robustness and generalization.

Conclusion: The obtained results show high performance and flexibility of some of the applied ML models compared to other well-known algorithms. In general, the empirical results confirm that ensemble methods, such as AdaBoost, generalized well and perform better than other approaches.

Keywords: Atrial fibrillation; Cardiac arrhythmias; Classification; Electrocardiogram (ECG); Machine learning.

MeSH terms

Algorithms
Atrial Fibrillation* / diagnosis
Electrocardiography / methods
Heart Rate
Humans
Support Vector Machine