This study aimed to establish and verify an effective machine learning (ML) model to predict the prognosis of follicular thyroid cancer (FTC), and compare it with the eighth edition of the American Joint Committee on Cancer (AJCC) model.
Kaplan-Meier method and Cox regression model were used to analyze the risk factors of cancer-specific survival (CSS). Propensity-score matching (PSM) was used to adjust the confounding factors of different surgeries. Nine different ML algorithms,including eXtreme Gradient Boosting (XGBoost), Light Gradient Boosting Machine (LightGBM), Random Forests (RF), Logistic Regression (LR), Adaptive Boosting (AdaBoost), Gaussian Naive Bayes (GaussianNB), K-Nearest Neighbor (KNN), Support Vector Machine (SVM) and Multi-Layer Perceptron (MLP),were used to build prognostic models of FTC.10-fold cross-validation and SHapley Additive exPlanations were used to train and visualize the optimal ML model.The AJCC model was built by multivariate Cox regression and visualized through nomogram. The performance of the XGBoost model and AJCC model was mainly assessed using the area under the receiver operating characteristic (AUROC).
Multivariate Cox regression showed that age, surgical methods, marital status, T classification, N classification and M classification were independent risk factors of CSS. Among different surgeries, the prognosis of one-sided thyroid lobectomy plus isthmectomy (LO plus IO) was the best, followed by total thyroidectomy (hazard ratios: One-sided thyroid LO plus IO, 0.086[95% confidence interval (CI),0.025-0.290],
For high-risk groups, effective surgical methods and well marital status can improve the prognosis of FTC. Compared with the traditional AJCC model, the XGBoost model has relatively better prediction accuracy and clinical usage.