Skip to main content

Advertisement

SpringerLink
Log in

Breast cancer masses classification using deep convolutional neural networks and transfer learning

  • Published:
Multimedia Tools and Applications Aims and scope Submit manuscript

Abstract

With the recent advances in the deep learning field, the use of deep convolutional neural networks (DCNNs) in biomedical image processing becomes very encouraging. This paper presents a new classification model for breast cancer masses based on DCNNs. We investigated the use of transfer learning from AlexNet and GoogleNet pre-trained models to suit this task. We experimentally determined the best DCNN model for accurate classification by comparing different models, which vary according to the design and hyper-parameters. The effectiveness of these models were demonstrated using four mammogram databases. All models were trained and tested using a mammographic dataset from CBIS-DDSM and INbreast databases to select the best AlexNet and GoogleNet models. The performance of the two proposed models was further verified using images from Egyptian National Cancer Institute (NCI) and MIAS database. When tested on CBIS-DDSM and INbreast databases, the proposed AlexNet model achieved an accuracy of 100% for both databases. While, the proposed GoogleNet model achieved accuracy of 98.46% and 92.5%, respectively. When tested on NCI images and MIAS databases, AlexNet achieved an accuracy of 97.89% with AUC of 98.32%, and accuracy of 98.53% with AUC of 98.95%, respectively. GoogleNet achieved an accuracy of 91.58% with AUC of 96.5%, and accuracy of 88.24% with AUC of 94.65%, respectively. These results suggest that AlexNet has better performance and more robustness than GoogleNet. To the best of our knowledge, the proposed AlexNet model outperformed the latest methods. It achieved the highest accuracy and AUC score and the lowest testing time reported on CBIS-DDSM, INbreast and MIAS databases.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Log in via an institution

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1.
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7

Similar content being viewed by others

