Machine Learning Model-Based Prediction of In-Hospital Acute Kidney Injury Risk in Acute Aortic Dissection Patients
Zhili Wei , Shidong Liu , Yang Chen , Hongxu Liu , Guangzu Liu , Yuan Hu , Bing Song
Reviews in Cardiovascular Medicine ›› 2025, Vol. 26 ›› Issue (2) : 25768
This study aimed to identify the risk factors for in-hospital acute kidney injury (AKI) in patients with acute aortic dissection (AAD) and to establish a machine learning model for predicting in-hospital AKI.
We extracted data on patients with AAD from the Medical Information Mart for Intensive Care (MIMIC)-IV database and developed seven machine learning models: support vector machine (SVM), gradient boosting machine (GBM), neural network (NNET), eXtreme gradient boosting (XGBoost), K-nearest neighbors (KNN), light gradient boosting machine (LightGBM), and categorical boosting (CatBoost). Model performance was assessed using the area under the receiver operating characteristic curve (AUC), and the optimal model was interpreted using Shapley Additive explanations (SHAP) visualization analysis.
A total of 325 patients with AAD were identified from the MIMIC-IV database, of which 84 patients (25.85%) developed in-hospital AKI. This study collected 42 features, with nine selected for model building. A total of 70% of the patients were randomly allocated to the training set, while the remaining 30% were allocated to the test set. Machine learning models were built on the training set and validated using the test set. In addition, we collected AAD patient data from the MIMIC-III database for external validation. Among the seven machine learning models, the CatBoost model performed the best, with an AUC of 0.876 in the training set and 0.723 in the test set. CatBoost also performed strongly during the validation, achieving an AUC of 0.712. SHAP visualization analysis identified the most important risk factors for in-hospital AKI in AAD patients as maximum blood urea nitrogen (BUN), body mass index (BMI), urine output, maximum glucose (GLU), minimum BUN, minimum creatinine, maximum creatinine, weight and acute physiology score III (APSIII).
The CatBoost model, constructed using risk factors including maximum and minimum BUN levels, BMI, urine output, and maximum GLU, effectively predicts the risk of in-hospital AKI in AAD patients and shows compelling results in further validations.
acute aortic dissection / acute kidney injury / machine learning / prediction model
| [1] |
Berretta P, Patel HJ, Gleason TG, Sundt TM, Myrmel T, Desai N, et al. IRAD experience on surgical type A acute dissection patients: results and predictors of mortality. Annals of Cardiothoracic Surgery. 2016; 5: 346–351. https://doi.org/10.21037/acs.2016.05.10 |
| [2] |
Nienaber CA, Clough RE. Management of acute aortic dissection. Lancet. 2015; 385: 800–811. https://doi.org/10.1016/S0140-6736(14)61005-9 |
| [3] |
Blakeslee-Carter J, Menon AJ, Novak Z, Spangler EL, Beck AW, McFarland GE. Association of Mental Health Disorders and Aortic Dissection. Annals of Vascular Surgery. 2021; 77: 217–225. https://doi.org/10.1016/j.avsg.2021.05.054 |
| [4] |
Sen I, Erben YM, Franco-Mesa C, DeMartino RR. Epidemiology of aortic dissection. Seminars in Vascular Surgery. 2021; 34: 10–17. https://doi.org/10.1053/j.semvascsurg.2021.02.003 |
| [5] |
Hagan PG, Nienaber CA, Isselbacher EM, Bruckman D, Karavite DJ, Russman PL, et al. The International Registry of Acute Aortic Dissection (IRAD): new insights into an old disease. JAMA. 2000; 283: 897–903. https://doi.org/10.1001/jama.283.7.897 |
| [6] |
Woo KMC, Schneider JI. High-risk chief complaints I: chest pain–the big three. Emergency Medicine Clinics of North America. 2009; 27: 685–712, x. https://doi.org/10.1016/j.emc.2009.07.007 |
| [7] |
Van den Eynde J, Cloet N, Van Lerberghe R, Sá MPBO, Vlasselaers D, Toelen J, et al. Strategies to Prevent Acute Kidney Injury after Pediatric Cardiac Surgery: A Network Meta-Analysis. Clinical Journal of the American Society of Nephrology. 2021; 16: 1480–1490. https://doi.org/10.2215/CJN.05800421 |
| [8] |
Li L, Zhou J, Hao X, Zhang W, Yu D, Xie Y, et al. The Incidence, Risk Factors and In-Hospital Mortality of Acute Kidney Injury in Patients After Surgery for Acute Type A Aortic Dissection: A Single-Center Retrospective Analysis of 335 Patients. Frontiers in Medicine. 2020; 7: 557044. https://doi.org/10.3389/fmed.2020.557044 |
| [9] |
Xu Y, Dai ST, Liu LZ, Kong DM, Guo SK, Gong KM. Risk factors for postoperative acute kidney injury in overweight patients with acute type A aortic dissection. Journal of Cardiothoracic Surgery. 2023; 18: 115. https://doi.org/10.1186/s13019-023-02218-0 |
| [10] |
Katiyar A, Mohanty A, Hua J, Chao S, Lopes R, Datta A, et al. A Bayesian approach to determine the composition of heterogeneous cancer tissue. BMC Bioinformatics. 2018; 19: 90. https://doi.org/10.1186/s12859-018-2062-0 |
| [11] |
Cypko MA, Stoehr M, Oeltze-Jafra S, Dietz A, Lemke HU. A Guide for Constructing Bayesian Network Graphs of Cancer Treatment Decisions. Studies in Health Technology and Informatics. 2017; 245: 1355. |
| [12] |
Dong J, Feng T, Thapa-Chhetry B, Cho BG, Shum T, Inwald DP, et al. Machine learning model for early prediction of acute kidney injury (AKI) in pediatric critical care. Critical Care. 2021; 25: 288. https://doi.org/10.1186/s13054-021-03724-0 |
| [13] |
Heo J, Yoon JG, Park H, Kim YD, Nam HS, Heo JH. Machine Learning-Based Model for Prediction of Outcomes in Acute Stroke. Stroke. 2019; 50: 1263–1265. https://doi.org/10.1161/STROKEAHA.118.024293 |
| [14] |
Bonkhoff AK, Grefkes C. Precision medicine in stroke: towards personalized outcome predictions using artificial intelligence. Brain. 2022; 145: 457–475. https://doi.org/10.1093/brain/awab439 |
| [15] |
Zhang L, Wang Z, Xu F, Han D, Li S, Yin H, et al. Effects of Stress Hyperglycemia on Short-Term Prognosis of Patients Without Diabetes Mellitus in Coronary Care Unit. Frontiers in Cardiovascular Medicine. 2021; 8: 683932. https://doi.org/10.3389/fcvm.2021.683932 |
| [16] |
Wu WT, Li YJ, Feng AZ, Li L, Huang T, Xu AD, et al. Data mining in clinical big data: the frequently used databases, steps, and methodological models. Military Medical Research. 2021; 8: 44. https://doi.org/10.1186/s40779-021-00338-z |
| [17] |
Xing L, Zhou Y, Han Y, Chen C, Dong Z, Zheng X, et al. Simple Death Risk Models to Predict In-hospital Outcomes in Acute Aortic Dissection in Emergency Department. Frontiers in Medicine. 2022; 9: 890567. https://doi.org/10.3389/fmed.2022.890567 |
| [18] |
Yuan Y, Xia Z, Wang L, Sun Q, Wang W, Chai C, et al. Risk factors for in-hospital death in 2,179 patients with acute aortic dissection. Frontiers in Cardiovascular Medicine. 2023; 10: 1159475. https://doi.org/10.3389/fcvm.2023.1159475 |
| [19] |
Cai D, Xiao T, Zou A, Mao L, Chi B, Wang Y, et al. Predicting acute kidney injury risk in acute myocardial infarction patients: An artificial intelligence model using medical information mart for intensive care databases. Frontiers in Cardiovascular Medicine. 2022; 9: 964894. https://doi.org/10.3389/fcvm.2022.964894 |
| [20] |
Kellum JA, Lameire N, KDIGO AKI Guideline Work Group. Diagnosis, evaluation, and management of acute kidney injury: a KDIGO summary (Part 1). Critical Care. 2013; 17: 204. https://doi.org/10.1186/cc11454 |
| [21] |
Goyal A, Sulaiman SA, Pancholi V, Fatima L, Yakkali S, Doshi A, et al. Exploring Acute Kidney Injury Following Aortic Dissection: A Comprehensive Review of Machine Learning Models for Predicting Risk, Management Strategies, Complications, and Racial and Gender Disparities. Cardiology in Review. 2024. (online ahead of print). https://doi.org/10.1097/CRD.0000000000000717 |
| [22] |
Chen X, Fang M, Yang J, Wang S, Wang X, Li L, et al. Incidence and outcomes of acute kidney disease in patients after type A aortic dissection surgery. Asian Journal of Surgery. 2023; 46: 1207–1214. https://doi.org/10.1016/j.asjsur.2022.08.002 |
| [23] |
Meng W, Li R, E L, Zha N. Postoperative acute kidney injury and early and long-term mortality in acute aortic dissection patients: A meta-analysis. Medicine. 2021; 100: e23426. https://doi.org/10.1097/MD.0000000000023426 |
| [24] |
Dai A, Zhou Z, Jiang F, Guo Y, Asante DO, Feng Y, et al. Incorporating intraoperative blood pressure time-series variables to assist in prediction of acute kidney injury after type a acute aortic dissection repair: an interpretable machine learning model. Annals of Medicine. 2023; 55: 2266458. https://doi.org/10.1080/07853890.2023.2266458 |
| [25] |
Luo CC, Zhong YL, Qiao ZY, Li CN, Liu YM, Zheng J, et al. Development and validation of a nomogram for postoperative severe acute kidney injury in acute type A aortic dissection. Journal of Geriatric Cardiology. 2022; 19: 734–742. https://doi.org/10.11909/j.issn.1671-5411.2022.10.003 |
| [26] |
Zhang C, Chen S, Yang J, Pan G. Postoperative nomogram and risk calculator of acute renal failure for Stanford type A aortic dissection surgery. General Thoracic and Cardiovascular Surgery. 2023; 71: 639–647. https://doi.org/10.1007/s11748-023-01935-6 |
| [27] |
Chenitz KB, Lane-Fall MB. Decreased urine output and acute kidney injury in the postanesthesia care unit. Anesthesiology Clinics. 2012; 30: 513–526. https://doi.org/10.1016/j.anclin.2012.07.004 |
| [28] |
Oh HJ, Shin DH, Lee MJ, Ko KI, Kim CH, Koo HM, et al. Urine output is associated with prognosis in patients with acute kidney injury requiring continuous renal replacement therapy. Journal of Critical Care. 2013; 28: 379–388. https://doi.org/10.1016/j.jcrc.2012.11.019 |
| [29] |
Cheruku SR, Raphael J, Neyra JA, Fox AA. Acute Kidney Injury after Cardiac Surgery: Prediction, Prevention, and Management. Anesthesiology. 2023; 139: 880–898. https://doi.org/10.1097/ALN.0000000000004734 |
| [30] |
Vandijck DM, Oeyen S, Decruyenaere JM, Annemans L, Hoste EA. Acute kidney injury, length of stay, and costs in patients hospitalized in the intensive care unit. Acta Clinica Belgica. 2007; 62: 341–345. https://doi.org/10.1179/acb.2007.077 |
| [31] |
Jiang R, Huang Y, Zeng M, Xing C, Mao H, Wu B. A marked elevation in serum creatinine/cystatin C ratio may indicate pseudo-acute kidney injury due to urinary ascites: a case report and literature review. BMC Nephrology. 2023; 24: 243. https://doi.org/10.1186/s12882-023-03289-w |
| [32] |
Lu P, Lin D, Chen N, Wang L, Zhang X, Chen H, et al. CNN-assisted SERS enables ultra-sensitive and simultaneous detection of Scr and BUN for rapid kidney function assessment. Analytical Methods: Advancing Methods and Applications. 2023; 15: 322–332. https://doi.org/10.1039/d2ay01573k |
| [33] |
Jerant A, Franks P. Body mass index, diabetes, hypertension, and short-term mortality: a population-based observational study, 2000-2006. Journal of the American Board of Family Medicine. 2012; 25: 422–431. https://doi.org/10.3122/jabfm.2012.04.110289 |
| [34] |
Gharaibeh KA, Hamadah AM, Sierra RJ, Leung N, Kremers WK, El-Zoghby ZM. The Rate of Acute Kidney Injury After Total Hip Arthroplasty Is Low but Increases Significantly in Patients with Specific Comorbidities. The Journal of Bone and Joint Surgery. American Volume. 2017; 99: 1819–1826. https://doi.org/10.2106/JBJS.16.01027 |
| [35] |
Fan S, Ma J. The value of five scoring systems in predicting the prognosis of patients with sepsis-associated acute respiratory failure. Scientific Reports. 2024; 14: 4760. https://doi.org/10.1038/s41598-024-55257-5 |
| [36] |
Pan X, Xu J, Wu H, Wang J, Kong W. Prognostic value of the systemic immune-inflammation index in patients with acute respiratory distress syndrome: A retrospective study. Heliyon. 2024; 10: e26569. https://doi.org/10.1016/j.heliyon.2024.e26569 |
Natural Science Foundation of Gansu Province(21JR7RA379)
/
| 〈 |
|
〉 |