Breast Cancer Image Classification Based on Attention Mechanisms

Yuyuan Guo; Sinuo Duan; Fangqi Wang

doi:10.62051/4y666a74

Authors

Yuyuan Guo
Sinuo Duan
Fangqi Wang

DOI:

https://doi.org/10.62051/4y666a74

Keywords:

Deep learning; Image Classification; Attention Mechanisms.

Abstract

Convolutional neural networks have been frequently utilized in computer-aided diagnosis (CAD). Breast cancer image classification is one of the vital applications of CAD. The purpose of this study was to explore the role of attention mechanism in breast cancer image classification. Specifically, this study introduces attention mechanism into the classical image classification deep learning network and constructs a new breast cancer image classification model. Test results indicate that convolutional neural networks perform better in classification when an attention mechanism is added, and they also perform better in terms of training loss and accuracy.

Downloads

Download data is not yet available.

References

[1] Zaalouk AM,Ebrahim GA,Mohamed HK, et al. A Deep Learning Computer-Aided Diagnosis Approach for Breast Cancer. Bioengineering (Basel). 2022;9 (8):. doi:10.3390/bioengineering9080391

[2] Li J, Chen Z, Su K, Zeng J. Clinicopathological classification and traditional prognostic indicators of breast cancer. Int J Clin Exp Pathol. 2015 Jul 1;8(7):8500-5. PMID: 26339424; PMCID: PMC4555752.

[3] Francesco Tacchino,Chiara Macchiavello,Dario Gerace, et al. An artificial neuron implemented on an actual quantum processor. npj Quantum Information. 2019;5 (1):0-0. doi:10.1038/s41534-019-0140-4

[4] Jiang, Yun; Chen, Li; Zhang, Hai; Xiao, Xiao; Hu, Jianjun (2019). Breast cancer histopathological image classification using convolutional neural networks with small SE-ResNet module. PLOS ONE, 14(3), e0214587–. doi:10.1371/journal.pone.0214587 no. 10, pp. 1563–1572, 2013.

[5] P. Filipczuk, T. Fevens, A. Krzy˙zak, and R. Monczak, “Computer-aided breast cancer diagnosis based on the analysis of cytological images of fine needle biopsies,” IEEE Transactions on Medical Imaging, vol. 32,no. 12, pp. 2169–2178, 2013.

[6] Sarvamangala DR, Kulkarni RV. Convolutional neural networks in medical image understanding: a survey. Evol Intell. 2022;15(1):1-22. doi: 10.1007/s12065-020-00540-3. Epub 2021 Jan 3. PMID: 33425040; PMCID: PMC7778711.

[7] Chen X, Wang X, Zhang K, Fung KM, Thai TC, Moore K, Mannel RS, Liu H, Zheng B, Qiu Y. Recent advances and clinical applications of deep learning in medical image analysis. Med Image Anal. 2022 Jul;79:102444. doi: 10.1016/j.media.2022.102444. Epub 2022 Apr 4. PMID: 35472844; PMCID: PMC9156578.

[8] Zuo T, Zheng Y, He L, Chen T, Zheng B, Zheng S, You J, Li X, Liu R, Bai J, Si S, Wang Y, Zhang S, Wang L, Chen J. Automated Classification of Papillary Renal Cell Carcinoma and Chromophobe Renal Cell Carcinoma Based on a Small Computed Tomography Imaging Dataset Using Deep Learning. Front Oncol. 2021 Nov 18;11:746750. doi: 10.3389/fonc.2021.746750. PMID: 34868946; PMCID: PMC8637858.

[9] Zopes J, Platscher M, Paganucci S, Federau C. Multi-Modal Segmentation of 3D Brain Scans Using Neural Networks. Front Neurol. 2021 Jul 14;12:653375. doi: 10.3389/fneur.2021.653375. PMID: 34335436; PMCID: PMC8318570.

[10] Zhou X, Ma L, Mubarak HK, Little JV, Chen AY, Myers LL, Sumer BD, Fei B. Automatic detection of head and neck squamous cell carcinoma on pathologic slides using polarized hyperspectral imaging and deep learning. Proc SPIE Int Soc Opt Eng. 2022 Feb-Mar;12039:120390G. doi: 10.1117/12.2614624. Epub 2022 Apr 4. PMID: 36798940; PMCID: PMC9930132.

[11] Krizhevsky A , Sutskever I , Hinton G . ImageNet Classification with Deep Convolutional Neural Networks[J]. Advances in neural information processing systems, 2012, 25(2).

[12] Simonyan K , Zisserman A . Very Deep Convolutional Networks for Large-Scale Image Recognition[J]. Computer Science, 2014.

[13] K. He, X. Zhang, S. Ren and J. Sun, "Deep Residual Learning for Image Recognition," 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA, 2016, pp. 770-778, doi: 10.1109/CVPR.2016.90.

[14] G. Huang, Z. Liu, L. Van Der Maaten and K. Q. Weinberger, "Densely Connected Convolutional Networks," 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA, 2017, pp. 2261-2269, doi: 10.1109/CVPR.2017.243.

[15] Meng-Hao Guo1, Tian-Xing Xu1, Jiang-Jiang Liu2.Attention mechanisms in computer vision: A survey..Computational Visual Media.2022;https://doi.org/10.1007/s12065-020-00540-3.

[16] HU J, SHEN L, SUN G． Squeeze－and－excitation networks[C]/ /Proceedings of the IEEE conference on computer vision and pattern recognition． 2018: 7132－7141.

[17] ＲOY A G, NAVAB N, WACHINGEＲC．Concurrent spatial and channel‘squeeze ＆ excitation’in fully convolutional networks［C］/ /International conference on medical image computing and computer－assisted intervention．Springer, Cham, 2018: 421－429．

[18] Woo S, Park J, Lee J Y, et al. Cbam: Convolutional block attention module[C]//Proceedings of the European Conference on Computer Vision (ECCV). 2018: 3-19.

[19] Spanhol F A, Oliveira L S, Petitjean C, et al. A dataset for breast cancer histopathological image classification[J]. Ieee transactions on biomedical engineering, IEEE, 2015, 63(7): 1455-1462.

[20] Abdallah N, Marion JM, Tauber C, Carlier T, Hatt M, Chauvet P. Enhancing histopathological image classification of invasive ductal carcinoma using hybrid harmonization techniques. Sci Rep. 2023 Nov 16;13(1):20014