Leveraging Machine Learning for Developing and Validating a Neonatal Acute Kidney Injury Prediction Model (NEPHRO): A Comprehensive Evidence-Based Neonatal AKI Risk Stratification Tool
Mohamed, T. H.; Bambach, S.; Spencer, J. D.; Rust, L.; Patel, S.; magers, j.; Neyra, J.; Wilson, F. P.; Ning, X.; Newland, J.; Rust, S.; Slaughter, J. L.
Show abstract
BackgroundAcute kidney injury (AKI) is a serious and common complication among critically ill neonates. Preventing or treating AKI early requires timely prediction, but current tools to forecast AKI in neonates are limited. We proposed that machine learning could help predict AKI by analyzing routinely collected clinical data. MethodsWe conducted a retrospective analysis of 8,059 critically ill neonates admitted to our level IV neonatal intensive care unit (NICU) from January 2017 to December 2021 for development and cross-validation (n=5,443, 68%), and from January 2022 to June 2024 (n=2,616, 32%) for temporally validating a neonatal AKI predictive model. Risk factors for model input were identified from the literature, and data were extracted from electronic health records. According to the neonatal modification of kidney disease: Improving Global Outcomes criteria, an increase in serum creatinine (SCr) and/or a decrease in urine output (UOP) defines neonatal AKI. We trained a machine learning model using a least absolute shrinkage and selection operator (LASSO) algorithm to develop and validate a predictive AKI model. The area under the receiver operating characteristic curve (AUROC) and F1-scores evaluated the models performance. FindingsAmong 206,220 NICU patient days, AKI occurred in 881 (11%) neonates. Using 27 potential AKI risk factors, LASSO identified key AKI predictors: fluid balance, hypotension requiring vasopressors, invasive ventilation, sepsis, surgical procedures, and congenital kidney and urinary tract anomalies. The model predicted the occurrence of a critical SCr increase or UOP decrease over the next 48 hours, with an AUROC of 0.814 (95% CI: 0.787-0.843) in development and 0.815 (0.795-0.834) in validation datasets. InterpretationWe developed a machine learning-based model that reliably predicted neonatal AKI before it became clinically apparent by conventional parameters. By identifying high-risk neonates earlier, timely interventions can be deployed to improve outcomes.
Matching journals
The top 5 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Evaluating the kidney disease progression using a comprehensive patient profiling algorithm: A hybrid clustering approach 94%
- Ranked severe maternal morbidity index for population-level surveillance at delivery hospitalization based on hospital discharge data 94%
- A comparison of machine learning models versus clinical evaluation for mortality prediction in patients with sepsis 93%
Similar papers in this journal
- Predicting Prognosis in COVID-19 Patients using Machine Learning and Readily Available Clinical Data 93%
- Predicting mortality in SARS-COV-2 (COVID-19) positive patients in the inpatient setting using a Novel Deep Neural Network 91%
- Identification of an ANCA-Associated Vasculitis Cohort Using Deep Learning and Electronic Health Records 91%
Similar papers in this journal
- AKI Risk Score (AKI-RiSc): Developing an Interpretable Clinical Score for Early Identification of Acute Kidney Injury for Patients Presenting to the Emergency Department 95%
- Machine learning vs. traditional regression analysis for fluid overload prediction in the ICU 93%
- Predicting bloodstream infection outcome using machine learning 93%
Similar papers in this journal
- Prediction models for post-discharge mortality among under-five children with suspected sepsis in Uganda: A multicohort analysis 93%
- Improving the quality of in-patient neonatal routine data as a pre-requisite for monitoring and improving quality of care at scale: A multi-site retrospective cohort study in Kenyan hospitals 92%
- Real world external validation of metabolic gestational age assessment in Kenya 91%
Similar papers in this journal
- A new approach to recognize term and preterm infants with impaired kidney function (IKF) during the first week of life. 93%
- Development and Validation of a Convolutional Neural Network Model for ICU Acute Kidney Injury Prediction 93%
- Shared Decision-Making in Renal Replacement Therapy Selection: Patient Perceptions, Preferences, and Influencing Factors in a Nationwide Cross-Sectional Study in Japan 90%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.