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

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.

[1]  Mohamed Medhat Gaber,et al.  DeTrac: Transfer Learning of Class Decomposed Medical Images in Convolutional Neural Networks , 2020, IEEE Access.

[2]  István Csabai,et al.  Detecting and classifying lesions in mammograms with Deep Learning , 2017, Scientific Reports.

[3]  Donghong Ji,et al.  DE-Ada*: A novel model for breast mass classification using cross-modal pathological semantic mining and organic integration of multi-feature fusions , 2020, Inf. Sci..

[4]  Tae-Seong Kim,et al.  Evaluation of deep learning detection and classification towards computer-aided diagnosis of breast lesions in digital X-ray mammograms , 2020, Comput. Methods Programs Biomed..

[5]  J. Dinesh Peter,et al.  Classification of Mammogram Images Using Multiscale all Convolutional Neural Network (MA-CNN) , 2019, Journal of Medical Systems.

[6]  Balasubramanian Raman,et al.  Deep feature–based automatic classification of mammograms , 2020, Medical & Biological Engineering & Computing.

[7]  José García Rodríguez,et al.  A Review on Deep Learning Techniques Applied to Semantic Segmentation , 2017, ArXiv.

[8]  Li Shen,et al.  Deep Learning to Improve Breast Cancer Detection on Screening Mammography , 2017, Scientific Reports.

[9]  U. Michelucci Advanced Applied Deep Learning: Convolutional Neural Networks and Object Detection , 2019 .

[10]  David R. Dance,et al.  Mammographic Image Analysis Society (MIAS) database v1.21 , 2015 .

[11]  Zhang Yi,et al.  Deep Neural Networks With Region-Based Pooling Structures for Mammographic Image Classification , 2020, IEEE Transactions on Medical Imaging.

[12]  Jaime S. Cardoso,et al.  INbreast: toward a full-field digital mammographic database. , 2012, Academic radiology.

[13]  Bram van Ginneken,et al.  A survey on deep learning in medical image analysis , 2017, Medical Image Anal..

[14]  Tae-Seong Kim,et al.  A fully integrated computer-aided diagnosis system for digital X-ray mammograms via deep learning detection, segmentation, and classification , 2018, Int. J. Medical Informatics.

[15]  Ning Qian,et al.  On the momentum term in gradient descent learning algorithms , 1999, Neural Networks.

[16]  Edwin Valarezo,et al.  Simultaneous Detection and Classification of Breast Masses in Digital Mammograms via a Deep Learning YOLO-based CAD System , 2018, Comput. Methods Programs Biomed..

[17]  Mohammed Bennamoun,et al.  A Guide to Convolutional Neural Networks for Computer Vision , 2018, A Guide to Convolutional Neural Networks for Computer Vision.

[18]  Sheida Nabavi,et al.  Deep convolutional neural networks for mammography: advances, challenges and applications , 2019, BMC Bioinformatics.

[19]  Xavier Lladó,et al.  Automatic mass detection in mammograms using deep convolutional neural networks , 2019, Journal of medical imaging.

[20]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[21]  Mohammed S. Sayed,et al.  Detection of breast cancer mass using MSER detector and features matching , 2019, Multimedia Tools and Applications.

[22]  Tianfu Wang,et al.  Breast Cancer Detection and Diagnosis Using Mammographic Data: Systematic Review , 2019, Journal of medical Internet research.

[23]  Stephen Marshall,et al.  Breast cancer detection using deep convolutional neural networks and support vector machines , 2019, PeerJ.

[24]  Tae-Seong Kim,et al.  Deep Learning Computer-Aided Diagnosis for Breast Lesion in Digital Mammogram. , 2020, Advances in experimental medicine and biology.

[25]  Qaisar Abbas,et al.  DeepCAD: A Computer-Aided Diagnosis System for Mammographic Masses Using Deep Invariant Features , 2016, Comput..

[26]  Fakhri Alam Khan,et al.  Computer-aided diagnosis for burnt skin images using deep convolutional neural network , 2020, Multimedia Tools and Applications.

[27]  Muhammad Imran Razzak,et al.  A Deep Learning-Based Framework for Automatic Brain Tumors Classification Using Transfer Learning , 2019, Circuits, Systems, and Signal Processing.

[28]  Taranjit Kaur,et al.  Deep convolutional neural networks with transfer learning for automated brain image classification , 2020, Machine Vision and Applications.

[29]  Shohreh Kasaei,et al.  Benign and malignant breast tumors classification based on region growing and CNN segmentation , 2015, Expert Syst. Appl..

[30]  J. Hornaday,et al.  Cancer Facts & Figures 2004 , 2004 .

[31]  Yasser M. Kadah,et al.  An Automatic Computer-Aided Diagnosis System for Breast Cancer in Digital Mammograms via Deep Belief Network , 2018 .

[32]  Dumitru Erhan,et al.  Going deeper with convolutions , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[33]  Gustavo Carneiro,et al.  A deep learning approach for the analysis of masses in mammograms with minimal user intervention , 2017, Medical Image Anal..

[34]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[35]  Richard H. Moore,et al.  THE DIGITAL DATABASE FOR SCREENING MAMMOGRAPHY , 2007 .