Interpretable machine learning model for predicting acute kidney injury in critically ill patients

Xunliang Li,Peng Wang,Yuke Zhu,Wenman Zhao,Haifeng Pan,Deguang Wang
DOI: https://doi.org/10.1186/s12911-024-02537-9
IF: 3.298
2024-06-02
BMC Medical Informatics and Decision Making
Abstract:This study aimed to create a method for promptly predicting acute kidney injury (AKI) in intensive care patients by applying interpretable, explainable artificial intelligence techniques.
medical informatics
What problem does this paper attempt to address?
The paper aims to create a method for quickly predicting the risk of acute kidney injury (AKI) in intensive care unit (ICU) patients by applying interpretable artificial intelligence techniques. Specifically, the study utilizes machine learning (ML) techniques to build six predictive models based on data from ICU patients in the MIMIC IV database. These models include logistic regression, support vector machine (SVM), K-nearest neighbors (KNN), decision tree, random forest (RF), and extreme gradient boosting (XGBoost). The main objective of the study is to evaluate the performance of these models in predicting AKI and to enhance the interpretability and transparency of the models using Local Interpretable Model-agnostic Explanations (LIME) and Shapley Additive Explanations (SHAP). The study results show that the XGBoost model performs the best in predicting AKI in ICU patients, with an area under the curve (AUC) of 0.816. The top four factors influencing the XGBoost model's prediction of AKI risk are Sequential Organ Failure Assessment (SOFA) score, weight, mechanical ventilation, and Simplified Acute Physiology Score II (SAPS II). Overall, the study provides a powerful tool for early warning of AKI in ICU patients by constructing machine learning models based on clinical features. This can help doctors take timely interventions to reduce the incidence and severity of AKI.