Deep Adaptively Feature Extracting Network for Cervical Squamous Lesion Cell Detection

Abstract

AbstractCervical cancer is one of the most widespread malignancies affecting women’s health worldwide today. However, the task of detection is particularly difficult due to the complex background of the cervical smear, where cells are previously stacked in clusters. To address this problem, we utilize the YOLOv5 as the baseline and build on YOLOv5 by using the simple Transformer Block which only combines with multi-head self attention layers and MLP to better extract cell features as well as to obtain global information. In addition, we allow the model to refine features adaptively to assist with detection by using Convolutional Block Attention Module (CBAM), an attention module being simple and effective for feed-forward convolutional neural networks, in the complex background information. Finally, we compare the model with YOLOv5 as baseline. In CDetector dataset, our model obtains 52.5% mAP@.5, which is 6% better than baseline. In transfer learning, it is 62.2%, which outperforms baseline by 3.2%.KeywordsCervical squamous lesion cellsYOLOv5TransformerDeep learning