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

被引:27
作者
Jahan, Masud Shah [1 ]
Mansourvar, Marjan [1 ,2 ]
Puthusserypady, Sadasivan [3 ]
Wiil, Uffe Kock [1 ]
Peimankar, Abdolrahman [1 ]
机构
[1] Univ Southern Denmark, Maersk Mc Kinney Moller Inst, Ctr Hlth Informat & Technol, DK-5230 Odense, Denmark
[2] Univ Southern Denmark, Dept Math & Comp Sci, DK-5230 Odense, Denmark
[3] Tech Univ Denmark, Dept Hlth Technol, DK-2800 Lyngby, Denmark
关键词
Atrial fibrillation; Cardiac arrhythmias; Classification; Electrocardiogram (ECG); Machine learning; COST;
D O I
10.1016/j.ijmedinf.2022.104790
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
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 F1-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.
引用
收藏
页数:11
相关论文
共 59 条
[1]   Comparison of supervised machine learning classification techniques in prediction of locoregional recurrences in early oral tongue cancer [J].
Alabi, Rasheed Omobolaji ;
Elmusrati, Mohammed ;
Sawazaki-Calone, Iris ;
Kowalski, Luiz Paulo ;
Haglund, Caj ;
Coletta, Ricardo D. ;
Makitie, Antti A. ;
Salo, Tuula ;
Almangush, Alhadi ;
Leivo, Ilmo .
INTERNATIONAL JOURNAL OF MEDICAL INFORMATICS, 2020, 136
[2]   Feature selection using support vector machines and bootstrap methods for ventricular fibrillation detection [J].
Alonso-Atienza, Felipe ;
Luis Rojo-Alvarez, Jose ;
Rosado-Munoz, Alfredo ;
Vinagre, Juan J. ;
Garcia-Alberola, Arcadi ;
Camps-Valls, Gustavo .
EXPERT SYSTEMS WITH APPLICATIONS, 2012, 39 (02) :1956-1967
[3]   A deep learning approach for real-time detection of atrial fibrillation [J].
Andersen, Rasmus S. ;
Peimankar, Abdolrahman ;
Puthusserypady, Sadasivan .
EXPERT SYSTEMS WITH APPLICATIONS, 2019, 115 :465-473
[4]  
Andersen RS, 2017, IEEE ENG MED BIO, P2039, DOI 10.1109/EMBC.2017.8037253
[5]  
[Anonymous], 2017, Classification and regression trees, DOI [DOI 10.1201/9781315139470-8, 10.1201/9781315139470-8]
[6]  
[Anonymous], 2021, IEEE Trans. Broadcast.
[7]  
[Anonymous], 2010, MIT BIH ARRHYTHMIA D
[8]   Automatic detection of atrial fibrillation using stationary wavelet transform and support vector machine [J].
Asgari, Shadnaz ;
Mehrnia, Alireza ;
Moussavi, Maryam .
COMPUTERS IN BIOLOGY AND MEDICINE, 2015, 60 :132-142
[9]   Early hospital mortality prediction of intensive care unit patients using an ensemble learning approach [J].
Awad, Aya ;
Bader-El-Den, Mohamed ;
McNicholas, James ;
Briggs, Jim .
INTERNATIONAL JOURNAL OF MEDICAL INFORMATICS, 2017, 108 :185-195
[10]   TSFEL: Time Series Feature Extraction Library [J].
Barandas, Marilia ;
Folgado, Duarte ;
Fernandes, Leticia ;
Santos, Sara ;
Abreu, Mariana ;
Bota, Patricia ;
Liu, Hui ;
Schultz, Tanja ;
Gamboa, Hugo .
SOFTWAREX, 2020, 11