Acute kidney injury (AKI) is a common and often under-recognized complication in intensive care units. Timely detection is critical because delayed recognition narrows the window for effective intervention and worsens outcomes. The authors aimed to create a more transportable and clinically implementable prediction model that uses a compact feature set and offers interpretable outputs. To that end they developed a 37-feature XGBoost classifier trained on MIMIC‑IV with the goals of temporal validation, external transportability, probability calibration, and per-prediction explainability.
The study used two publicly available, de-identified critical care datasets accessible via PhysioNet: MIMIC‑IV v3.1 for model development and internal temporal testing, and the eICU Collaborative Research Database v2.0 for external validation across multiple hospitals. The MIMIC‑IV training cohort had an overall AKI positivity rate of 5.4%. Access to both datasets required completion of CITI training and PhysioNet credentialing; code is available from the corresponding author upon request.
The model used a reduced set of 37 features chosen to balance predictive performance and transportability across different electronic health record ecosystems. The manuscript highlights the intent to avoid the hundreds-to-thousands of features commonly used in other machine learning or deep learning AKI models, which can limit generalisability and integration into clinical workflows. Specific feature names and preprocessing steps were described in the source; when details were not reported in the abstract-level text, those particulars are not restated here.
An XGBoost classifier served as the predictive engine. Hyperparameter tuning was performed using Optuna. To produce well-calibrated risk probabilities suitable for clinical interpretation, model outputs were calibrated via isotonic regression. The authors also used SHAP TreeExplainer to provide local, feature-level explanations for individual predictions.
Internal validation used a patient-level temporal split to emulate prospective deployment. Training was performed on admissions from 2008–2016, and testing on admissions from 2017–2022. This design evaluates temporal generalisability and reduces optimistic bias that can arise when training and testing are not separated by time. On the internal temporal test set the model predicted AKI onset within a 12–24 hour window with an AUROC of 0.794. The reported internal 95% confidence interval for AUROC was 0.789–0.799, indicating narrow estimate variability.
To assess transportability across care settings, the trained MIMIC‑IV model was applied to the eICU Collaborative Research Database that aggregates data from 208 US hospitals. The model was deployed unchanged (no retraining or fine-tuning) and achieved an AUROC of 0.750 on the external dataset. This result demonstrates preserved discrimination when applied across a multi-centre dataset without site-specific retraining.
The authors used SHAP TreeExplainer to generate feature-level explanations for individual predictions produced by the tree-based model. This approach provides interpretable attribution scores that can support clinical review and decision-making for flagged predictions. The paper frames explainability as part of the strategy to facilitate clinical acceptance and auditability of model outputs.
According to the report, the model exhibited equitable discrimination across multiple patient subgroups in both internal and external evaluations. Subgroup analyses included gender, age groups, presence of chronic kidney disease, self-reported race, and AKI stages. The authors concluded that discrimination was consistent across these categories on both datasets; the internal AUROC confidence interval (0.789–0.799) was cited to support estimate stability.
Ethical approval for the work was provided by the Ethics Committee of the School of Computing, Miva Open University. The authors declared no competing interests. Data sources (MIMIC‑IV and eICU CRD) are available via PhysioNet under Data Use Agreements and require standard credentialing; the manuscript states that code is available upon request from the corresponding author. The study adhered to relevant reporting guidelines and confirmed necessary approvals and consents as applicable.
The study suggests that clinically useful AKI prediction systems can be built with a substantially reduced feature set while maintaining acceptable discrimination and transportability. The combination of temporal validation, external multi-centre testing without retraining, probability calibration, and SHAP-based explainability are presented as a pragmatic pathway toward integration into diverse electronic health-record environments. The results support further evaluation and potential prospective implementation studies; specific operational details for deployment and full feature lists were provided in the source manuscript and codebase accessibility is noted as available by request.