A predictive model for radiation esophagitis in esophageal cancer patients based on machine learning
10.3760/cma.j.cn371439-20240806-00004
- VernacularTitle:基于机器学习构建食管癌患者放射性食管炎预测模型
- Author:
Wei GAO
1
;
Ling ZHANG
;
Tianlei WU
;
Lili HU
;
Feng RONG
Author Information
1. 安徽医科大学附属六安医院肿瘤中心放疗科，六安 237000
- Keywords:
Esophageal neoplasms;
Chemoradiotherapy;
Machine learning;
Radiation esophagitis;
SHAP analysis
- From:
Journal of International Oncology
2025;52(1):31-37
- CountryChina
- Language:Chinese
-
Abstract:
Objective:To construct a predictive model of ≥ grade 2 radiation esophagitis (RE) in patients with esophageal cancer during concurrent radiochemotherapy (CRT) based on machine learning (ML) algorithm.Methods:A retrospective analysis was conducted on the clinical data of 276 patients with esophageal cancer who had received CRT at Lu'an Hospital of Anhui Medical University from January 2018 to January 2023. The occurrence of RE was evaluated according to grading criteria of RE developed by American Radiation Therapy Oncology Group, with ≥ grade 2 RE as the outcome event. After screening variables through the least absolute shrinkage and selection operator (LASSO) regression, the dataset was re-established. The dataset was then divided into training set ( n=193) and testing set ( n=83) in a 7∶3 ratio and included in four ML models: random forest (RF) , decision tree (DT) , extreme gradient boosting (XGBoost) , and support vector machine (SVM) . In the models, data training and model optimization were conducted in the training set, and model performance was evaluated in the testing set using the receiver operator characteristic (ROC) curve. The area under the curve (AUC) , accuracy, precision, sensitivity, and F1 score were calculated to assess the model. SHAP analysis was used to explain the optimal model. Results:By the end of follow-up, 91 cases (32.97%) of esophageal cancer patients had experienced ≥ grade 2 RE during CRT. There were statistically significant differences in tumor lesion length ( Z=-5.53, P<0.001) , Karnofsky performance status (KPS) score ( χ2=5.92, P=0.015) , the Eastern Cooperative Oncology Group (ECOG) score ( χ2=4.01, P=0.045) , hypertension ( χ2=15.35, P<0.001) , diabetes ( χ2=13.06, P<0.001) , white blood cell count ( Z=-6.59, P<0.001) , neutrophil count ( Z=-6.72, P<0.001) , and radiotherapy dose ( χ2=9.81, P=0.002) between ≥ grade 2 RE occurrence group ( n=91) and no occurrence group ( n=185) . After LASSO regression screening, 7 characteristic variables were ultimately selected, which were tumor lesion length, ECOG score, KPS score, neutrophil count, hypertension, diabetes, and radiotherapy dose. ROC curve analysis showed that the XGBoost model had better predictive performance, with an AUC of 0.90, accuracy of 0.82, precision of 0.80, sensitivity of 0.73, and F1 score of 0.76. The AUC, accuracy, precision, sensitivity, and F1 score of RF model were 0.89, 0.78, 0.76, 0.48, and 0.59, respectively. The AUC, accuracy, precision, sensitivity, and F1 score of DT model were 0.72, 0.72, 0.44, 0.60, and 0.52, respectively. The AUC of SVM model was 0.74, with an accuracy of 0.82, precision of 0.52, sensitivity of 0.88, and F1 score of 0.65. The XGBoost model was explained using SHAP analysis, which indicated that the tumor lesion length, neutrophil count, hypertension, diabetes, and radiotherapy dose had a strong predictive ability for the occurrence of ≥ grade 2 RE during CRT in esophageal cancer patients. Conclusions:The model established based on the XGBoost method has good predictive performance for the occurrence of ≥ grade 2 RE in esophageal cancer patients during CRT. Meanwhile, combined with SHAP analysis, it can provide an intuitive understanding of the impact of important features in the model on the outcome.
