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

被引:0
|
作者
Zhang, Ji [1 ]
Chen, Qing [1 ]
Zhang, Yu [1 ]
Zhou, Jie [2 ]
机构
[1] Huazhong Univ Sci & Technol, Union Hosp, Tongji Med Coll, Dept Hepatobiliary Surg, Wuhan, Peoples R China
[2] Huazhong Univ Sci & Technol, Tongji Med Coll, Dept Biochem & Mol Biol, Wuhan, Peoples R China
基金
中国国家自然科学基金;
关键词
Hepatocellular carcinoma; Hepatectomy; Early recurrence; Machine learning; PERIOPERATIVE BLOOD-TRANSFUSION; INTRAHEPATIC RECURRENCE; CURATIVE RESECTION; CANCER;
D O I
10.1186/s12885-024-13366-4
中图分类号
R73 [肿瘤学];
学科分类号
100214 ;
摘要
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 models: 6, 12 and 18 months. For the training group, the AUC value for the RSF model at 6, 12 and 18 months was 0.971 (95% CI: 0.955-0.988), 0.919 (95% CI: 0.887-0.951) and 0.899 (95% CI: 0.867-0.932), respectively. For the validation cohort, the AUC value for the RSF model at 6, 12 and 18 months was 0.830 (95% CI: 0.728-0.932), 0.856 (95% CI: 0.787-0.924) and 0.832 (95% CI: 0.764-0.901), respectively. The AUC values were higher in the RSF model, when compared to the CPH regression model and ALBI grade model, in both groups. The DCA results revealed that the net clinical benefits associated to the RSF model were superior to those associated to the CPH regression model and ALBI grade model in both groups, suggesting a higher level of clinical utility in the RSF model. The Brier score for the RSF model at 6, 12 and 18 months was 0.062, 0.125 and 0.178, respectively, in the training group, and 0.111, 0.128 and 0.149, respectively, in the validation group. In summary, the RSF model demonstrated superior performance, when compared to the CPH regression model and ALBI grade model. Furthermore, the RSF model demonstrated superior predictive ability, accuracy, clinical practicality, and overall performance, when compared to the CPH regression model and ALBI grade model. In addition, the RSF model was able to successfully stratify patients into three distinct risk groups (low-risk, medium-risk and high-risk) in both groups (p < 0.001). Conclusions: The RSF model demonstrates efficacy in predicting early recurrence following HCC surgery, exhibiting superior performance, when compared to the CPH regression model and ALBI grade model. For patients undergoing HCC surgery, the RSF model can serve as a valuable tool for clinicians to postoperatively stratify patients into distinct risk categories, offering guidance for subsequent follow-up care.
引用
收藏
页数:18
相关论文
共 50 条
  • [1] Development of a machine learning model to predict early recurrence for hepatocellular carcinoma after curative resection
    Zeng, Jianxing
    Zeng, Jinhua
    Lin, Kongying
    Lin, Haitao
    Wu, Qionglan
    Guo, Pengfei
    Zhou, Weiping
    Liu, Jingfeng
    HEPATOBILIARY SURGERY AND NUTRITION, 2020,
  • [2] Prediction model for hepatocellular carcinoma recurrence after hepatectomy: Machine learning-based development and interpretation study
    Liu, Rongqiang
    Wu, Shinan
    Yu, Hao yuan
    Zeng, Kaining
    Liang, Zhixing
    Li, Siqi
    Hu, Yongwei
    Yang, Yang
    Ye, Linsen
    HELIYON, 2023, 9 (11)
  • [3] A novel machine learning algorithm to predict disease free survival after resection of hepatocellular carcinoma
    Schoenberg, Markus Bo
    Bucher, Julian Nikolaus
    Koch, Donnuik
    Boerner, Nikolaus
    Hesse, Sebastian
    De Toni, Enrico Narciso
    Seidensticker, Max
    Angele, Martin Kurt
    Klein, Christoph
    Bazhin, Alexandr, V
    Werner, Jens
    Guba, Markus Otto
    ANNALS OF TRANSLATIONAL MEDICINE, 2020, 8 (07)
  • [4] Model to Predict Overall Survival in Patients With Hepatocellular Carcinoma After Curative Hepatectomy
    Zhang, Li-xiang
    Luo, Pan-quan
    Chen, Lei
    Song, Dong-da
    Xu, A-man
    Xu, Peng
    Xu, Jia
    FRONTIERS IN ONCOLOGY, 2021, 10
  • [5] PREDICTION MODELS OF RECURRENCE AND SURVIVAL FOR HEPATOCELLULAR CARCINOMA AFTER HEPATECTOMY
    Park, Sang Jae
    Lee, Kwong-Woong
    Jung, Hyung-Joong
    Kim, Young Kyu
    Park, Joong-Won
    Kim, Chang-Min
    HEPATOLOGY, 2008, 48 (04) : 962A - 962A
  • [6] Preoperative inflammation-based markers predict early and late recurrence of hepatocellular carcinoma after curative hepatectomy
    Yue Liu
    Zhong-Xia Wang
    Yin Cao
    Guang Zhang
    Wei-Bo Chen
    Chun-Ping Jiang
    Hepatobiliary&PancreaticDiseasesInternational, 2016, 15 (03) : 266 - 274
  • [7] Preoperative inflammation-based markers predict early and late recurrence of hepatocellular carcinoma after curative hepatectomy
    Liu, Yue
    Wang, Zhong-Xia
    Cao, Yin
    Zhang, Guang
    Chen, Wei-Bo
    Jiang, Chun-Ping
    HEPATOBILIARY & PANCREATIC DISEASES INTERNATIONAL, 2016, 15 (03) : 266 - 274
  • [8] The risk factors of early recurrence after hepatectomy in hepatocellular carcinoma
    Kwon, Soon-Keun
    Yun, Sung-Su
    Kim, Hong-Jin
    Lee, Dong-Shik
    ANNALS OF SURGICAL TREATMENT AND RESEARCH, 2014, 86 (06) : 283 - 288
  • [9] Diffusion-weighted imaging of hepatocellular carcinoma for predicting early recurrence and survival after hepatectomy
    Ali Muhi
    Tomoaki Ichikawa
    Utaroh Motosugi
    Katsuhiro Sano
    Zareen Fatima
    Masanori Matsuda
    Hideki Fujii
    Nobuyuki Enomoto
    Tsutomu Araki
    Hepatology International, 2013, 7 : 662 - 668
  • [10] Diffusion-weighted imaging of hepatocellular carcinoma for predicting early recurrence and survival after hepatectomy
    Muhi, Ali
    Ichikawa, Tomoaki
    Motosugi, Utaroh
    Sano, Katsuhiro
    Fatima, Zareen
    Matsuda, Masanori
    Fujii, Hideki
    Enomoto, Nobuyuki
    Araki, Tsutomu
    HEPATOLOGY INTERNATIONAL, 2013, 7 (02) : 662 - 668