Comparison of machine and deep learning for the classification of cervical cancer based on cervicography imagesopen access
- Authors
- Park, Ye Rang; Kim, Young Jae; Ju, Woong; Nam, Kyehyun; Kim, Soonyung; Kim, Kwang Gi
- Issue Date
- 9-Aug-2021
- Publisher
- Nature Publishing Group
- Citation
- Scientific Reports, v.11, no.1
- Journal Title
- Scientific Reports
- Volume
- 11
- Number
- 1
- URI
- https://scholarworks.bwise.kr/sch/handle/2021.sw.sch/19836
- DOI
- 10.1038/s41598-021-95748-3
- ISSN
- 2045-2322
- Abstract
- Cervical cancer is the second most common cancer in women worldwide with a mortality rate of 60%. Cervical cancer begins with no overt signs and has a long latent period, making early detection through regular checkups vitally immportant. In this study, we compare the performance of two different models, machine learning and deep learning, for the purpose of identifying signs of cervical cancer using cervicography images. Using the deep learning model ResNet-50 and the machine learning models XGB, SVM, and RF, we classified 4119 Cervicography images as positive or negative for cervical cancer using square images in which the vaginal wall regions were removed. The machine learning models extracted 10 major features from a total of 300 features. All tests were validated by fivefold cross-validation and receiver operating characteristics (ROC) analysis yielded the following AUCs: ResNet-50 0.97(CI 95% 0.949-0.976), XGB 0.82(CI 95% 0.797-0.851), SVM 0.84(CI 95% 0.801-0.854), RF 0.79(CI 95% 0.804-0.856). The ResNet-50 model showed a 0.15 point improvement (p < 0.05) over the average (0.82) of the three machine learning methods. Our data suggest that the ResNet-50 deep learning algorithm could offer greater performance than current machine learning models for the purpose of identifying cervical cancer using cervicography images.
- Files in This Item
- There are no files associated with this item.
- Appears in
Collections - College of Medicine > Department of Obstetrics and Gynecology > 1. Journal Articles
Items in ScholarWorks are protected by copyright, with all rights reserved, unless otherwise indicated.