Predicting Six-Month Re-Admission Risk in Heart Failure Patients Using Multiple Machine Learning Methods: A Study Based on the Chinese Heart Failure Population Database

Abstract

Since most patients with heart failure are re-admitted to the hospital, accurately identifying the risk of re-admission of patients with heart failure is important for clinical decision making and management. This study plans to develop an interpretable predictive model based on a Chinese population for predicting six-month re-admission rates in heart failure patients. Research data were obtained from the PhysioNet portal. To ensure robustness, we used three approaches for variable selection. Six different machine learning models were estimated based on selected variables. The ROC curve, prediction accuracy, sensitivity, and specificity were used to evaluate the performance of the established models. In addition, we visualized the optimized model with a nomogram. In all, 2002 patients with heart failure were included in this study. Of these, 773 patients experienced re-admission and a six-month re-admission incidence of 38.61%. Based on evaluation metrics, the logistic regression model performed best in the validation cohort, with an AUC of 0.634 (95%CI: 0.599–0.646) and an accuracy of 0.652. A nomogram was also generated. The established prediction model has good discrimination ability in predicting. Our findings are helpful and could provide useful information for the allocation of healthcare resources and for improving the quality of survival of heart failure patients.