Surgical Methods and Social Factors Are Associated With Long-Term Survival in Follicular Thyroid Carcinoma: Construction and Validation of a Prognostic Model Based on Machine Learning Algorithms

Abstract

Background: 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. Methods: 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). Results: <0.001). Among different ML algorithms, the XGBoost model had the best performance, followed by Gaussian NB, RF, LR, MLP, LightGBM, AdaBoost, KNN and SVM. In predicting FTC prognosis, the predictive performance of the XGBoost model was relatively better than the AJCC model (AUROC: 0.886 vs. 0.814). Conclusion: 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.

Ähnliche Arbeiten