Construction of a random survival forest model based on a machine learning algorithm to predict early recurrence after hepatectomy for adult hepatocellular carcinoma

Abstract

Background and aims Hepatocellular carcinoma (HCC) exhibits a propensity for early recurrence following liver resection, resulting in a bleak prognosis. At present, majority of the predictive models for the early postoperative recurrence of HCC rely on the linear assumption of the Cox Proportional Hazard (CPH) model. However, the predictive efficacy of this model is constrained by the intricate nature of clinical data. The present study aims to investigate the efficacy of the random survival forest (RSF) model, which is a machine learning algorithm, in predicting the early postoperative recurrence of HCC, and compare its performance with that of the traditional CPH model. This analysis seeks to elucidate the potential advantages of the RSF model over the CPH model in addressing this clinical challenge. Methods The present retrospective cohort study was conducted at a single center. After excluding 41 patients, a total of 541 patients were included in the final model construction and subsequent analysis. The patients were randomly divided into two groups at a 7:3 ratio: training group ( n = 378) and validation group ( n = 163). The least absolute shrinkage and selection operator (LASSO) regression was used to identify the risk factors in the training group. Then, the identified factors were used to develop the RSF and CPH regression models. The predictive ability of the model was assessed using the concordance index (C-index). The accuracy of the model predictions was evaluated using the receiver operating characteristic curve (ROC) and area under the receiver operating characteristic curve (AUC). The clinical practicality of the model was measured by decision curve analysis (DCA), and the overall performance of the model was evaluated using the Brier score. The RSF model was visually represented using the Shapley additive explanations (SHAP) framework. Then, the RSF, CPH regression, and albumin-bilirubin (ALBI) grade models were compared. Results The following variables were examined by LASSO regression: alpha fetoprotein (AFP), gamma-glutamyl transpeptidase to platelet ratio (GPR), blood transfusion (BT), microvascular invasion (MVI), large vessel invasion (LVI), Edmondson-Steiner (ES) grade, liver capsule invasion (LCI), satellite nodule (SN), and Barcelona clinic liver cancer (BCLC) grade. Then, a RSF model was developed using 500 trees, and the variable importance (VIMP) ranking was MVI, LCI, SN, BT, BCLC, ESG, AFP, GPR and LVI. After these aforementioned factors were applied, the RSF and CPH regression models were developed and compared using the ALBI grade model. The C-index for the RSF model (0.896 and 0.798, respectively) outperformed that of the CPH regression model (0.803 and 0.772, respectively) and ALBI grade model (0.517 and 0.515, respectively), in both the training and validation groups. Three time points were selected to assess the predictive capabilities of these mode...