Application of Interpretable Machine Learning Algorithms to Predict Acute Kidney Injury in Patients with Cerebral Infarction in ICU

Xiaochi Lu,Yi Chen,Gongping Zhang,Xu Zeng,Linjie Lai,Chaojun Qu
DOI: https://doi.org/10.1016/j.jstrokecerebrovasdis.2024.107729
IF: 2.677
2024-04-25
Journal of Stroke and Cerebrovascular Diseases
Abstract:Background Acute kidney injury (AKI) is not only a complication but also a serious threat to patients with cerebral infarction (CI). This study aimed to explore the application of interpretable machine learning algorithms in predicting AKI in patients with cerebral infarction. Methods The study included 3,920 patients with CI admitted to the Intensive Care Unit and Emergency Medicine of the Central Hospital of Lishui City, Zhejiang Province. Nine machine learning techniques, including XGBoost, logistics, LightGBM, random forest (RF), AdaBoost, GaussianNB (GNB), Multi-Layer Perceptron (MLP), support vector machine (SVM), and k-nearest neighbors (KNN) classification, were used to develop a predictive model for AKI in these patients. SHapley Additive exPlanations (SHAP) analysis provided visual explanations for each patient. Finally, model effectiveness was assessed using metrics such as average precision (AP), sensitivity, specificity, accuracy, F1 score, precision-recall (PR) curve, calibration plot, and decision curve analysis (DCA). Results The XGBoost model performed better in the internal validation set and the external validation set, with an AUC of 0.940 and 0.887, respectively. The five most important variables in the model were, in order, glomerular filtration rate, low-density lipoprotein, total cholesterol, hemiplegia and serum kalium. Conclusion This study demonstrates the potential of interpretable machine learning algorithms in predicting CI patients with AKI.
neurosciences,peripheral vascular disease
What problem does this paper attempt to address?