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.

Predicting acute kidney injury onset using a random forest algorithm using electronic medical records of COVID-19 patients: the CRACoV-AKI model

Machine Learning for Predicting Chronic Renal Disease Progression in COVID-19 Patients with Acute Renal Injury: A Feasibility Study

Predictive value of machine learning for the risk of acute kidney injury (AKI) in hospital intensive care units (ICU) patients: a systematic review and meta-analysis

Development of a Federated Learning Approach to Predict Acute Kidney Injury in Adult Hospitalized Patients with COVID-19 in New York City.

The Development of a Machine Learning Inpatient Acute Kidney Injury Prediction Model*

Prediction of acute kidney injury after cardiac surgery: model development using a Chinese electronic health record dataset

A novel multivariable time series prediction model for acute kidney injury in general hospitalization

Predicting In-Hospital Mortality of Patients with Acute Kidney Injury in the ICU Using Random Forest Model.

Predictive modeling of COVID-19 mortality risk in chronic kidney disease patients using multiple machine learning algorithms

Predicting in-hospital outcomes of patients with acute kidney injury

Machine learning for prediction of acute kidney injury in patients diagnosed with sepsis in critical care

Acute kidney injury prediction for non-critical care patients: a retrospective external and internal validation study

Machine learning for the prediction of acute kidney injury in patients with sepsis

Artificial intelligence algorithms permits rapid acute kidney injury risk classification of patients with acute myocardial infarction

Predicting acute kidney injury risk in acute myocardial infarction patients: An artificial intelligence model using medical information mart for intensive care databases

Development, External Validation, and Visualization of Machine Learning Models for Predicting Occurrence of Acute Kidney Injury after Cardiac Surgery

Predicting acute kidney injury at hospital re-entry using high-dimensional electronic health record data

A validation study of UCSD-Mayo risk score in predicting hospital-acquired acute kidney injury in COVID-19 patients

Forecasting acute kidney injury and resource utilization in ICU patients using longitudinal, multimodal models

Health Care Analytics With Time-Invariant and Time-Variant Feature Importance to Predict Hospital-Acquired Acute Kidney Injury: Observational Longitudinal Study

Construction and evaluation of a risk prediction model for pulmonary infection‐associated acute kidney injury in intensive care units