Electrocardiogram Abnormality Detection Using Machine Learning on Summary Data and Biometric Features
<b>Background/Objectives:</b> 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. <b>Methods:</b> 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. <b>Results:</b> 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. <b>Conclusions:</b> 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.
Authors
- Michael Nizich (ORCID: https://orcid.org/0000-0002-3922-5659)
- Michael Ruane
- Milan Toma (ORCID: https://orcid.org/0000-0003-2154-3728)
- A. Robert Singh
- Daniel Nasef
- C Hartnett
- Kennette James Basco (ORCID: https://orcid.org/0009-0005-7688-9349)
- Jason Tagliarino
Institutions
- New York Institute of Technology (US)
Publication Details
- Journal
- Diagnostics
- Published
- 2025-04-01
- DOI
- https://doi.org/10.3390/diagnostics15070903
- Citations
- 2
- Primary Topic
- ECG Monitoring and Analysis
- Type
- article
- Field-Weighted Citation Impact
- 2.78