Predicting Thyroid Cancer Recurrence After Radioactive Iodine Therapy Using Random Forest and Neural Network Models
Abstract
Thyroid cancer recurrence following Radioactive Iodine (RAI) therapy remains a clinical concern, necessitating accurate and timely risk prediction to guide post-treatment management. This study aims to evaluate the effectiveness of machine learning models—Random Forest and Neural Networks—in predicting recurrence using a structured clinical dataset consisting of 383 patient records and 13 diagnostic and pathological attributes. All categorical features were encoded ordinally, and the dataset was partitioned into training and testing sets with appropriate normalization for neural network processing. Both models were evaluated using standard metrics including accuracy, precision, recall, and F1-score. The Random Forest model achieved an accuracy of 97.39%, outperforming the Neural Network which recorded 93.04%. Moreover, Random Forest showed better recall in detecting recurrence cases, making it a more suitable model for clinical application. These results demonstrate that machine learning, particularly ensemble-based methods, can offer a practical and interpretable solution for recurrence prediction, supporting data-driven decision-making in thyroid cancer follow-up care.
References
K. Bharath, “Predicting Recurrence in Differentiated Thyroid Cancer: A Machine Learning Approach,” 2024 International Conference on Advances in Data Engineering and Intelligent Computing Systems, ADICS 2024. 2024, doi: 10.1109/ADICS58448.2024.10533649.
M. A. S. Ahmad, “An Explainable AI Model for Predicting the Recurrence of Differentiated Thyroid Cancer,” 2024 2nd Jordanian International Biomedical Engineering Conference, JIBEC 2024. pp. 84–89, 2024, doi: 10.1109/JIBEC63210.2024.10932125.
M. G. Alfarizi, “Optimized Random Forest Model for Remaining Useful Life Prediction of Experimental Bearings,” IEEE Trans. Ind. Informatics, vol. 19, no. 6, pp. 7771–7779, 2023, doi: 10.1109/TII.2022.3206339.
N. Aida, “Comparison of Extreme Machine Learning and Hidden Markov Model Algorithm in Predicting The Recurrence Of Differentiated Thyroid Cancer Using SMOTE,” J. Electron. Electromed. Eng. Med. Informatics, vol. 6, no. 4, pp. 429–444, 2024, doi: 10.35882/jeeemi.v6i4.467.
H. Wang et al., “Development and validation of prediction models for papillary thyroid cancer structural recurrence using machine learning approaches,” BMC Cancer, vol. 24, no. 1, p. 427, Apr. 2024, doi: 10.1186/s12885-024-12146-4.
D. Iqbal and A. Shahzad, “Prediction of thyroid cancer recurrence with machine learning models,” Pakistan J. Nucl. Med., pp. 49–55, 2024, doi: 10.24911/PJNMed.175-1721068107.
O. Bülbül and D. Nak, “Predicting excellent response to radioiodine in differentiated thyroid cancer using machine learning,” Acta Otorhinolaryngol. Ital., vol. 44, no. 4, pp. 261–268, Aug. 2024, doi: 10.14639/0392-100X-N3029.
B. K and A. Sai Sabitha, “Predicting Recurrence in Differentiated Thyroid Cancer: A Machine Learning Approach,” in 2024 International Conference on Advances in Data Engineering and Intelligent Computing Systems (ADICS), Apr. 2024, pp. 1–6, doi: 10.1109/ADICS58448.2024.10533649.
Y. Camgözlü, “Leaf Image Classification Based on Pre-trained Convolutional Neural Network Models,” Nat. Eng. Sci., vol. 8, no. 3, pp. 214–232, 2023, doi: 10.28978/nesciences.1405175.
L. Lei, “An evolutionary deep learning model based on EWKM, random forest algorithm, SSA and BiLSTM for building energy consumption prediction,” Energy, vol. 288, 2024, doi: 10.1016/j.energy.2023.129795.
S. M. Simon, “Interpreting random forest analysis of ecological models to move from prediction to explanation,” Sci. Rep., vol. 13, no. 1, 2023, doi: 10.1038/s41598-023-30313-8.
S. Sharma, “A Deep Learning based model for the Detection of Pneumonia from Chest X-Ray Images using VGG-16 and Neural Networks,” Procedia Computer Science, vol. 218. pp. 357–366, 2022, doi: 10.1016/j.procs.2023.01.018.
H. Bhatt, “A Convolutional Neural Network ensemble model for Pneumonia Detection using chest X-ray images,” Healthc. Anal., vol. 3, 2023, doi: 10.1016/j.health.2023.100176.
U. K. Lilhore, “Hybrid model for precise hepatitis-C classification using improved random forest and SVM method,” Sci. Rep., vol. 13, no. 1, 2023, doi: 10.1038/s41598-023-36605-3.
L. Hua, “An evolutionary deep learning soft sensor model based on random forest feature selection technique for penicillin fermentation process,” ISA Trans., vol. 136, pp. 139–151, 2023, doi: 10.1016/j.isatra.2022.10.044.
S. Das, “A hybrid approach for Bangla sign language recognition using deep transfer learning model with random forest classifier,” Expert Syst. Appl., vol. 213, 2023, doi: 10.1016/j.eswa.2022.118914.
A. Elbeltagi, “Prediction of meteorological drought and standardized precipitation index based on the random forest (RF), random tree (RT), and Gaussian process regression (GPR) models,” Environ. Sci. Pollut. Res., vol. 30, no. 15, pp. 43183–43202, 2023, doi: 10.1007/s11356-023-25221-3.
F. Huang, “Slope stability prediction based on a long short-term memory neural network: comparisons with convolutional neural networks, support vector machines and random forest models,” Int. J. Coal Sci. Technol., vol. 10, no. 1, 2023, doi: 10.1007/s40789-023-00579-4.
G. Ali, “A Hybrid Convolutional Neural Network Model for Automatic Diabetic Retinopathy Classification From Fundus Images,” IEEE J. Transl. Eng. Heal. Med., vol. 11, pp. 341–350, 2023, doi: 10.1109/JTEHM.2023.3282104.
M. Gupta, “A real-time adaptive model for bearing fault classification and remaining useful life estimation using deep neural network,” Knowledge-Based Syst., vol. 259, 2023, doi: 10.1016/j.knosys.2022.110070.
S. Sharma, “A Deep Learning Model for Early Prediction of Pneumonia Using VGG19 and Neural Networks,” Lecture Notes in Networks and Systems, vol. 588. pp. 597–612, 2023, doi: 10.1007/978-981-19-7982-8_50.
X. Zhao, “Model-Assisted Multi-source Fusion Hypergraph Convolutional Neural Networks for intelligent few-shot fault diagnosis to Electro-Hydrostatic Actuator,” Inf. Fusion, vol. 104, 2024, doi: 10.1016/j.inffus.2023.102186.
S. Suneel, “Quantum mesh neural network model in precise image diagnosing,” Opt. Quantum Electron., vol. 56, no. 4, 2024, doi: 10.1007/s11082-023-06245-y.
A. Zermane, “Predicting fatal fall from heights accidents using random forest classification machine learning model,” Saf. Sci., vol. 159, 2023, doi: 10.1016/j.ssci.2022.106023.
D. M. Ge, “Estimation of rapid chloride permeability of SCC using hyperparameters optimized random forest models,” J. Sustain. Cem. Mater., vol. 12, no. 5, pp. 542–560, 2023, doi: 10.1080/21650373.2022.2093291.
P. Celard, “A survey on deep learning applied to medical images: from simple artificial neural networks to generative models,” Neural Computing and Applications, vol. 35, no. 3. pp. 2291–2323, 2023, doi: 10.1007/s00521-022-07953-4.
