Comparative Analysis of Machine Learning Techniques for Arrhythmia Detection

Abstract

Cardiovascular arrhythmia is indeed one of the most prevalent cardiac issues globally. In this paper, the primary objective was to develop and evaluate an automated classification system. This system utilizes a comprehensive database of electro- cardiogram (ECG) data, with a particular focus on improving the detection of minority arrhythmia classes. In this study, the focus was on investigating the performance of three different supervised machine learning models in the context of arrhythmia detection. These models included Support Vector Machine (SVM), Logistic Regression (LR) and Random Forest (RF). An analysis was conducted using real inter-patient electrocardiogram (ECG) records, which is a more realistic scenario in a clinical environment where ECG data comes from various patients. The study evaluated the models’ performances based on four important metrics: accuracy, precision, recall, and f1-score. After thorough experimentation, the results highlighted that the Random Forest (RF) classifier outperformed the other methods in all of the metrics used in the experiments. This classifier achieved an impressive accuracy of 0.94, indicating its effectiveness in accurately detecting arrhythmia in diverse ECG signals collected from different patients.

Keywords

• arrhythmia
• electrocardiography (ecg)
• machine learning
• support vector machine (svm)
• logistic regression (lg)
• random forest (rf).