Towards an explainable machine learning model to reduce readmission risks for diabetes patients

Abstract

Objective:

Hospital readmission of Diabetes patients is a persistent burden on the healthcare industry. Artificial Intelligence (AI) based Machine Learning (ML) techniques offer the potential to predict readmission rates and related risk features for diabetic patients. However, complex machine learning-based solutions are often not explainable and hard to understand for the relevant parties. To this end, this study designs and implements an explainable model to predict readmission rates and identify the risk factors associated with readmission in patients with diabetes.

Methods:

The model employs various explainable visualization techniques, including the permutation importance plot, partial dependence plot (PDP), SHapley Additive exPlanations (SHAP), and interpretable classifiers on a publicly available dataset from US hospitals.

Results:

The bagging random forest model yields the best results, achieving 89% accuracy and 67% precision.

Conclusion:

The explainability visualization techniques reveal that the number of inpatient admissions and emergency visits in a year is the two most critical risk factors for the readmission rate of diabetic patients.