RR Interval-based Atrial Fibrillation Detection using Traditional and Ensemble Machine Learning Algorithms

J Med Signals Sens. 2023 Jul 12;13(3):224-232. doi: 10.4103/jmss.jmss_4_22. eCollection 2023 Jul-Sep.

Abstract

Atrial fibrillation (AF) is a life threatening disease and can cause stroke, heart failure, and sometimes death. To reduce the rate of mortality and morbidity due to increased prevalence of AF, early detection of the same becomes a prior concern. Traditional machine learning (TML) algorithms and ensemble machine learning (EML) algorithms are proposed to detect AF in this article. The performances of both these methods are compared in this study. Methodology involves computation of RR interval features extracted from electrocardiogram and its classification into: normal, AF, and other rhythms. TML techniques such as Classification and Regression Tree, K Nearest Neighbor, C4.5, Iterative Dichotomiser 3, Support Vector Machine and EML classifier such as Random Forest (RF), and Rotation Forest are used for classification. The proposed method is evaluated using PhysioNet challenge 2017. During the tenfold cross validation, it is observed that RF classifier provided good classification accuracy of 99.10% with area under the curve of 0.998. Apart from contributing a new methodology, the proposed study also experimentally proves higher performance with ensemble learning method, RF. The methodology has many applications in health care management systems including defibrillators, cardiac pacemakers, etc.

Keywords: Atrial fibrillation; C4.5; Discrete wavelet transform; Electrocardiogram; Iterative Dichotomiser 3; K-NN; Random Forest; Support Vector Machine; area under the curve; classification and regression tree; rotation forest.