Prediction of Hospital Mortality in Sepsis-Associated Acute Kidney Injury using a Machine-Learning Approach: A Multi-Center Study Using SHAP Interpretability Analysis
作者
Jiawei Lai,Lujiao Mo,Xiaoyuan Shen,Riliang Fang
出处
期刊:Ndt Plus [Oxford University Press] 日期:2025-11-28
Abstract Background Sepsis-associated acute kidney injury (S-AKI) represents a critical complication with high mortality rates in intensive care units. Current risk stratification tools lack precision and interpretability for clinical decision-making. This study aimed to develop and validate interpretable machine learning models for predicting hospital mortality in S-AKI patients. Methods This retrospective cohort study utilized five international critical care databases: MIMIC-IV (n=12,966), MIMIC-III-CareVue (n=2,209), eICU (n=8,210), NWICU (n=2,207), and SICdb (n=1,893). Adult patients with S-AKI meeting sepsis-3.0 and acute kidney injury criteria were included. Feature selection used the Boruta algorithm on MIMIC-IV, MIMIC-III, and eICU databases. Eleven machine learning algorithms were trained using MIMIC-IV data with external validation on all other datasets. Performance was evaluated using ROC analysis, calibration plots, and decision curve analysis. SHAP analysis provided model interpretability. Results Among 27,485 S-AKI patients, hospital mortality was 27.5%. Boruta identified 21 consensus features including severity scores (SAPS II, SOFA, OASIS), vital signs, and laboratory parameters. Gradient Boosting Machine emerged as optimal with AUC values of 0.770 (training), 0.731 (internal validation), and 0.732-0.778 across four external validation cohorts. The model demonstrated excellent calibration and minimal overfitting (3.9% AUC difference). Decision curve analysis revealed superior clinical utility across probability thresholds of 4-82%. SHAP analysis identified SAPS II as the most important predictor, with scores >60 and SOFA >15 associated with substantially increased mortality risk. Complete case analysis confirmed model robustness (AUC 0.766-0.847). Conclusions The interpretable machine learning model demonstrated excellent performance and robust generalizability for S-AKI mortality prediction across five international databases. SHAP analysis provided clinically meaningful insights supporting personalized risk stratification and evidence-based clinical decision-making.