Electrocardiogram Abnormality Detection Using Machine Learning on Summary Data and Biometric Features

Diagnostics (Basel). 2025 Apr 1;15(7):903. doi: 10.3390/diagnostics15070903.

Abstract

Background/Objectives: Electrocardiogram data are widely used to diagnose cardiovascular diseases, a leading cause of death globally. Traditional interpretation methods are manual, time-consuming, and prone to error. Machine learning offers a promising alternative for automating the classification of electrocardiogram abnormalities. This study explores the use of machine learning models to classify electrocardiogram abnormalities using a dataset that combines clinical features (e.g., age, weight, smoking status) with key electrocardiogram measurements, without relying on time-series data. Methods: The dataset included demographic and electrocardiogram-related biometric data. Preprocessing steps addressed class imbalance, outliers, feature scaling, and the encoding of categorical variables. Five machine learning models-Gaussian Naive Bayes, support vector machines, random forest trees, extremely randomized trees, gradient boosted trees, and an ensemble of top-performing classifiers-were trained and optimized using stratified k-fold cross-validation. Model performance was evaluated on a reserved testing set using metrics such as accuracy, precision, recall, and F1-score. Results: The extremely randomized trees model achieved the best performance, with a testing accuracy of 66.79%, recall of 66.79%, and F1-score of 62.93%. Ventricular rate, QRS duration, and QTC (Bezet) were identified as the most important features. Challenges in classifying borderline cases were noted due to class imbalance and overlapping features. Conclusions: This study demonstrates the potential of machine learning models, particularly extremely randomized trees, in classifying electrocardiogram abnormalities using demographic and biometric data. While promising, the absence of time-series data limits diagnostic accuracy. Future work incorporating time-series signals and advanced deep learning techniques could further improve performance and clinical relevance.

Keywords: ECG-related biometrics; ECG/EKG; Extremely randomized trees; gradient boosted trees; support vector machines.