Prediction of heart failure patients with distinct left ventricular ejection fraction levels using circadian ECG

Sona M Al Younis1, Leontios J Hadjileontiadis1,2, Ahsan H Khandoker1

  • 1Department of Biomedical Engineering, Healthcare Engineering Innovation Centre (HEIC), Khalifa University, Abu Dhabi, United Arab Emirates.

Plos One
|May 13, 2024
PubMed

Insights

Machine learning models accurately classify heart failure (HF) patients using electrocardiograms (ECG). Decision Tree and KNN models achieved over 90% accuracy, identifying optimal times for screening coronary artery disease (CAD) patients.

Area of Science:

  • Cardiology
  • Biomedical Engineering
  • Artificial Intelligence in Medicine

Background:

  • Heart failure (HF) is a growing global health concern with increasing prevalence and healthcare costs.
  • Accurate classification of HF patients into reduced (HFrEF), mid-range (HFmEF), and preserved (HFpEF) ejection fraction categories is crucial for management.
  • Echocardiography is standard for ejection fraction assessment, but electrocardiograms (ECG) offer a cost-effective, continuous alternative.

Purpose of the Study:

  • To evaluate machine learning (ML) models for classifying left ventricular ejection fraction (LVEF) in HF patients using 24-hour ECG recordings.
  • To compare the performance of K-nearest neighbors (KNN), neural networks (NN), support vector machines (SVM), and decision trees (TREE) for HF classification.
  • To identify optimal time intervals for ECG-based HF classification, potentially aiding in automated screening for coronary artery disease (CAD) patients.

Main Methods:

  • Utilized a multicenter dataset of 303 HF patients (HFpEF, HFmEF, HFrEF) from American and Greek populations.
  • Extracted features from 24-hour ECG recordings and trained ML models (KNN, NN, SVM, TREE) at hourly intervals.
  • Employed nested cross-validation for hyperparameter tuning to optimize LVEF classification accuracy in CAD patients.

Main Results:

  • Decision Tree (TREE) and KNN models demonstrated superior performance, achieving 91.2% and 90.9% accuracy, respectively.
  • Both TREE and KNN models attained high average area under the receiver operating characteristics curve (AUROC) of 0.98 and 0.99.
  • Peak classification accuracy was observed during specific time windows: midnight-1 am, 8-9 am, and 10-11 pm, suggesting circadian influences.

Conclusions:

  • ML models, particularly TREE and KNN, can effectively classify LVEF in HF patients using ECG data.
  • ECG-based ML classification offers a promising, non-invasive, and cost-effective approach for HF patient stratification.
  • The findings support the development of an automated screening system for CAD patients, leveraging optimized ECG measurement timings aligned with circadian rhythms.