Laser & Optoelectronics Progress, Volume. 59, Issue 4, 0410003(2022)

Improved Breast Mass Recognition YOLOv3 Algorithm Based on Cross-Layer Feature Aggregation

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

    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.

    Tools

    Get Citation

    Copy Citation Text

    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

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Feb. 8, 2021

    Accepted: Mar. 22, 2021

    Published Online: Jan. 25, 2022

    The Author Email: Hu Yiying (yiyinghu1125@163.com)

    DOI:10.3788/LOP202259.0410003

    Topics