References

  1. Abbas Q (2016) DeepCAD: a computer-aided diagnosis system for mammographic masses using deep invariant features. Computers 5(4):28. https://doi.org/10.3390/computers5040028

    Article  Google Scholar 

  2. Abbas A, Abdelsamea MM, Gaber MM (2020) Detrac: transfer learning of class decomposed medical images in convolutional neural networks. IEEE Access 8:74901–74913

    Article  Google Scholar 

  3. Abdelhafiz D, Yang C, Ammar R, Nabavi S (2019) Deep convolutional neural networks for mammography: advances, challenges and applications. BMC bioinformatics 20(11):281

    Article  Google Scholar 

  4. Agarwal R, Diaz O, Lladó X, Yap MH, Martí R (2019) Automatic mass detection in mammograms using deep convolutional neural networks. Journal of Medical Imaging 6(3):031409

    Article  Google Scholar 

  5. Agnes SA, Anitha J, Pandian SIA, Peter JD (2020) Classification of mammogram images using multiscale all convolutional neural network (MA-CNN). J Med Syst 44:30. https://doi.org/10.1007/s10916-019-1494-z

    Article  Google Scholar 

  6. Al-antari MA, Al-masni MA, Choi MT, Han SM, Kim TS (2018) A fully integrated computer-aided diagnosis system for digital X-ray mammograms via deep learning detection, segmentation, and classification. International journal of medical informatics 117:44–54

    Article  Google Scholar 

  7. Al-antari MA, Al-masni MA, Kim TS (2020) Deep learning computer-aided diagnosis for breast lesion in digital mammogram. In: Lee G, Fujita H (eds) Deep learning in medical image analysis. Advances in experimental medicine and biology, vol 1213. Springer, Cham

    Google Scholar 

  8. Al-antari MA, Al-masni MA, Park SU, Park J, Metwally MK, Kadah YM, … Kim TS (2018) An automatic computer-aided diagnosis system for breast cancer in digital mammograms via deep belief network. Journal of Medical and Biological Engineering 38(3):443–456. https://doi.org/10.1007/s40846-017-0321-6

    Article  Google Scholar 

  9. Al-Antari MA, Han SM, Kim TS (2020) Evaluation of deep learning detection and classification towards computer-aided diagnosis of breast lesions in digital X-ray mammograms. Comput Methods Prog Biomed 196:105584. https://doi.org/10.1016/j.cmpb.2020.105584

    Article  Google Scholar 

  10. Al-masni MA, Al-antari MA, Park JM, Gi G, Kim TY, Rivera P, … Kim TS (2018) Simultaneous detection and classification of breast masses in digital mammograms via a deep learning YOLO-based CAD system. Computer methods and programs in biomedicine 157:85–94

    Article  Google Scholar 

  11. American Cancer Society (2019) Cancer facts & figures 2019. American Cancer Society, Atlanta. http://www.cancer.org/content/dam/cancer-org/research/cancer-facts-and-statistics/annual-cancer-facts-and-figures/2019/cancer-facts-and-figures-2019.pdf

  12. Arora R, Rai PK, Raman B (2020) Deep feature–based automatic classification of mammograms. Med Biol Eng Comput 58:1199–1211. https://doi.org/10.1007/s11517-020-02150-8

    Article  Google Scholar 

  13. Dhungel N, Carneiro G, Bradley AP (2017) A deep learning approach for the analysis of masses in mammograms with minimal user intervention. Med Image Anal 37:114–128

    Article  Google Scholar 

  14. Garcia-Garcia, A, Orts-Escolano, S, Oprea, S, Villena-Martinez, V and Garcia-Rodriguez, J (2017). A review on deep learning techniques applied to semantic segmentation. arXiv preprint arXiv:1704.06857

  15. Gardezi SJS, Elazab A, Lei B, Wang T (2019) Breast cancer detection and diagnosis using mammographic data: systematic review. J Med Internet Res 21(7):e14464

    Article  Google Scholar 

  16. Hassan SA, Sayed MS, Abdalla MI, Rashwan MA (2019) Detection of breast cancer mass using MSER detector and features matching. Multimed Tools Appl 78(14):20239–20262. https://doi.org/10.1007/s11042-019-7358-1

    Article  Google Scholar 

  17. Heath, M, Bowyer, K, Kopans, D, Moore, R and Kegelmeyer, WP (2000, June). The digital database for screening mammography. In proceedings of the 5th international workshop on digital mammography (pp. 212-218). Medical physics publishing. http://marathon.csee.usf.edu/Mammography/Database.html

  18. Kaur T, Gandhi TK (2020) Deep convolutional neural networks with transfer learning for automated brain image classification. Mach Vis Appl 31:20. https://doi.org/10.1007/s00138-020-01069-2

    Article  Google Scholar 

  19. Khan, FA, Butt, AUR, Asif, M, Ahmad W, Nawaz M, Jamjoom M, Alabdulkreem E (2020). Computer-aided diagnosis for burnt skin images using deep convolutional neural network . Multimed Tools Appl . https://doi.org/10.1007/s11042-020-08768-y

  20. Khan S, Rahmani H, Shah SAA, Bennamoun M (2018) A guide to convolutional neural networks for computer vision. Synthesis Lectures on Computer Vision 8(1):1–207

    Article  Google Scholar 

  21. Kingma, DPandBa, J (2014). Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980

  22. Krizhevsky, A, Sutskever, I and Hinton, GE (2012). Imagenet classification with deep convolutional neural networks. In advances in neural information processing systems (pp. 1097-1105)

  23. Lee, RS, Gimenez, F, Hoogi, A and Rubin, D (2016). Curated breast imaging subset of DDSM The cancer imaging archive, 8. https://doi.org/10.7937/K9/TCIA.2016.7O02S9CY

  24. Litjens G, Kooi T, Bejnordi BE, Setio AAA, Ciompi F, Ghafoorian M, van der Laak JAWM, van Ginneken B, Sánchez CI (2017) A survey on deep learning in medical image analysis. Med Image Anal 42:60–88

    Article  Google Scholar 

  25. Michelucci, U (2019). Advanced applied deep learning: convolutional neural networks and object detection. Apress.

  26. Moreira IC, Amaral I, Domingues I, Cardoso A, Cardoso MJ, Cardoso JS (2012) Inbreast: toward a full-field digital mammographic database. Acad Radiol 19(2):236–248

    Article  Google Scholar 

  27. Qian N (1999) On the momentum term in gradient descent learning algorithms. Neural Netw 12(1):145–151

    Article  MathSciNet  Google Scholar 

  28. Ragab DA, Sharkas M, Marshall S, Ren J (2019) Breast cancer detection using deep convolutional neural networks and support vector machines. PeerJ 7:e6201

    Article  Google Scholar 

  29. Rehman A, Naz S, Razzak MI, Akram F, Imran M (2020) A deep learning-based framework for automatic brain tumors classification using transfer learning. Circuits Syst Signal Process 39:757–775. https://doi.org/10.1007/s00034-019-01246-3

    Article  Google Scholar 

  30. Ribli D, Horváth A, Unger Z, Pollner P, Csabai I (2018) Detecting and classifying lesions in mammograms with deep learning. Sci Rep 8(1):4165

    Article  Google Scholar 

  31. Rouhi R, Jafari M, Kasaei S, Keshavarzian P (2015) Benign and malignant breast tumors classification based on region growing and CNN segmentation. Expert Syst Appl 42(3):990–1002

    Article  Google Scholar 

  32. Shen, L, Margolies, LR, Rothstein, JH, Fluder, E, McBride, R and Sieh, W (2019). Deep learning to improve breast Cancer detection on screening mammography. Sci Rep, 9

  33. Shu, X, Zhang, L, Wang, Z, Lv, Q and Yi, Z (2020). Deep neural networks with region-based pooling structures for mammographic image classification. IEEE transactions on medical imaging, 1–1. doi:https://doi.org/10.1109/tmi.2020.2968397

  34. Suckling, J, Parker, J, Dance, D, et al. (2015). Mammographic Image Analysis Society (MIAS) database v1.21 [Dataset]. https://www.repository.cam.ac.uk/handle/1810/250394

  35. Sundararajan, D (2017). Digital image processing: a signal processing and algorithmic approach. Springer..

  36. Szegedy, C, Liu, W, Jia, Y, Sermanet, P, Reed, S, Anguelov, D, ... and Rabinovich, A (2015). Going deeper with convolutions. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 1–9)

  37. U.S. Breast Cancer statistics (2019), www.breastcancer.org/symptoms/understand_bc/statistics Accessed 4 November 2019

  38. World health organization, Breast cancer [online] (2019) https://www.who.int/cancer/prevention/diagnosis-screening/breast-cancer/en/ Accessed 4 November 2019

  39. World health organization, cancer fact sheet [online] (2019) https://www.who.int/en/news-room/fact-sheets/detail/cancer Accessed 4 November 2019.

  40. Zhang H, Wu R, Yuan T, Jiang Z, Huang S, Wu J, Hua J, Niu Z, Ji D (2020) DE-Ada*: a novel model for breast mass classification using cross-modal pathological semantic mining and organic integration of multi-feature fusions. Inf Sci 539:461–486. https://doi.org/10.1016/j.ins.2020.05.080

    Article  Google Scholar 

