A simple machine learning model for the prediction of acute kidney injury following noncardiac surgery in geriatric patients: a prospective cohort study

Abstract

Abstract Background Surgery in geriatric patients often poses risk of major postoperative complications. Acute kidney injury (AKI) is a common complication following noncardiac surgery and is associated with increased mortality. Early identification of geriatric patients at high risk of AKI could facilitate preventive measures and improve patient prognosis. This study used machine learning methods to identify important features and predict AKI following noncardiac surgery in geriatric patients. Methods The data for this study were obtained from a prospective cohort. Patients aged ≥ 65 years who received noncardiac surgery from June 2019 to December 2021 were enrolled. Data were split into training set (from June 2019 to March 2021) and internal validation set (from April 2021 to December 2021) by time. The least absolute shrinkage and selection operator (LASSO) regularization algorithm and the random forest recursive feature elimination algorithm (RF-RFE) were used to screen important predictors. Models were trained through extreme gradient boosting (XGBoost), random forest, and LASSO. The SHapley Additive exPlanations (SHAP) package was used to interpret the machine learning model. Results The training set included 6753 geriatric patients. Of these, 250 (3.70%) patients developed AKI. The XGBoost model with RF-RFE selected features outperformed other models with an area under the precision-recall curve (AUPRC) of 0.505 (95% confidence interval [CI]: 0.369–0.626) and an area under the receiver operating characteristic curve (AUROC) of 0.806 (95%CI: 0.733–0.875). The model incorporated ten predictors, including operation site and hypertension. The internal validation set included 3808 geriatric patients, and 96 (2.52%) patients developed AKI. The model maintained good predictive performance with an AUPRC of 0.431 (95%CI: 0.331–0.524) and an AUROC of 0.845 (95%CI: 0.796–0.888) in the internal validation. Conclusions This study developed a simple machine learning model and a web calculator for predicting AKI following noncardiac surgery in geriatric patients. This model may be a valuable tool for guiding preventive measures and improving patient prognosis. Trial registration The protocol of this study was approved by the Committee of Ethics from West China Hospital of Sichuan University (2019–473) with a waiver of informed consent and registered at www.chictr.org.cn (ChiCTR1900025160, 15/08/2019).