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
- Laser & Optoelectronics Progress
- Vol. 58, Issue 4, 0410007 (2021)
Fig. 1. Workflow of the proposed method
Fig. 2. Mammograms and the pixel-level lesion labels of different views (L-CC, L-MLO, R-CC and R-MLO) from CBIS-DDSM dataset
Fig. 3. Images from different datasets. (a)Local breast mass patch image from dataset P; (b)whole breast mammogram from CBIS-DDSM dataset;(c)global breast mammogram from dataset G
Fig. 4. Residual block structure in ResNet50 Network
Fig. 5. Structure of channel attention module
Fig. 6. Structure of spatial attention module
Fig. 7. Mammogram from dataset G and heat maps of the corresponding Grad-CAM. (a)Sample 1(benign); (b)sample 2(benign);(c)sample 3(malignant);(d)sample 4(malignant);(e)sample 5(malignant)
|
Table 1. Comparison of classification performance of different network models on patch dataset P
|
Table 2. Comparison of classification performance of different network models and different transfer learning methods on global breast dataset G
Set citation alerts for the article
Please enter your email address
© Copyright 2018-2021 | Chinese Laser Press. All Rights Reserved 沪ICP备15018463号-20