Oral cancer remains a critical global health challenge, characterized by high morbidity and mortality due to late-stage diagnosis. This paper addresses the need for improved diagnostic accuracy by introducing a novel hybrid-attention aided deep learning model for oral cancer classification. Our approach integrates the EfficientNetB0 model for feature extraction with a customized hybrid attention module, combining self, spatial, and soft attention mechanisms to emphasize crucial features. The model has been trained and evaluated on both microscopic and macroscopic oral cancer datasets, demonstrating its generalizability. Our proposed model achieves an accuracy of 96.97% on a macroscopic dataset and 92% on a microscopic dataset, with respective F1-scores of 0.97 and 0.89, outperforming baseline models. This hybrid attention-aided deep learning model shows significant potential in improving early detection and treatment planning of oral cancer. Source code and additional results of this work can be found at the GitHub repository.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Oral Cancer Classification Using a Hybrid Attention-Aided Deep Learning Model

  • Srinjan Dutta,
  • Jibesh Roy,
  • Diptarka Mandal,
  • Dmitrii Kaplun,
  • Irina Shpakovskaya,
  • Ram Sarkar

摘要

Oral cancer remains a critical global health challenge, characterized by high morbidity and mortality due to late-stage diagnosis. This paper addresses the need for improved diagnostic accuracy by introducing a novel hybrid-attention aided deep learning model for oral cancer classification. Our approach integrates the EfficientNetB0 model for feature extraction with a customized hybrid attention module, combining self, spatial, and soft attention mechanisms to emphasize crucial features. The model has been trained and evaluated on both microscopic and macroscopic oral cancer datasets, demonstrating its generalizability. Our proposed model achieves an accuracy of 96.97% on a macroscopic dataset and 92% on a microscopic dataset, with respective F1-scores of 0.97 and 0.89, outperforming baseline models. This hybrid attention-aided deep learning model shows significant potential in improving early detection and treatment planning of oral cancer. Source code and additional results of this work can be found at the GitHub repository.