Generic placeholder image

Current Medical Imaging


ISSN (Print): 1573-4056
ISSN (Online): 1875-6603

Research Article

Multi-Class Breast Cancer Classification Using Ensemble of Pretrained models and Transfer Learning

Author(s): Perumalla Murali Mallikarjuna Rao, Sanjay Kumar Singh*, Aditya Khamparia, Bharat Bhushan and Prajoy Podder

Volume 18, Issue 4, 2022

Published on: 18 February, 2021

Article ID: e150322191535 Pages: 8

DOI: 10.2174/1573405617666210218101418

Price: $65


Aims: Early detection of breast cancer has reduced many deaths. Earlier CAD systems used to be the second opinion for radiologists and clinicians. Machine learning and deep learning have brought tremendous changes in medical diagnosis and imagining.

Background: Breast cancer is the most commonly occurring cancer in women and it is the second most common cancer overall. According to the 2018 statistics, there were over 2million cases all over the world. Belgium and Luxembourg have the highest rate of cancer.

Objective: A method for breast cancer detection has been proposed using Ensemble learning. 2- class and 8-class classification is performed.

Methods: To deal with imbalance classification, the authors have proposed an ensemble of pretrained models.

Results: 98.5% training accuracy and 89% of test accuracy are achieved on 8-class classification. Moreover, 99.1% and 98% train and test accuracy are achieved on 2 class classification.

Conclusion: it is found that there are high misclassifications in class DC when compared to the other classes, this is due to the imbalance in the dataset. In the future, one can increase the size of the datasets or use different methods. In implement this research work, authors have used 2 Nvidia Tesla V100 GPU’s in google cloud platform.

Keywords: Machine learning, deep learning, transfer learning, ensemble learning, resnet, mobilenet, densenet, pyTorch, breast cancer classification.

Graphical Abstract
”Breast cancer statistics | World Cancer Research Fund”. [Apr. 24, 2019].
”Breast Cancer: Statistics | CancerNet”. [Apr. 25, 2019].
”PyTorch” [Sep. 16, 2020]”.
Samala RK, Chan HP, Hadjiiski L, Helvie MA, Wei J, Cha K. Mass detection in digital breast tomosynthesis: Deep convolutional neural network with transfer learning from mammography. Med Phys 2016; 43(12): 6654-66.
[] [PMID: 27908154]
Dibble EH, Lourenco AP, Baird GL, Ward RC, Maynard AS, Mainiero MB. Comparison of digital mammography and digital breast tomosynthesis in the detection of architectural distortion. Eur Radiol 2018; 28(1): 3-10.
[] [PMID: 28710582]
Zhang X, Zhang Y, Han EY, et al. Classification of whole mammogram and tomosynthesis images using deep convolutional neural networks. IEEE Trans Nanobioscience 2018; 17(3): 237-42.
[] [PMID: 29994219]
Zhang X, Liu W, Dundar M, Badve S, Zhang S. Towards large-scale histopathological image analysis: hashing-based image retrieval. IEEE Trans Med Imaging 2015; 34(2): 496-506.
[] [PMID: 25314696]
Zhang X. Boosting twin support vector machine approach for MCs detection 2009; 1: 149-52.
Wang Z, et al. Breast Cancer Detection Using Extreme Learning Machine Based on Feature Fusion with CNN Deep Features. IEEE Access 2009; 4(c): 1-1.
Guo ZW, Suchkov M, Jiang L, Yan LZ. Breast Cancer Detection Via Wavelet Energy and Support Vector Machine RO-MAN 2018 - 27th IEEE. Int Symp Robot Hum Interact Commun 758-63.
Hagos YB, Narayanan PL, Akarca AU, Marafioti T, Yuan Y. ConCORDe-Net: cell count regularized convolutional neural network for cell detection in multiplex immunohistochemistry images 2019.
Hasan MK, Aleef TA. Automatic mass detection in breast using deep convolutional neural network and SVM classifier. 2019; pp. 1-11.
Gao F, Yoon H, Wu T, Chu X. A feature transfer enabled multi-task deep learning model on medical imaging. Expert Syst Appl 2020; 143: 112957.
Agrawal S, Rangnekar R, Gala D, Paul S, Kalbande D. Detection of breast cancer from mammograms using a hybrid approach of deep learning and linear classification. 2018; Int Conf Smart City Emerg Technol ICSCET 2018 1-6.
Simonyan K, Zisserman A. Very deep convolutional networks for large-scale image recognition. 2014; pp. 1-14.
Kassani SH, Kassani PH, Wesolowski MJ, Schneider KA, Deters R. Classification of histopathological biopsy images using ensemble of deep learning networks 2019.
Nguyen PT. Multiclass breast cancer classification using convolutional neural network. 2019 International Symposium on Electrical and Electronics Engineering (ISEE). Ho Chi Minh, Vietnam. 2019; pp. 130-4.
He K, Zhang X, Ren S, Sun J. Deep residual learning for image recognition. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). Las Vegas, NV, USA. 2016; pp. 770-8.
Sandler M, Howard A, Zhu M, Zhmoginov A, Chen LC. MobileNetV2: Inverted Residuals and Linear Bottlenecks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR) 2018; 4510-20.
Huang G, Liu Z, Van Der Maaten L, Weinberger KQ. Densely connected convolutional networks Proc - 30th IEEE Conf Comput Vis Pattern Recognition, CVPR 2017; 2261-9.
Gandomkar Z, Brennan PC, Mello-Thoms C. MuDeRN: Multi-category classification of breast histopathological image using deep residual networks. Artif Intell Med 2018; 88: 14-24.
[] [PMID: 29705552]
Spanhol FA, Oliveira LS, Petitjean C, Heutte L, et al. Breast cancer histopathological image classification using convolutional neural networks Int Jt Conf Neural Networks (IJCNN 2016)
Vo DM, Nguyen NQ, Lee SW. Classification of breast cancer histology images using incremental boosting convolution networks. Inf Sci (Ny) 2019; 482: 123-38.
Pratiher S, Chattoraj S. Diving Deep onto Discriminative Ensemble of Histological Hashing Class-Specific Manifold Learning for Multi-class Breast Carcinoma Taxonomy ICASSP. IEEE Int Conf Acoust Speech Signal Process 2019; 1025-9
Karthiga R, Narasimhan K. Automated diagnosis of breast cancer using wavelet based entropy. 2018; Second Int Conf Electron Commun Aerosp Technol 274-9.
“Cloud Computing Services | Google Cloud”. [Sep. 16, 2020].

Rights & Permissions Print Cite
© 2024 Bentham Science Publishers | Privacy Policy