Predicting Differentiated Thyroid Cancer Recurrence Using Machine Learning
摘要
A large percentage of thyroid cancers, 90%, are classified as differentiated thyroid cancer (DTC), and this comprises differentiated forms. Identifying which patients are likely to experience return of disease can increase the quality of life by avoiding side effects from unnecessary treatment for recurrence in differentiated thyroid cancer. Conventional methods used for predicting the re-occurrence of DTC heavily rely on clinical features and imaging, which often lack specificity, and struggle with dynamic risk assessment, rare aggressive variants, and personalized patient management, leading to potential overtreatment or undertreatment. While machine learning models could derive predictions using a larger set of features therefore more personal and accurate predictions, this work investigates various ML algorithms such as Neural Networks (NN), Support Vector Machines (SVM), Tree-based methods, K-Nearest Neighbor (KNN) to improve the prediction for recurrence in Differentiated Thyroid Cancer. The dataset used in this study consists of 383 instances and is characterized by 16 unique features. The accuracy of each model’s recurrence classification has been assessed. With an accuracy of 95.6%, the ensemble bagged trees model produced the most accurate prediction out of all the models. These findings suggest that machine learning approaches are more effective at forecasting DTC recurrences and may prove to be a valuable healthcare tool by providing a more accurate and tailored approach to DTC recurrence prediction.