IIARD International Institute of Academic Research and Development

INTERNATIONAL JOURNAL OF COMPUTER SCIENCE AND MATHEMATICAL THEORY (IJCSMT )

E-ISSN 2545-5699
P-ISSN 2695-1924
VOL. 9 NO. 4 2023
DOI: https://doi.org/10.56201/ijcsmt.v9.no4.2023.pg23.44

---

Comparative Analysis of Supervised Machine Learning Algorithms for ECG Arrhythmia Detection using Small Dataset

Igiri , C.G, Anireh, V.I.E, Nwiabu, N.D, Matthias, Daniel

---

Abstract

Automatic detection and analysis of arrhythmias from Electrocardiogram (ECG) signal is beginning to take a center stage in recent times, due to delay the nature of ECG signals, coupled with the subjective interpretation of these signals by cardiologist. Arrhythmia detection plays a vital role in diagnosing and managing cardiovascular diseases. With the advancements in machine learning techniques, various supervised algorithms have been employed to improve the accuracy of arrhythmia detection. However, each supervised ML algorithm has its strength and weakness in predicting the various classes of arrhythmia. This research study presents a comparative analysis of five popular supervised machine learning algorithms: support vector machine (SVM), K-Nearest Neighbors (KNN), Random Forest (RF), Naïve Baye (NB) and Decision Tree (DT) when applied to ECG arrhythmia detection with down sampled dataset. The goal is to evaluate and compare the performance of these algorithms in terms of accuracy, precision, recall, and F1 score. The study utilizes the MIT-BIH benchmark dataset, and experimental results provide insights into the strengths and limitations of each algorithm, aiding in the selection of the most suitable algorithm for accurate ECG arrhythmia detection. The Random Forest algorithm outperformed other algorithms in terms of accuracy, achieving an accuracy of 89.9% with RR interval based feature set.

keywords:

Electrocardiogram, Cardiovascular, Arrhythmia, Machine Learning

---

References:

Ahamed, J., Koli, A.M., Ahmad, K., Jamal, A.M., & Gupta, B.B (2021) CDPS-IoT:
Cardiovascular Disease Prediction System Based on IoT Using Machine Learning.
International Journal of Interactive Multimedia and Artificial Intelligence, 7(4). DOI:
10.9781/ijimai.2021.09.002
Acharya, U. R., Fujita, H., Adam, M., Lih, O. S., Sudarshan, V. K., & Tan, J. H., et al. (2016).
Automated characterization and classification of coronary artery disease and myocardial
infarction by decomposition of ecg signals: a comparative study. Information Sciences,
377, (pp.17-29).
Ahamed, J., Koli, A.M., Ahmad, K., Jamal, M.A & Gupta, B.B. (2021). CDPS-IoT:
cardiovascular disease prediction system based on IoT using machine learning.
International Journal of Interactive Multimedia and Artifcial Intelligence, 7(4), 78–86.
Alarsan, F.I., & Younes, M. (2019). Analysis and Classification of Heart Diseases using Heartbeat
Features and Machine Learning Algorithms . Journal of Big Data 6(81).
Asl, B.M., Setarehdan, S.K., & Mohebbi, M. (2008). Support vector machine-based arrhythmia
classification using reduced features of heart rate variability signal. Artificial . Intelligence.
Medical. 44, 51–64.
Batra, A., & Jawa, V. (2016). Classification of arrhythmia using conjunction of machine
learning algorithms and ECG diagnostic criteria. International Journal of Biology.
Biomed. 1, 1–7.

DOWNLOAD PDF

---

Back