Laser & Optoelectronics Progress, Volume. 58, Issue 4, 0410007(2021)
Mass Classification of Breast Mammogram Based on Attention Mechanism and Transfer Learning
Figures & Tables(9)
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. 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
View tableTable 1. Comparison of classification performance of different network models on patch dataset P
Model Sensitivity Specificity Accuracy AUC VGG16 VGG16+CBAM 0.7619 0.7649 0.7748 0.8073 0.7698 0.7831 0.8395 0.8503 Inception V3 Inception V3+CBAM 0.7687 0.7699 0.7792 0.7806 0.7751 0.7772 0.8403 0.8477 ResNet50 0.7551 0.7835 0.7724 0.8348 Proposed network 0.7993 0.8074 0.8042 0.8607
Table 2. Comparison of classification performance of different network models and different transfer learning methods on global breast dataset G
View tableTable 2. Comparison of classification performance of different network models and different transfer learning methods on global breast dataset G
Method Transfer learning Sensitivity Specificity Accuracy AUC VGG16 ImageNet Patch data 0.6835 0.7143 0.7272 0.7380 0.7097 0.7288 0.7599 0.7749 VGG16+CBAM ImageNet Patch data 0.7194 0.7278 0.7177 0.7408 0.7184 0.7335 0.7885 0.7913 Inception V3 ImageNet Patch data 0.6619 0.7122 0.6794 0.7416 0.6724 0.7298 0.7541 0.7941 Inception V3+CBAM ImageNet Patch data 0.6835 0.7482 0.7321 0.6986 0.7126 0.7184 0.7954 0.7656 ResNet50 ImageNet Patch data 0.7122 0.7482 0.7464 0.7037 0.7327 0.7341 0.7881 0.7929 Proposed network ImageNet Patch data 0.7350 0.7698 0.7456 0.7416 0.7414 0.7529 0.7982 0.8081
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
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)
© Copyright 2018-2021 | Chinese Laser Press.
All Rights Reserved 沪ICP备15018463号-20