Background: Acute kidney injury (AKI) and acute kidney disease (AKD) are prevalent among pediatric patients, both linked to increased mortality and extended hospital stays. Early detection of kidney injury is crucial for improving outcomes. This study presents a machine learning-based risk prediction model for AKI and AKD in pediatric patients, enabling personalized risk predictions.
Methods: Data from 2,346 hospitalized pediatric patients, collected between January 2020 and January 2023, were divided into an 85% training set and a 15% test set. Predictive models were constructed using eight machine learning algorithms and two ensemble algorithms, with the optimal model identified through AUROC. SHAP was used to interpret the model, and an online prediction tool was developed with Streamlit to predict AKI and AKD.
Results: The incidence of AKI and AKD were 14.90% and 16.26%, respectively. Patients with AKD combined with AKI had the highest mortality rate, at 6.94%, when analyzed by renal function trajectories. The LightGBM algorithm showed superior predictive performance for both AKI and AKD (AUROC: 0.813, 0.744). SHAP identified top predictors for AKI as serum creatinine, white blood cell count, neutrophil count, and lactate dehydrogenase, while key predictors for AKD included proton pump inhibitor, blood glucose, hemoglobin, and AKI grade.
Conclusion: The high incidence of AKI and AKD among hospitalized children warrants attention. Renal function trajectories are strongly associated with prognosis. Supported by a web-based tool, machine learning models can effectively predict AKI and AKD, facilitating early identification of high-risk pediatric patients and potentially improving outcomes.
Keywords: Pediatric; acute kidney disease; acute kidney injury; machine learning; prediction model; renal function trajectory.