Classification of Atrial Fibrillation using Random Forest Algorithm
Article Sidebar
Main Article Content
Abstract
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.
Article Details
References
García, Manuel, Juan Ródenas, Raúl Alcaraz, and José J. Rieta. "Atrial fibrillation screening through combined timing features of short single-lead electrocardiograms." In 2017 Computing in Cardiology (CinC), pp. 1-4. IEEE, 2017.
Liu, Yang, Kuanquan Wang, Qince Li, Runnan He, Yong Xia, Zhen Li, Hao Liu, and Henggui Zhang. "Diagnosis of AF based on time and frequency features by using a hierarchical classifier." In 2017 Computing in Cardiology (CinC), pp. 1-4. IEEE, 2017.
Zabihi, Morteza, Ali Bahrami Rad, Aggelos K. Katsaggelos, Serkan Kiranyaz, Susanna Narkilahti, and Moncef Gabbouj. "Detection of atrial fibrillation in ECG hand-held devices using a random forest classifier." In 2017 Computing in Cardiology (CinC), pp. 1-4. IEEE, 2017.
Jiménez-Serrano, Santiago, et al. "Atrial fibrillation detection using feedforward neural networks and automatically extracted signal features." 2017 Computing in Cardiology (CinC). IEEE, 2017.
Yaseen Alkubaisi, A. N. ., Abbas Abood, E. ., & Mohammed, M. H. . (2023). Computational Modelling Applications for the Optimal Design of Prefabricated Industrial Buildings According to the Harmonious Research Method . International Journal of Intelligent Systems and Applications in Engineering, 11(4s), 302–312. Retrieved from https://ijisae.org/index.php/IJISAE/article/view/2668
Behar, Joachim A., et al. "Rhythm and quality classification from short ECGs recorded using a mobile device." 2017 Computing in Cardiology (CinC). IEEE, 2017.
Andreotti, Fernando, et al. "Comparing feature-based classifiers and convolutional neural networks to detect arrhythmia from short segments of ECG." 2017 Computing in Cardiology (CinC). IEEE, 2017.
Billeci, Lucia, et al. "Detection of AF and other rhythms using RR variability and ECG spectral measures." 2017 Computing in Cardiology (CinC). IEEE, 2017.
Bin, Guangyu, et al. "Detection of atrial fibrillation using decision tree ensemble." 2017 Computing in Cardiology (CinC). IEEE, 2017.
V. H. C. de Albuquerque, T. M. Nunes, D. R. Pereira, E. J. D. S. Luz, D. Menotti, J. P. Papa, and J. M. R. Tavares, "Robust automated cardiac arrhythmia detection in ECG beat signals". Neural Computing and Applications, pp. 1-15, 2016.
Alickovic, E., & Subasi, A. Medical decision support system for diagnosis of heart arrhythmia using DWT and random forests classifier. Journal of medical systems, 40(4), 108. 2016.
Shadmand, S., & Mashoufi, B. (2016). A new personalized ECG signal classification algorithm using block-based neural network and particle swarm optimization. Biomedical Signal Processing and Control, 25, 12-23.
Mohapatra, S. K., Palo, H. K., & Mohanty, M. N. (2017). Detection of Arrhythmia using Neural Network. Annals of Computer Science and Information Systems, 14, 97-100.
Mustaqeem, A., Anwar, S. M., Majid, M., & Khan, A. R. (2017, July). Wrapper method for feature selection to classify cardiac arrhythmia. In Engineering in Medicine and Biology Society (EMBC), 2017 39th Annual International Conference of the IEEE (pp. 3656-3659). IEEE.
Shensheng Xu, S., Mak, M. W., & Cheung, C. C. (2017, July). Deep neural networks versus support vector machines for ECG arrhythmia classification. In Multimedia & Expo Workshops (ICMEW), 2017 IEEE International Conference on(pp. 127-132). IEEE.
Zellar, P. I. . (2021). Business Security Design Improvement Using Digitization. International Journal of New Practices in Management and Engineering, 10(01), 19–21. https://doi.org/10.17762/ijnpme.v10i01.98
Pandey, S. K., & Janghel, R. R. (2019). ECG Arrhythmia Classification Using Artificial Neural Networks. In Proceedings of 2nd International Conference on Communication, Computing and Networking (pp. 645-652). Springer, Singapore.
Kher, Rahul. "Signal processing techniques for removing noise from ECG signals." J. Biomed. Eng. Res 3 (2019): 1-9
Rodrigues, Tiago, Sirisack Samoutphonh, Hugo Silva, and Ana Fred. "A Low-Complexity R-peak Detection Algorithm with Adaptive Thresholding for Wearable Devices." In 2020 25th International Conference on Pattern Recognition (ICPR), pp. 1-8. IEEE, 2021.
Gutiérrez-Rivas, Raquel, Juan Jesus Garcia, William P. Marnane, and Alvaro Hernández. "Novel real-time low-complexity QRS complex detector based on adaptive thresholding." IEEE Sensors Journal 15, no. 10 (2015): 6036-6043.
Shaffer, Fred, and James P. Ginsberg. "An overview of heart rate variability metrics and norms." Frontiers in public health 5 (2017): 258.
Veerabhadrappa, S.T., Vyas, A.L. and Anand, S., 2015. Changes in heart rate variability and pulse wave characteristics during normal pregnancy and postpartum. International Journal of Biomedical Engineering and Technology, 17(2), pp.99-114.
Kung BH, Hu PY, Huang CC, Lee CC, Yao CY, Kuan CH. An Efficient ECG Classification System Using Resource-Saving Architecture and Random Forest. IEEE J Biomed Health Inform. 2021 Jun;25(6):1904-1914. doi: 10.1109/JBHI.2020.3035191. Epub 2021 Jun 3. PMID: 33136548.