Comparison of linear and non-linear machine learning models for time-dependent readmission or mortality prediction among hospitalized heart failure patients

Abstract

Although many models are available to predict prognosis of heart failure patients, most tools combining survival analysis are based on proportional hazard model. Non-linear machine learning algorithms would overcome the limitation of the time-independent hazard ratio assumption and provide more information in readmission or mortality prediction among heart failure patients.The present study collected the clinical information of 1796 hospitalized heart failure patients surviving during hospitalization in a Chinese clinical center from December 2016 to June 2019. A traditional multivariate Cox regression model and three machine learning survival models were developed in derivation cohort. Uno's concordance index and integrated Brier score in validation cohort were calculated to evaluate the discrimination and calibration of different models. Time-dependent AUC and Brier score curves were plotted to assess the performance of models at different time phases.

Keywords

• Heart failure
• Readmission and mortality
• Survival analysis
• Machine learning