Download references

Author information

Authors and Affiliations

  1. Department of Electronics and Communications Engineering, Zagazig University, Zagazig, Egypt

    Shayma’a A. Hassan, Mohammed S. Sayed & Mahmoud I Abdalla

  2. Department of Electronics and Communications Engineering, Egypt-Japan University of Science and Technology, New Borg El-Arab City, Alexandria, Egypt

    Mohammed S. Sayed

  3. IAEMS (International Academy for Engineering & Media Science), Giza, Egypt

    Mahmoud I Abdalla

  4. Department of Electronics and Communications Engineering, Cairo University, Cairo, Egypt

    Mohsen A. Rashwan

Authors
  1. Shayma’a A. Hassan
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Mohammed S. Sayed
    View author publications

    You can also search for this author in PubMed Google Scholar

  3. Mahmoud I Abdalla
    View author publications

    You can also search for this author in PubMed Google Scholar

  4. Mohsen A. Rashwan
    View author publications

    You can also search for this author in PubMed Google Scholar

Corresponding author

Correspondence to Shayma’a A. Hassan.

Ethics declarations

Conflict of interest

The 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

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Hassan, S.A., Sayed, M.S., Abdalla, M.I. et al. Breast cancer masses classification using deep convolutional neural networks and transfer learning. Multimed Tools Appl 79, 30735–30768 (2020). https://doi.org/10.1007/s11042-020-09518-w

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s11042-020-09518-w

Keywords

Search

Navigation