Investigating automated regression models for estimating left ventricular ejection fraction levels in heart failure patients using circadian ECG features

by Sona M. Al Younis, Leontios J. Hadjileontiadis, Aamna M. Al Shehhi, Cesare Stefanini, Mohanad Alkhodari, Stergios Soulaidopoulos, Petros Arsenos, Ioannis Doundoulakis, Konstantinos A. Gatzoulis, Konstantinos Tsioufis, Ahsan H. Khandoker

Heart Failure (HF) significantly impacts approximately 26 million people worldwide, causing disruptions in the normal functioning of their hearts. The estimation of left ventricular ejection fraction (LVEF) plays a crucial role in the diagnosis, risk stratification, treatment selection, and monitoring of heart failure. However, achieving a definitive assessment is challenging, necessitating the use of echocardiography. Electrocardiogram (ECG) is a relatively simple, quick to obtain, provides continuous monitoring of patient’s cardiac rhythm, and cost-effective procedure compared to echocardiography. In this study, we compare several regression models (support vector machine (SVM), extreme gradient boosting (XGBOOST), gaussian process regression (GPR) and decision tree) for the estimation of LVEF for three groups of HF patients at hourly intervals using 24-hour ECG recordings. Data from 303 HF patients with preserved, mid-range, or reduced LVEF were obtained from a multicentre cohort (American and Greek). ECG extracted features were used to train the different regression models in one-hour intervals. To enhance the best possible LVEF level estimations, hyperparameters tuning in nested loop approach was implemented (the outer loop divides the data into training and testing sets, while the inner loop further divides the training set into smaller sets for cross-validation). LVEF levels were best estimated using rational quadratic GPR and fine decision tree regression models with an average root mean square error (RMSE) of 3.83% and 3.42%, and correlation coefficients of 0.92 (p