Investigating automated regression models for estimating left ventricular ejection fraction levels in heart failure

Sona M Al Younis1, Leontios J Hadjileontiadis1,2, Aamna M Al Shehhi1

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

Plos One
|December 11, 2023
PubMed

Insights

Electrocardiogram (ECG) can estimate left ventricular ejection fraction (LVEF) in heart failure (HF) patients. Machine learning models, particularly Gaussian Process Regression and Decision Trees, show high accuracy in predicting LVEF from ECG data.

Area of Science:

  • Cardiology
  • Biomedical Engineering
  • Machine Learning

Background:

  • Heart Failure (HF) affects millions globally, necessitating accurate Left Ventricular Ejection Fraction (LVEF) assessment for diagnosis and management.
  • Echocardiography is standard for LVEF estimation, but Electrocardiogram (ECG) offers a simpler, cost-effective, and continuous monitoring alternative.
  • Developing non-invasive methods to estimate LVEF from ECG is crucial for widespread HF monitoring.

Purpose of the Study:

  • To compare the efficacy of various regression models (SVM, XGBOOST, GPR, Decision Tree) for estimating LVEF using 24-hour ECG recordings in HF patients.
  • To identify the optimal machine learning models and time intervals for accurate LVEF prediction from ECG data.
  • To explore the potential for an automated screening system for coronary artery disease (CAD) based on ECG-derived LVEF.

Main Methods:

  • Utilized 24-hour ECG recordings from 303 HF patients across a multicentre cohort (American and Greek).
  • Extracted ECG features to train regression models (SVM, XGBOOST, GPR, Decision Tree) at hourly intervals.
  • Implemented hyperparameter tuning using a nested loop approach for model optimization and cross-validation.

Main Results:

  • Rational Quadratic Gaussian Process Regression (GPR) and Fine Decision Tree models achieved the best LVEF estimations.
  • Achieved average Root Mean Square Errors (RMSE) of 3.83% for GPR and 3.42% for Decision Trees, with correlation coefficients of 0.92 and 0.91, respectively (p<0.01).
  • Identified specific circadian time periods (midnight-1 am, 8-9 am, 10-11 pm) with the lowest RMSE between actual and predicted LVEF.

Conclusions:

  • Machine learning models, particularly GPR and Decision Trees, can accurately estimate LVEF from ECG data in heart failure patients.
  • The findings support the development of automated systems for HF monitoring and potentially CAD screening using ECG.
  • Optimizing measurement timings based on circadian rhythms can enhance the accuracy of ECG-based LVEF estimation.