An explainable machine learning model to predict early and late acute kidney injury after major hepatectomy

Abstract

Background: Risk assessment models for acute kidney injury (AKI) after major hepatectomy that differentiate between early and late AKI are lacking. This retrospective study aimed to create a model predicting AKI through machine learning and identify features that contribute to the development of early and late AKI.

Methods: Patients that underwent major hepatectomy were categorized into the No-AKI, Early-AKI (within 48 h) or Late-AKI group (between 48 h and 7 days). Modeling was done with 20 perioperative features and the performance of prediction models were measured by the area under the receiver operating characteristic curve (AUROCC). Shapley Additive Explanation (SHAP) values were utilized to explain the outcome of the prediction model.

Results: Of the 1383 patients included in this study, 1229, 110 and 44 patients were categorized into the No-AKI, Early-AKI and Late-AKI group, respectively. The CatBoost classifier exhibited the greatest AUROCC of 0.758 (95% CI: 0.671-0.847) and was found to differentiate well between Early and Late-AKI. We identified different perioperative features for predicting each outcome and found 1-year mortality to be greater for Early-AKI.

Conclusions: Our results suggest that risk factors are different for Early and Late-AKI after major hepatectomy, and 1-year mortality is greater for Early-AKI.