Improved FCOS for Detecting Breast Cancers

Abstract

Purpose: Breast cancer ranks first among cancers affecting women's health. Our goal is to develop a fast, high-precision, and fully automated breast cancer detection algorithm to improve the early detection rate of breast cancer.

Methods: We compare different object detection algorithms, including anchor-based and anchor-free object detection algorithms for detecting breast lesions. Finally, we find that the fully convolutional onestage object detection (FCOS) showed the best performance in the detection of breast lesions, which is an anchor-free algorithm. 1) Considering that the detection of breast lesions requires the context information of the ultrasound images, we introduce the non-local technique, which models long-range dependency between pixels to the FCOS algorithm, providing the global context information for the detection of the breast lesions. 2) The variety of shapes and sizes of breast lesions makes detection difficult. We propose a new deformable spatial attention (DSA) module and add it to the FCOS algorithm.

Results: The detection performance of the original FCOS is that the average precision (AP) for benign lesions is 0.818, and for malignant lesions is 0.888. The FCOS with a non-local module improves the performance of the breast detection; the AP of benign lesions was 0.819, and that of malignant lesions was 0.894. Combining the DSA module with the FCOS improves the performance of breast detection; the AP for benign lesions and malignant lesions is 0.840 and 0.899, respectively.

Conclusion: We propose two methods to improve the FCOS algorithm from different perspectives to improve its performance in detecting breast lesions. We find that FCOS combined with DSA is beneficial in improving the localization and classification of breast tumors and can provide auxiliary diagnostic advice for ultrasound physicians, which has a certain clinical application value.

Keywords: FCOS; Object detection algorithm; breast cancers; deformable spatial attention; non-local module; ultrasound.