International Journal of Medical Informatics, volume 136, pages 104068
Comparison of supervised machine learning classification techniques in prediction of locoregional recurrences in early oral tongue cancer
Rasheed Omobolaji Alabi
1
,
Mohammed Elmusrati
1
,
Íris Sawazaki Calone
2
,
Luiz Paulo Kowalski
3
,
Caj Haglund
4
,
Ricardo Della Coletta
5, 6, 7
,
Antti A Makitie
8, 9, 10
,
Ilmo Leivo
9
3
Department of Head and Neck Surgery and Otorhinolaryngology, A.C. Camargo Cancer Center, São Paulo, SP, Brazil.
|
5
Department of Oral Diagnosis
7
Piracicaba São Paulo Brazil
|
Publication type: Journal Article
Publication date: 2020-04-01
scimago Q1
SJR: 1.110
CiteScore: 8.9
Impact factor: 3.7
ISSN: 13865056, 18728243
Health Informatics
Abstract
The proper estimate of the risk of recurrences in early-stage oral tongue squamous cell carcinoma (OTSCC) is mandatory for individual treatment-decision making. However, this remains a challenge even for experienced multidisciplinary centers.We compared the performance of four machine learning (ML) algorithms for predicting the risk of locoregional recurrences in patients with OTSCC. These algorithms were Support Vector Machine (SVM), Naive Bayes (NB), Boosted Decision Tree (BDT), and Decision Forest (DF).The study cohort comprised 311 cases from the five University Hospitals in Finland and A.C. Camargo Cancer Center, São Paulo, Brazil. For comparison of the algorithms, we used the harmonic mean of precision and recall called F1 score, specificity, and accuracy values. These algorithms and their corresponding permutation feature importance (PFI) with the input parameters were externally tested on 59 new cases. Furthermore, we compared the performance of the algorithm that showed the highest prediction accuracy with the prognostic significance of depth of invasion (DOI).The results showed that the average specificity of all the algorithms was 71% . The SVM showed an accuracy of 68% and F1 score of 0.63, NB an accuracy of 70% and F1 score of 0.64, BDT an accuracy of 81% and F1 score of 0.78, and DF an accuracy of 78% and F1 score of 0.70. Additionally, these algorithms outperformed the DOI-based approach, which gave an accuracy of 63%. With PFI-analysis, there was no significant difference in the overall accuracies of three of the algorithms; PFI-BDT accuracy increased to 83.1%, PFI-DF increased to 80%, PFI-SVM decreased to 64.4%, while PFI-NB accuracy increased significantly to 81.4%.Our findings show that the best classification accuracy was achieved with the boosted decision tree algorithm. Additionally, these algorithms outperformed the DOI-based approach. Furthermore, with few parameters identified in the PFI analysis, ML technique still showed the ability to predict locoregional recurrence. The application of boosted decision tree machine learning algorithm can stratify OTSCC patients and thus aid in their individual treatment planning.
Nothing found, try to update filter.
Prediction of lung cancer patient survival via supervised machine learning classification techniques
Lynch C.M., Abdollahi B., Fuqua J.D., de Carlo A.R., Bartholomai J.A., Balgemann R.N., van Berkel V.H., Frieboes H.B.
Radiomic machine-learning classifiers for prognostic biomarkers of advanced nasopharyngeal carcinoma
Zhang B., He X., Ouyang F., Gu D., Dong Y., Zhang L., Mo X., Huang W., Tian J., Zhang S.
Are you a researcher?
Create a profile to get free access to personal recommendations for colleagues and new articles.