Development of an open-access and explainable machine learning prediction system to assess the mortality and recurrence risk factors of Clostridioides difficile infection patients

Abstract

Identifying Clostridioides difficile infection (CDI) patients at risk of mortality or recurrence facilitates prevention, timely treatment, and improves clinical outcomes. The aim herein is to establish an open‐access web‐based prediction system, which estimates CDI patients’ mortality and recurrence outcomes and explains machine learning prediction with patients’ characteristics. Prognostic models are developed using four various types of machine learning algorithms and the statistical logistics regression model utilizing over 15 000 CDI patients from 41 hospitals in Hong Kong. The boosting‐based machine learning algorithm gradient boosting machine (GBM) (Mortality AUC: 0.7878; Recurrence AUC: 0.7076) outperforms statistical models (Mortality AUC: 0.7573; Recurrence AUC: 0.6927) and other machine learning algorithms. As the difficulty to interpret complex machine learning results limits their use in the medical area, Shapley additive explanations (SHAP) are adapted to identify which features are crucial to the machine learning models and associate them with clinical findings. SHAP analysis shows that older age, reduced albumin levels, higher creatinine levels, and higher white blood cell count are the most highly associated mortality features, which is consistent with existing clinical findings. The open‐access prediction system for clinicians to assess and interpret the risk factors of CDI patients is now available at https://www.cdiml.care/.