IMR Press / RCM / Volume 24 / Issue 8 / DOI: 10.31083/j.rcm2408229
Open Access Original Research
Development, External Validation, and Visualization of Machine Learning Models for Predicting Occurrence of Acute Kidney Injury after Cardiac Surgery
Show Less
1 Medical School of Chinese PLA, 100853 Beijing, China
2 Department of Vascular and Endovascular Surgery, The First Medical Center of Chinese PLA General Hospital, 100853 Beijing, China
3 Department of Cardiovascular Surgery, the First Medical Centre of Chinese PLA General Hospital, 100853 Beijing, China
4 Department of Cardiovascular Medicine, The First Hospital of Hebei Medical University, 050000 Shijiazhuang, Hebei, China
5 Department of Cardiovascular Surgery, the Sixth Medical Centre of Chinese PLA General Hospital, 100048 Beijing, China
*Correspondence: QinHD301hospital@163.com (Handai Qin); LaiSW301hospital@163.com (Shengwei Lai); FanYL301hospital@163.com (Yunlong Fan)
These authors contributed equally.
Rev. Cardiovasc. Med. 2023, 24(8), 229; https://doi.org/10.31083/j.rcm2408229
Submitted: 19 October 2022 | Revised: 6 February 2023 | Accepted: 17 February 2023 | Published: 9 August 2023
Copyright: © 2023 The Author(s). Published by IMR Press.
This is an open access article under the CC BY 4.0 license.
Abstract

Background: Cardiac surgery-associated acute kidney injury (CSA-AKI) is a major complication that results in short- and long-term mortality among patients. Here, we adopted machine learning algorithms to build prediction models with the overarching goal of identifying patients who are at a high risk of such unfavorable kidney outcomes. Methods: A total of 1686 patients (development cohort) and 422 patients (validation cohort), with 126 pre- and intra-operative variables, were recruited from the First Medical Centre and the Sixth Medical Centre of Chinese PLA General Hospital in Beijing, China, respectively. Analyses were performed using six machine learning techniques, namely K-nearest neighbor, logistic regression, decision tree, random forest (RF), support vector machine, and neural network, and the APPROACH score, a previously established risk score for CSA-AKI. For model tuning, optimal hyperparameter was achieved by using GridSearch with 5-fold cross-validation from the scikit-learn library. Model performance was externally assessed via the receiver operating characteristic (ROC) and decision curve analysis (DCA). Explainable machine learning was performed using the Python SHapley Additive exPlanation (SHAP) package and Seaborn library, which allow the calculation of marginal contributory SHAP value. Results: 637 patients (30.2%) developed CSA-AKI within seven days after surgery. In the external validation, the RF classifier exhibited the best performance among the six machine learning techniques, as shown by the ROC curve and DCA, while the traditional APPROACH risk score showed a relatively poor performance. Further analysis found no specific causative factor contributing to the development of CSA-AKI; rather, the development of CSA-AKI appeared to be a complex process resulting from a complex interplay of multiple risk factors. The SHAP summary plot illustrated the positive or negative contribution of RF-top 20 variables and extrapolated risk of developing CSA-AKI at individual levels. The Seaborn library showed the effect of each single feature on the model output of the RF prediction. Conclusions: Efficient machine learning approaches were successfully established to predict patients with a high probability of developing acute kidney injury after cardiac surgery. These findings are expected to help clinicians to optimize treatment strategies and minimize postoperative complications. Clinical Trial Registration: The study protocol was registered at the ClinicalTrials Registration System (https://www.clinicaltrials.gov/, #NCT04966598) on July 26, 2021.

Keywords
acute kidney injury
cardiac surgery
machine learning
prediction model
precision medicine
Figures
Fig. 1.
Share
Back to top