An Ensemble Machine Learning Model for Early Prediction of Vancomycin-Induced Acute Kidney Injury in ICU Patients
Archives of Academic Emergency Medicine,
Vol. 13 No. 1 (2025),
6 September 2025
,
Page e45
https://doi.org/10.22037/aaemj.v13i1.2560
Abstract
Introduction: Acute Kidney Injury (AKI) is a severe complication of vancomycin treatment due to its nephrotoxic effects. However, research on predicting AKI in this high-risk group remains limited. This study presents a stacking ensemble machine learning model designed to predict the onset of AKI in this patient population.
Methods: Leveraging data from 314 ICU patients, the model incorporates SHapley Additive exPlanations (SHAP) for enhanced interpretability, identifying key predictors such as serum creatinine levels, glucose variability, and patient age. The model achieved an Area Under the Curve (AUC) of 0.94, outperforming existing predictive approaches. By utilizing readily available clinical data and determining an optimal temporal prediction window, this model facilitates proactive clinical decision-making, aiming to reduce the risk of AKI and improve patient outcomes.
Results: The stacking ensemble model achieved 92\% accuracy, 93\% precision, 92\% sensitivity, and 0.94 AUC in 314 ICU patients, pinpointing creatinine, glucose variability, and age as critical AKI predictors.
Conclusion: The findings suggest that integrating advanced machine learning techniques with interpretable artificial intelligence (AI) can provide a scalable and cost-effective solution for early AKI detection in diverse healthcare settings.
- Acute kidney injury
- Vancomycin
- machine learning
- Prediction
- Intensive Care Unit
How to Cite
References
Chen Y-W, Wu M-Y, Mao C-H, Yeh Y-T, Chen T-T, Liao C-T, et al. Severe acute kidney disease is associated with worse kidney outcome among acute kidney injury patients. Scientific Reports. 2022;12(1).
Kellum JA, Romagnani P, Ashuntantang G, Ronco C, Zarbock A, Anders H-J. Acute kidney injury. Nature Reviews Disease Primers. 2021;7(1):52.
Pickkers P, Darmon M, Hoste E, Joannidis M, Legrand M, Ostermann M, et al. Acute kidney injury in the critically ill: an updated review on pathophysiology and management. Intensive Care Medicine. 2021;47(8):835-50.
Ronco C, Bellomo R, Kellum JA. Acute kidney injury. The Lancet. 2019;394(10212):1949-64.
Khwaja A. KDIGO Clinical Practice Guidelines for Acute Kidney Injury. Nephron Clinical Practice. 2012;120(4):c179-c84.
Liu C, Bayer A, Cosgrove SE, Daum RS, Fridkin SK, Gorwitz RJ, et al. Clinical Practice Guidelines by the Infectious Diseases Society of America for the Treatment of Methicillin-Resistant Staphylococcus aureus Infections in Adults and Children. Clinical Infectious Diseases. 2011;52(3):e18-e55.
Kwak S, Kim JY, Cho H. Vancomycin-induced nephrotoxicity in non-intensive care unit pediatric patients. Scientific Reports. 2021;11(1).
Yang Z, Wang C, Wang H, Wang S, Liu R, Wang X, et al. Cross-sectional survey on adult acute kidney injury in Chinese ICU: the study protocol (CARE-AKI). BMJ Open. 2018;8(6):e020766.
Filippone E, Kraft W, Farber J. The Nephrotoxicity of Vancomycin. Clinical Pharmacology & Therapeutics. 2017;102(3):459-69.
Van Hal SJ, Paterson DL, Lodise TP. Systematic Review and Meta-Analysis of Vancomycin-Induced Nephrotoxicity Associated with Dosing Schedules That Maintain Troughs between 15 and 20 Milligrams per Liter. Antimicrobial Agents and Chemotherapy. 2013;57(2):734-44.
Notice. Kidney International Supplements. 2012;2(1):1.
Bindra S, Jain R. Artificial intelligence in medical science: a review. Irish journal of medical science. 2023;193.
Boursalie O, Samavi R, Doyle TE. Machine Learning and Mobile Health Monitoring Platforms: A Case Study on Research and Implementation Challenges. Journal of Healthcare Informatics Research. 2018;2(1):179-203.
Alyousef AA, Nihtyanova S, Denton C, Bosoni P, Bellazzi R, Tucker A. Nearest Consensus Clustering Classification to Identify Subclasses and Predict Disease. Journal of Healthcare Informatics Research. 2018;2(4):402-22.
Hashemi S, Yousefzadeh Z, Abin AA, Ejmalian A, Nabavi S, Dabbagh A. Machine Learning-Guided Anesthesiology: A Review of Recent Advances and Clinical Applications. Journal of Cellular & Molecular Anesthesia. 2024;9(1).
Nikkinen O, Kolehmainen T, Aaltonen T, Jämsä E, Alahuhta S, Vakkala M. Developing a supervised machine learning model for predicting perioperative acute kidney injury in arthroplasty patients. Computers in Biology and Medicine. 2022;144:105351.
Gao W, Wang J, Zhou L, Luo Q, Lao Y, Lyu H, et al. Prediction of acute kidney injury in ICU with gradient boosting decision tree algorithms. Computers in Biology and Medicine. 2022;140:105097.
Ejmalian A, Aghaei A, Nabavi S, Abedzadeh Darabad M, Tajbakhsh A, Abin AA, et al. Prediction of Acute Kidney Injury After Cardiac Surgery Using Interpretable Machine Learning. Anesth Pain Med. 2022;12(4):e127140.
Wu M, Jiang X, Du K, Xu Y, Zhang W. Ensemble machine learning algorithm for predicting acute kidney injury in patients admitted to the neurointensive care unit following brain surgery. Scientific Reports. 2023;13(1).
Li X, Wu R, Zhao W, Shi R, Zhu Y, Wang Z, et al. Machine learning algorithm to predict mortality in critically ill patients with sepsis-associated acute kidney injury. Scientific Reports. 2023;13(1).
Mu F, Cui C, Tang M, Guo G, Zhang H, Ge J, et al. Analysis of a machine learning–based risk stratification scheme for acute kidney injury in vancomycin. Frontiers in Pharmacology. 2022;13.
Kim JY, Kim KY, Yee J, Gwak HS. Risk Scoring System for Vancomycin-Associated Acute Kidney Injury. Frontiers in Pharmacology. 2022;13.
Miyai T, Imai S, Kashiwagi H, Sato Y, Kadomura S, Yoshida K, et al. A Risk Prediction Flowchart of Vancomycin-Induced Acute Kidney Injury to Use When Starting Vancomycin Administration: A Multicenter Retrospective Study. Antibiotics. 2020;9(12):920.
Biau G. Analysis of a random forests model. J Mach Learn Res. 2012;13(null):1063–95.
Yadav D. Prediction of Heart Disease Using Feature Selection and Random Forest Ensemble Method. International Journal for Pharmaceutical Research Scholars. 2020;12:56-66.
Yu Z, Wang K, Wan Z, Xie S, Lv Z. Popular deep learning algorithms for disease prediction: a review. Cluster Computing. 2023;26(2):1231-51.
Lisboa PJ, Taktak AFG. The use of artificial neural networks in decision support in cancer: A systematic review. Neural Networks. 2006;19(4):408-15.
Ogunleye A, Wang QG. XGBoost Model for Chronic Kidney Disease Diagnosis. IEEE/ACM Transactions on Computational Biology and Bioinformatics. 2020;17(6):2131-40.
Raihan MJ, Khan MA-M, Kee S-H, Nahid A-A. Detection of the chronic kidney disease using XGBoost classifier and explaining the influence of the attributes on the model using SHAP. Scientific Reports. 2023;13(1).
Bentéjac C, Csörgő A, Martínez-Muñoz G. A comparative analysis of gradient boosting algorithms. Artificial Intelligence Review. 2021;54(3):1937-67.
Zhang D, Gong Y. The Comparison of LightGBM and XGBoost Coupling Factor Analysis and Prediagnosis of Acute Liver Failure. IEEE Access. 2020;8:220990-1003.
Rufo DD, Debelee TG, Ibenthal A, Negera WG. Diagnosis of Diabetes Mellitus Using Gradient Boosting Machine (LightGBM). Diagnostics. 2021;11(9):1714.
Sharma D, Gujral R, Jain A. Breast cancer prediction based on neural networks and extra tree classifier using feature ensemble learning. Measurement: Sensors. 2022;24:100560.
Shafique R, Mehmood A, Ullah DS, Choi GS. Cardiovascular Disease Prediction System Using Extra Trees Classifier2019.
Hameed M, Khalid M, Khaleel F, Al-Ansari N. An Extra Tree Regression Model for Discharge Coefficient Prediction: Novel, Practical Applications in the Hydraulic Sector and Future Research Directions. Mathematical Problems in Engineering. 2021;2021:1-19.
Bergstra J, Bengio Y. Random search for hyper-parameter optimization. J Mach Learn Res. 2012;13(null):281–305.
Vujovic Z. Classification Model Evaluation Metrics. International Journal of Advanced Computer Science and Applications. 2021;Volume 12:599-606.
Yu K, Xie X. Predicting Hospital Readmission: A Joint Ensemble-Learning Model. IEEE Journal of Biomedical and Health Informatics. 2020;24(2):447-56.
Mahajan P, Uddin S, Hajati F, Moni MA. Ensemble Learning for Disease Prediction: A Review. Healthcare. 2023;11(12):1808.
Lundberg SM, Lee S-I. A unified approach to interpreting model predictions. Proceedings of the 31st International Conference on Neural Information Processing Systems; Long Beach, California, USA: Curran Associates Inc.; 2017. p. 4768–77.
- Abstract Viewed: 533 times
- pdf Downloaded: 1067 times
Download Statastics
Make a Submission
SJR
COPE
