• Laser & Optoelectronics Progress
  • Vol. 59, Issue 4, 0410003 (2022)
Shan Wang1, Yiying Hu1、*, Liang Feng2, and Linying Guo2
Author Affiliations
  • 1School of Information Engineering, East China JiaoTong University, Nanchang , Jiangxi 330013, China
  • 2Department of Breast Oncology, The Third Hospital of Nanchang, Nanchang , Jiangxi 330009, China
  • show less
    DOI: 10.3788/LOP202259.0410003 Cite this Article Set citation alerts
    Shan Wang, Yiying Hu, Liang Feng, Linying Guo. Improved Breast Mass Recognition YOLOv3 Algorithm Based on Cross-Layer Feature Aggregation[J]. Laser & Optoelectronics Progress, 2022, 59(4): 0410003 Copy Citation Text show less

    Abstract

    Aiming at the problem that small masses and occluded masses are easy to be missed in breast cancer diagnosis based on deep learning, an improved YOLOv3 algorithm for breast mass detection is proposed. First, a bottom-up path is added into the feature fusion module, and the cascading and cross-layer connections are adopted to make full use of the underlying feature information to improve the recognition accuracy of small masses. Second, to filter out more accurate prediction bounding boxes and avoid missed detection of masses that occlude each other, the distance intersection over union (DIoU) is introduced in soft non-maximum suppression (Soft-NMS) algorithm to suppress the redundant prediction bounding boxes. The experimental results demonstrate that the proposed breast mass detection algorithm has high accuracy and speed in detecting small masses and occluded masses, mean average precision (mAP@0.5) reaches 96.1%, which is 1.8 percentage point higher than that of YOLOv3, and the detection time of each mammogram target image is only 28 ms.
    Shan Wang, Yiying Hu, Liang Feng, Linying Guo. Improved Breast Mass Recognition YOLOv3 Algorithm Based on Cross-Layer Feature Aggregation[J]. Laser & Optoelectronics Progress, 2022, 59(4): 0410003
    Download Citation