Machine learning prediction of thyroid cancer recurrence for early screening and clinical decision pathways: a
Guobin Hu1, Bin Huang2, Li Cai3
1The Fourth Hospital of Hebei Medical University, Shijiazhuang, China.
Abstract:
Recurrence prediction in differentiated thyroid carcinoma (DTC) remains clinically challenging despite generally favorable outcomes and well-established treatment strategies. Improving early identification of patients at elevated recurrence risk may enhance individualized surveillance and therapeutic decision-making. This study evaluated the performance and clinical utility of machine-learning models for recurrence prediction using routinely collected clinicopathologic features from a publicly available cohort of 383 patients with long-term follow-up.16 variables were initially analyzed following international guideline definitions. Random Forest, XGBoost, and LightGBM models were developed using stratified training-test splits, SMOTE for class-imbalance correction, fivefold cross-validation, probability calibration, and decision-curve analysis. Shapley Additive Explanations (SHAP) was applied to quantify global and local feature contributions and to derive simplified feature subsets. Models trained with 4, 6, 8, and full feature sets were compared to assess the impact of dimensionality reduction on discrimination and interpretability. Full-feature models achieved strong performance, with Random Forest obtaining the highest AUC (0.931). Notably, a compact 4-feature Random Forest model-including Risk, N stage, T stage, and Age-maintained high discriminatory ability (AUC 0.913; accuracy 0.862; recall 0.750), demonstrating that substantial simplification preserved predictive value. Performance improvements plateaued beyond 6-8 features, indicating limited incremental benefit from larger feature sets. SHAP analysis consistently identified Risk, N, T, and Age as dominant predictors. These findings highlight that streamlined, interpretable ML models using a small number of clinically accessible features can provide accurate and explainable recurrence prediction in DTC. Such models offer advantages in computational efficiency, transparency, and real-world deployability, supporting their potential integration into electronic health record systems or point-of-care decision tools. Future work should prioritize multicenter external validation and incorporation of additional pathological or molecular markers to enhance generalizability and clinical applicability.
More Related Videos
03:55Computer-Aided Three-Dimensional Visualization in the Treatment of Locally Advanced Thyroid Cancer
Published on: June 9, 2023
07:13Comparison of Predictive Performance of Three Lymph Node Staging Systems in Colorectal Signet Ring Cell Carcinoma Based on Machine Learning Model
Published on: April 18, 2025
