We'd appreciate your feedback. Send feedback Subscribe to our newsletters and alerts


Journal of Current Research in Oral Surgery

2021 Volume 1

YOLOv11-MobileNetV2 Two-Stage AI Framework for Lesion Localization and Differentiation of Oral Cancer and Precancerous Lesions


, ,
  1. Department of Oral and Maxillofacial Surgery, School of Dentistry, University of Melbourne, Melbourne, Australia.
  2. Oral Surgery Unit, Royal Dental Hospital of Melbourne, Melbourne, Australia.
  3. Department of Oral and Maxillofacial Surgery, Monash Health, Melbourne, Australia.
Abstract

This work sought to construct and assess an artificial intelligence workflow that merges object-detection and image-classification models to support early recognition and distinction of oral lesions. A retrospective cross-sectional design was applied, using clinical photographs of oral potentially malignant disorders and oral squamous cell carcinoma. The primary dataset consisted of 773 images from the Faculdade de Odontologia de Piracicaba, Universidade Estadual de Campinas (FOP-UNICAMP), and an independent validation set included 132 images from the Federal University of Paraíba (UFPB). All images were captured before biopsy, each paired with histopathological confirmation. For lesion localization, ten YOLOv11 variants employing different augmentation schemes were trained for 200 epochs with pretrained COCO weights. For classification, three MobileNetV2 networks were trained on crops generated according to expert bounding boxes, each adopting distinct learning rate and augmentation configurations. After identifying the top-performing detection–classification pair, both components were linked in a two-stage pipeline in which the detector-generated crops were forwarded into the classifier. The optimal YOLOv11 model achieved an mAP50 of 0.820, precision of 0.897, recall of 0.744, and an F1-score of 0.813. The strongest MobileNetV2 model reached an accuracy of 0.846, precision of 0.871, recall of 0.846, F1-score of 0.844, and an AUC-ROC of 0.852. On the external set, the same classifier obtained an accuracy of 0.850, precision of 0.866, recall of 0.850, an F1-score of 0.851, and an AUC-ROC of 0.935. The integrated two-step framework, tested on the baseline dataset, achieved an accuracy of 0.784, precision of 0.793, recall of 0.784, F1-score of 0.784, and an AUC-ROC of 0.811. When applied to the independent dataset, it produced an accuracy of 0.863, a precision of 0.879, a recall of 0.863, F1-score of 0.866, and an AUC-ROC of 0.934. Visual review of the YOLO outputs showed consistent lesion localization across varied oral images, though 17.4% were not detected. The t-SNE map revealed partial clustering of OPMD and OSCC embeddings, suggesting the model captured relevant discriminative signals despite some overlap. This proof-of-concept investigation indicates that a coupled detection–classification AI framework can feasibly support early screening of oral lesions. Nonetheless, caution is necessary when interpreting two-stage results, since images not detected by YOLO do not advance to classification, potentially influencing the final metrics.


How to cite this article
Vancouver
Nguyen TK, Bennett SJ, Carter WJ. YOLOv11-MobileNetV2 Two-Stage AI Framework for Lesion Localization and Differentiation of Oral Cancer and Precancerous Lesions. J Curr Res Oral Surg. 2021;1:98-109. https://doi.org/10.51847/kMIYO81bAR
APA
Nguyen, T. K., Bennett, S. J., & Carter, W. J. (2021). YOLOv11-MobileNetV2 Two-Stage AI Framework for Lesion Localization and Differentiation of Oral Cancer and Precancerous Lesions. Journal of Current Research in Oral Surgery, 1, 98-109. https://doi.org/10.51847/kMIYO81bAR

About TSDP

Find out more

Our platform is dedicated to covering all facets of dental health, technology, education, and innovation. From general dentistry and orthodontics to cosmetic procedures, oral surgery, and the latest advancements in dental science, we strive to be a one-stop destination for professionals, students, and anyone passionate about dental care.

Our mission is to elevate the dental field by fostering knowledge sharing and encouraging the adoption of cutting-edge practices. We are committed to bridging the gap between innovation and application, making the latest research, trends, and technological breakthroughs accessible to everyone. Whether you're a seasoned practitioner seeking to refine your expertise, a student eager to stay ahead in your studies, or simply someone curious about oral health, our website is designed to empower and inspire.