Development and validation of an XGBoost model to predict 5-year survival in elderly patients with intrahepatic cholangiocarcinoma after surgery: a SEER-based study

Abstract

Nomograms have been established to predict survival in postoperative or elderly intrahepatic cholangiocarcinoma (ICC) patients. There are no models to predict postoperative survival in elderly ICC patients. Extreme gradient boosting (XGBoost) can adjust the errors generated by existing models. This retrospective cohort study aimed to develop and validate an XGBoost model to predict postoperative 5-year survival in elderly ICC patients. The Surveillance, Epidemiology, and End Results (SEER) program provided data on elderly ICC patients aged 60 years or older and undergoing surgery. The median follow-up time was 20 months. Totally 1,055 patients were classified as training (n=738) and testing (n=317) sets at a ratio of 7:3. The outcome was postoperative 5-year survival. Demographic, tumor-related and treatment-related variables were collected. Variables were screened using the XGBoost model. The predictive performance of the model was assessed by the area under the receiver operating characteristic (ROC) curve (AUC), sensitivity, specificity, positive predictive value (PPV), negative predictive value (NPV), and Kaplan-Meier curve. Cox regression analysis was conducted to estimate the risk of death in the predicted populations. The predictive abilities of the XGBoost model and the American Joint Commission on Cancer (AJCC) system (7th edition) were compared. The XGBoost model achieved an AUC of 0.811, a sensitivity of 0.573, a specificity of 0.890, and a PPV of 0.849 in the training set. In the testing set, the model had an AUC of 0.713, a sensitivity of 0.478, a specificity of 0.814, and a PPV of 0.726. The 5-year mortality risk of patients predicted to die was 2.91 times that of patients predicted to survive [hazard ratio (HR) =2.91, 95% confidence interval (CI): 2.42-3.50]. The XGBoost model showed a better predictive performance than the AJCC staging system both in the training and testing sets. AJCC stage, multiple (satellite) tumors/nodules, tumor-node-metastasis (TNM) stage, more than one lobe invaded, direct invasion of adjacent organs, tumor size, and radiotherapy were relatively important features in survival prediction. The XGBoost model exhibited some predictive capacity, which may be applied to predict postoperative 5-year survival for elderly ICC patients.