Clinical application of artificial intelligence algorithm for prediction of one-year mortality in heart failure patients.

ABSTRACT

Risk prediction for heart failure (HF) using machine learning methods (MLM) has not yet been established at practical application levels in clinical settings. This study aimed to create a new risk prediction model for HF with a minimum number of predictor variables using MLM. We used two datasets of hospitalized HF patients retrospective data for creating the model and prospectively registered data for model validation. Critical clinical events (CCEs) were defined as death or LV assist device implantation within 1 year from the discharge date. We randomly divided the retrospective data into training and testing datasets and created a risk prediction model based on the training dataset (MLM-risk model). The prediction model was validated using both the testing dataset and the prospectively registered data. Finally, we compared predictive power with published conventional risk models. In the patients with HF (n = 987), CCEs occurred in 142 patients. In the testing dataset, the substantial predictive power of the MLM-risk model was obtained (AUC = 0.87). We generated the model using 15 variables. Our MLM-risk model showed superior predictive power in the prospective study compared to conventional risk models such as the Seattle Heart Failure Model (c-statistics 0.86 vs. 0.68, p < 0.05). Notably, the model with an input variable number (n = 5) has comparable predictive power for CCE with the model (variable number = 15). This study developed and validated a model with minimized variables to predict mortality more accurately in patients with HF, using a MLM, than the existing risk scores.