Abstract
Breast cancer is currently one of the most common cancers among women worldwide. Computer-aided diagnosis (CAD) is crucial in reducing diagnostic costs and improving subsequent cancer treatment. Convolutional neural networks are widely used in CAD, but most existing models are high-complexity and multi-parameter architectures, which limits their practical application. To address this issue, this paper proposes LMCNet, a lightweight breast cancer multi-classification model. LMCNet reduces the model complexity by refactoring the basic units of ShuffleNetV2, introducing PAConv parallel convolution layers and combining depthwise separable convolution. In order to further improve the feature extraction ability of the model, an SAE attention module containing channel and spatial dimensions is introduced. The proposed model achieves high recognition accuracy with low resource consumption. Our method requires only 275.69 K parameters and 51.19 M FLOPs. Compared to ShuffleNetV2, LMCNet reduces parameters by approximately 78.63% and FLOPs by 66.18%. Compared with other classical models, the proposed model achieves the best recognition efficiency and the fastest inference speed.
Similar content being viewed by others
Data availability
The dataset used in this study is publicly available and can be accessed at https://web.inf.ufpr.br/vri/databases/breast-cancer-histopathological-database-breakhis/. and https://iciar2018-challenge.grand-challenge.org/Dataset/. All relevant data for the analysis are available from the corresponding author upon reasonable request.
References
Basu, P., Lucas, E., Carvalho, A.L., Sauvaget, C., Muwonge, R., Herrero, R., Sankaranarayanan, R.: Cancer screening in five continents. Lyon, France: International Agency for Research on Cancer. Available from: https://canscreen5.iarc.fr. Accessed [28/08/2024] (2019)
Barba, D., León-Sosa, A., Lugo, P., et al., 2021. Breast cancer, screening and diagnostic tools: all you need to know. Crit. Rev. Oncol. Hematol. 157, 103174. (2021) https://doi.org/10.1016/j.critrevonc.2020.103174
Zebari, D.A., Ibrahim, D.A., Zeebaree, D.Q., et al.: Systematic review of computing approaches for breast cancer detection based computer aided diagnosis using mammogram images. Appl. Artif. Intell. 35 (15), 2157–2203. (2021) https://doi.org/10.1080/08839514.2021.2001177.
Sun, Y., Xue, B., Zhang, M., et al.: Evolving deep convolutional neural networks for image classification. IEEE Trans. Evol. Comput. 24 (2), 394–407. (2019) https://doi.org/10.1109/TEVC.2019.2916183.
Dhillon, A., Verma, G.K.: Convolutional neural network: a review of models, methodologies and applications to object detection. Prog. in Artif. Intel. 9 (2), 85–112. (2020) https://doi.org/10.1007/s13748-019-00203-0.
Guo, Y., Liu, Y., Georgiou, T., et al.: A review of semantic segmentation using deep neural networks. Int. J. Multimed. Inf. Retr. 7, 87–93. (2018). https://doi.org/10.1007/s13735-017-0141-z.
Amrane, M., Oukid, S., Gagaoua, I., et al.: Breast cancer classification using machine learning. 2018 Electric Electronics, Computer Science, Biomedical Engineerings' Meeting.pp.1–4.(2018)https://doi.org/10.1109/EBBT.2018.8391453.
Atban, F., Ekinci, E., Garip, Z.: Traditional machine learning algorithms for breast cancer image classification with optimized deep features. Biomed. Signal Process. Control. 81, 104534. (2023). https://doi.org/10.1016/j.bspc.2022.104534.(2023)
Zhongyi H, Benzheng W, Yuanjie Z, et al.: Breast Cancer Multi-classification from Histopathological Images with Structured Deep Learning Model. [J]. Scientific reports (1–4:4172). (2017). https://doi.org/10.1038/s41598-017-04075-z.
Pawer, M.M., Pujari, S.D., Pawar, S.P., et al.: MuSCF-Net: multi-scale, multi-channel feature network using Resnet-based attention mechanism for breast histopathological image classification. Machine Learning and Deep Learning Techniques for Medical Science. CRC Press., pp. 243–261. (2022) https://doi.org/10.1201/9781003217497.
Gupta, S., Agrawal, S., Singh, S.K., et al.: A novel transfer learning-based model for ultrasound breast cancer image classification. Computational Vision and Bio-Inspired Computing. pp. 511–523. (2023). https://doi.org/10.1007/978-981-19-9819-5_37.
Li, J., Shi, J., Chen, J., et al.: Self-attention random forest for breast cancer image classification. Front. in Oncol. 13, 1043463. (2023) https://doi.org/10.3389/fonc.2023.1043463.
Nawaz M, Sewissy A.A., Soliman A.H.T.: Multi-class breast cancer classification using deep learning convolutional neural network. Int J Adv Comput Sci Appl (IJACSA). (2018) https://doi.org/10.14569/IJACSA.2018.090645.
Laxmisagar, H.S., Hanumantharaju, M.C.: Detection of breast cancer with lightweight deep neural networks for histology image classification. Crit. Rev. Biomed. Eng. 50 (2), 1–19. (2022) https://doi.org/10.1615/critrevbiomedeng.2022043417.
Xiao, M., Li, Y., Yan, X., et al.: Convolutional neural network classification of cancer cytopathology images: taking breast cancer as an example. In: Proceedings of the 2024 7th International Conference on Machine Vision and Applications. pp. 145–149. (2024) https://doi.org/10.1145/3653946.3653968.
Addo, D., Zhou, S., Sarpong, K., et al.: A hybrid lightweight breast cancer classification framework using the histopathological images. Biocybern. and Biomed. Eng. 44 (1), 31–54. (2024) https://doi.org/10.1016/j.bbe.2023.12.003.
Wakili, M.A., Shehu, H.A., Sharif, M.H., Sharif, M.H.U., Umar, A., Kusetogullari, H., Ince, I.F., Uyaver, S.: Classification of breast cancer histopathological images using densenet and transfer learning. Comput. Intell. Neurosci. (2022). https://doi.org/10.1155/2022/8904768
Imran I, Muhammad Y, Khuram W, et al.: Automated Multi-class Classification of Skin Lesions through Deep Convolutional Neural Network with Dermoscopic Images. Computerized Medical Imaging and Graphics, 88(prepublish), 101843 (2020). https://doi.org/10.1016/J.COMPMEDIMAG.2020.101843.
Spanhol, F., Oliveira, L.S., Petitjean, C., Heutte, L.: 2016, A dataset for breast cancer histopathological image classification. IEEE Trans. Biomed. Eng. (TBME) 63(7), 1455–1462 (2016). https://doi.org/10.1109/TBME.2015.2496264
Liu, Z., Lv, Q., Li, Y., et al.: Medaugment: universal automatic data augmentation plug-in for medical image analysis. arXiv preprint arXiv:2306.17466 (2023). https://doi.org/10.48550/arXiv.2306.17466.
Aresta G, Araújo T, Kwok S, et al.: BACH: Grand challenge on breast cancer histology images. Medical Image Analysis, 56122–139. (2019). https://doi.org/10.1016/j.media.2019.05.010.
Ma, N., Zhang, X., Zheng, H.T., et al.: Shufflenet v2: Practical guidelines for efficient cnn architecture design. Proceedings of the European Conference on Computer Vision. pp. 116–131 (2018). https://doi.org/10.48550/arXiv.1807.11164.
Zhang, Q.L., Yang, Y.B.: Sa-net: Shuffle attention for deep convolutional neural networks. ICASSP 2021–2021 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). IEEE. pp. 2235–2239 (2021). https://doi.org/10.1109/ICASSP39728.2021.9414568.
Woo, S., Park, J., Lee, J.-Y.: So Kweon. In: Proceedings of the European Conference on Computer Vision (ECCV), pp. 3–19 (2018). https://doi.org/10.48550/arXiv.1807.06521.
Hu, J., Shen, L., Sun, G.: Squeeze-and-excitation networks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. pp. 7132–7141 (2018). https://doi.org/10.48550/arXiv.1709.01507.
Szegedy, C., Vanhoucke, V., Ioffe, S., Shlens, J., Wojna, Z.: In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 2818–2826 (2016)
Wei, B., Han, Z., He, X., Yin, Y.: Deep learning model based breast cancer histopathological image classification. In: 2017 IEEE 2nd international conference on cloud computing and big data analysis (ICCCBDA) (pp. 348–353). IEEE (2017). https://doi.org/10.1109/ICCCBDA.2017.7951937.
Li, J., Shi, J., Su, H. and Gao, L.: Breast cancer histopathological image recognition based on pyramid gray level co-occurrence matrix and incremental broad learning. Electronics, 11(15), p.2322 (2022). https://doi.org/10.3390/electronics11152322.
He, K., Zhang, X., Ren, S., et al.: Deep residual learning for image recognition. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR). pp. 770–778 (2016)
Huang, G., Liu, Z., Maaten, L.V.D, et al.: Densely connected convolutional networks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. pp. 4700–4708 (2017). https://doi.org/10.48550/arXiv.1608.06993.
Sandler, M., Howard, A., Zhu, M., et al.: Mobilenetv2: Inverted residuals and linear bottlenecks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. pp. 4510–4520. (2018). https://doi.org/10.48550/arXiv.1801.04381.
Linxian L, Wenxiang F, Cheng C, et al.: Classification of breast cancer histology images using MSMV-PFENet. [J]. Scientific Reports, 12(1):17447–17447. (2022). https://doi.org/10.1038/S41598-022-22358-Y.
Li Guang-li, YE Yi-yuan, Wu Guang-ting. et al.: Breast cancer pathological image classification model via combining multi-view transformer coding and online fusion mutual learning. Acta Electron. Sin. 52(07), 2369–2381 (2024)
Singh, V.K., Kalafi, E.Y., Wang, S., et al.: Prior wavelet knowledge for multi-modal medical image segmentation using a lightweight neural network with attention guided features. Expert Syst. Appl. 209, 118166 (2022). https://doi.org/10.1016/j.eswa.2022.118166
Xie, X., Zhang, X., Tang, X., et al.: MACTFusion: lightweight cross transformer for adaptive multimodal medical image fusion. IEEE J. Biomed. Health Inform. 19 (2024). https://doi.org/10.1109/JBHI.2024.3391620.
Funding
This research is supported by the Natural Science Foundation of Hebei Province (Grant No. F2024207004), the National Natural Science Foundation of China (Grant No. 61902107) and the Scientific Research and Development Program of Hebei University of Economics and Business (Grant No.2021ZD02).
Author information
Authors and Affiliations
Contributions
XM wrote the main manuscript text, carried out the implementation of the software and conceptualised the experimental design. LS carried out the editing and review of the manuscript and participated in the conceptualisation of the experimental methodology. YD and JG carried out the data collection and processing, and performed the visualisation of the experimental data. All authors reviewed the manuscript.
Corresponding author
Ethics declarations
Conflict of interest
This paper does not contain any studies with human or animal subjects and all authors declare that they have no conflict of interest.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Ma, X., Sun, L., Gao, J. et al. LMCNet: a lightweight and efficient model for multi-classification of breast cancer images. SIViP 19, 175 (2025). https://doi.org/10.1007/s11760-024-03743-8
Received:
Revised:
Accepted:
Published:
DOI: https://doi.org/10.1007/s11760-024-03743-8