Laser & Optoelectronics Progress, Volume. 58, Issue 4, 0410007(2021)

Mass Classification of Breast Mammogram Based on Attention Mechanism and Transfer Learning

Wenhui Xu, Yijian Pei*, Donglin Gao, Jiude Zhu, and Yunkai Liu
Author Affiliations
  • School of Information, Yunnan University, Kunming, Yunnan 650500, China
  • show less

    This study aimed to address issues of difficult diagnosis of benign and malignant masses in breast mammogram. For medical imaging, this study proposed a classification method of benign and malignant masses in breast mammogram based on attention mechanism and transfer learning. First, a new network model was built by combining convolutional block attention module (CBAM) and the residual network ResNet50 to improve the ability of the network to extract the features of the mass lesions and enhance specific semantic feature representation. Then, a new transfer learning method was proposed; instead of traditional method using the ImageNet as the transfer learning source domain, the patch data were used as the transfer learning source domain to complete the domain adaptive learning from local mass patch images to global breast mammogram, which can improve the ability of the network to capture pathological features. The experimental results show that the proposed method achieves an area under the receiver operating characteristics curve (AUC) value of 0.8607 in the local breast mass patch dataset and an AUC value of 0.8081 in the global breast mammogram dataset. The results confirm the effectiveness of the proposed classification method.

    Tools

    Get Citation

    Copy Citation Text

    Wenhui Xu, Yijian Pei, Donglin Gao, Jiude Zhu, Yunkai Liu. Mass Classification of Breast Mammogram Based on Attention Mechanism and Transfer Learning[J]. Laser & Optoelectronics Progress, 2021, 58(4): 0410007

    Download Citation

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

    Category: Image Processing

    Received: Jun. 9, 2020

    Accepted: Aug. 6, 2020

    Published Online: Feb. 8, 2021

    The Author Email: Pei Yijian (pei3p@ynu.edu.cn)

    DOI:10.3788/LOP202158.0410007

    Topics