Predicting Distant Metastasis in Head and Neck Squamous Cell Carcinoma: A Machine Learning Analysis of 31,060
Changping Ma1, Xinyi Xu2, Yaoqi Chen1
1Ningxia Medical University, No. 1160, Sheng Li South Road, Yinchuan, 750004 Ningxia People's Republic of China.
Background:
Accurate assessment of the risk of distant metastasis of head and neck squamous cell carcinoma (HNSCC) is important for the development of personalized treatment and prognosis. The purpose of this study was to explore the risk factors for distant metastasis of HNSCC and to establish a predictive model using machine learning methods.
Materials And Methods:
We designed a retrospective cross-sectional study with a cohort from the SEER database (affiliated with the National Cancer Institute). A total of 31,060 cases of head and neck cancer were included by our inclusion criteria. We constructed four machine learning models-logistic regression, decision tree, XGBoost, and neural network-to predict the risk of distant metastasis in HNSCC patients and compared the performance of the four models. Accuracy, precision, recall, and F1-score were used to evaluate the performance of the model. The evaluation ability and clinical practicability of the model were verified by comparing the area under the curve and the receiver operating characteristic curve.
Results:
The receiver operating characteristic of the four models ranged from 0.681 to 0.847. The average accuracy of all algorithms was 77%, and XGBoost had the highest accuracy of 85.119%. Among the four models, XGBoost and logistic regression had the highest precision, both with precision greater than 79. Neural network had the highest recall and F1-score. Decision tree had the lowest accuracy and recall. Among the four models, XGBoost achieved the highest AUC (0.846) and accuracy (85.1%), outperforming neural networks (AUC: 0.837) and logistic regression (AUC: 0.838).
Conclusion:
XGBoost had the highest classification accuracy, so this machine learning method could be used to predict distant metastasis of HNSCC. The application of machine learning algorithms can stratify patients with HNSCC in clinic, which is conducive to the development of personalized treatment plans.
Clinical Relevance:
The findings of this study have significant implications clinical management of HMSCC. The model's top predictors were surgical treatment (OR: 0.33) and lymph node involvement (OR: 3.37), suggests that this machine learning algorithm could be effectively integrated into clinical practice to predict distant metastasis in HMSCC patients. This has the potential to enhance the accuracy of prognostic assessments, thereby facilitating more informed treatment planning and personalized care.
More Related Videos
07:15Machine Learning Algorithms for Early Detection of Bone Metastases in an Experimental Rat Model
Published on: August 16, 2020
04:09Predicting Treatment Response to Image-Guided Therapies Using Machine Learning: An Example for Trans-Arterial Treatment of Hepatocellular Carcinoma
Published on: October 10, 2018
