» Articles » PMID: 39777058

Prediction of Acute Kidney Injury in Intensive Care Unit Patients Based on Interpretable Machine Learning

Overview
Journal Digit Health
Date 2025 Jan 8
PMID 39777058
Authors
Affiliations
Soon will be listed here.
Abstract

Objective: Acute kidney injury (AKI) poses a lethal risk in intensive care unit (ICU) patients, where early detection is challenging. This study was to establish a prediction model for AKI 24 hours in advance for ICU patients and to help clinicians monitor patients at an early stage by key features.

Methods: In this study, the Medical Information Mart for Intensive Care (MIMIC) databases were used to construct a dataset of critically ill patients. Predictive models were constructed using five machine learning algorithms based on MIMIC-IV data, and the best predictive model was selected by multiple model evaluation metrics. MIMIC-III data were used for external validation. We conducted an interpretability analysis of the model using SHapley Additive exPlanations (SHAP) to clarify key features and decision-making mechanisms.

Results: A total of 18,186 patient data were included in this study. The analysis combining calibration and decision curves demonstrated that the eXtreme Gradient Boosting (XGBoost) exhibited superior performance among the five algorithms, achieving an area under the receiver operating characteristic curve of 0.88. Interpretability analysis based on the XGBoost model showed diuretic use, mechanical ventilation, vasopressor use, age, and antibiotic use as the most important decision factors of the model. The SHAP summary plot was used to illustrate the effects of the top 19 features attributed to the XGBoost.

Conclusions: The XGBoost algorithm can predict the occurrence of AKI more accurately. Interpretative analysis of the model reveals the mechanisms of key features, and reflects the individual differences between patients, providing an important clinical reference.

References
1.
Huang S, Teng Y, Du J, Zhou X, Duan F, Feng C . Internal and external validation of machine learning-assisted prediction models for mechanical ventilation-associated severe acute kidney injury. Aust Crit Care. 2022; 36(4):604-612. DOI: 10.1016/j.aucc.2022.06.001. View

2.
Wang Q, Yu S, Qi X, Hu Y, Zheng W, Shi J . [Overview of logistic regression model analysis and application]. Zhonghua Yu Fang Yi Xue Za Zhi. 2019; 53(9):955-960. DOI: 10.3760/cma.j.issn.0253-9624.2019.09.018. View

3.
Huang H, Liu Y, Wu M, Gao Y, Yu X . Development and validation of a risk stratification model for predicting the mortality of acute kidney injury in critical care patients. Ann Transl Med. 2021; 9(4):323. PMC: 7944298. DOI: 10.21037/atm-20-5723. View

4.
Li J, Liu S, Hu Y, Zhu L, Mao Y, Liu J . Predicting Mortality in Intensive Care Unit Patients With Heart Failure Using an Interpretable Machine Learning Model: Retrospective Cohort Study. J Med Internet Res. 2022; 24(8):e38082. PMC: 9399880. DOI: 10.2196/38082. View

5.
Privratsky J, Fuller M, Raghunathan K, Ohnuma T, Bartz R, Schroeder R . Postoperative Acute Kidney Injury by Age and Sex: A Retrospective Cohort Association Study. Anesthesiology. 2022; 138(2):184-194. PMC: 10439699. DOI: 10.1097/ALN.0000000000004436. View