Abstract
Breast cancer has the second highest frequency of death rate among women worldwide. Early-stage prevention becomes complex due to reasons unknown. However, some typical signatures like masses and micro-calcifications upon investigating mammograms can help diagnose women better. Manual diagnosis is a hard task the radiologists carry out frequently. For their assistance, many computer-aided diagnosis (CADx) approaches have been developed. To improve upon the state of the art, we proposed a deep ensemble transfer learning and neural network classifier for automatic feature extraction and classification. In computer-assisted mammography, deep learning–based architectures are generally not trained on mammogram images directly. Instead, the images are pre-processed beforehand, and then they are adopted to be given as input to the ensemble model proposed. The robust features extracted from the ensemble model are optimized into a feature vector which are further classified using the neural network (nntraintool). The network was trained and tested to separate out benign and malignant tumors, thus achieving an accuracy of 0.88 with an area under curve (AUC) of 0.88. The attained results show that the proposed methodology is a promising and robust CADx system for breast cancer classification.
Similar content being viewed by others
References
Afshar P, Mohammadi A, Plataniotis KN, Oikonomou A, Benali H (2019) From handcrafted to deep-learning-based cancer radiomics: challenges and opportunities. IEEE Signal Proc Mag 36(4):132–160
Anderson PG, Sassaroli A, Kainerstorfer JM, Krishnamurthy N, Kalli S, Makim SS, Graham RA, Fantini S (2016) Optical mammography: bilateral breast symmetry in hemoglobin saturation maps. J Biomed Opt 21(10):101403
Antropova N, Huynh B, Giger M (2017) Performance comparison of deep learning and segmentation-based radiomic methods in the task of distinguishing benign and malignant breast lesions on dce-mri. In: Medical imaging 2017: computer-aided diagnosis, vol 10134. International Society for Optics and Photonics, p 101341g
Arevalo J, González FA, Ramos-Pollán R, Oliveira JL, Lopez MAG (2016) Representation learning for mammography mass lesion classification with convolutional neural networks. Comput Meth Prog Bio 127:248–257
Bengio Y (2013) Deep learning of representations: looking forward. In: International conference on statistical language and speech processing, Springer, pp 1–37
Bick U (2014) Mammography: how to interpret microcalcifications. In: Diseases of the abdomen and pelvis 2014–2017, Springer, pp 313–318
Bray F, Ferlay J, Soerjomataram I, Siegel RL, Torre LA, Jemal A (2018) Global cancer statistics 2018: Globocan estimates of incidence and mortality worldwide for 36 cancers in 185 countries. CA: A Cancer J Clin 68(6):394–424
Chaieb R, Kalti K (2019) Feature subset selection for classification of malignant and benign breast masses in digital mammography. Pattern Anal Applic 22(3):803–829
Chen H, Dou Q, Ni D, Cheng JZ, Qin J, Li S, Heng PA (2015) Automatic fetal ultrasound standard plane detection using knowledge transferred recurrent neural networks. In: International conference on medical image computing and computer-assisted intervention, Springer, pp 507–514
Demuth H, Beale M (2000) Neural network toolbox. In: Lewin RA (ed) The mathworks, Inc. 4th edn. Chapman and Hall, Natick, pp 1–840
Garnica C, Boochs F, Twardochlib M (2000) A new approach to edge-preserving smoothing for edge extraction and image segmentation. International Archives of Photogrammetry and Remote Sensing 33(B3/1; PART 3):320–325
Giger ML, Karssemeijer N, Schnabel JA (2013) Breast image analysis for risk assessment, detection, diagnosis, and treatment of cancer. Annu Rev Biomed Eng 15:327–357
Guo W, Wu R, Chen Y, Zhu X (2018) Deep learning scene recognition method based on localization enhancement. Sensors 18(10):3376
He K, Sun J, Tang X (2012) Guided image filtering. IEEE Trans Pattern Anal Mach Intell 35(6):1397–1409
He K, Zhang X, Ren S, Sun J (2016) Deep residual learning for image recognition. In: Proceedings of the IEEE conference on computer vision and pattern recognition, pp 770–778
Jamieson AR, Drukker K, Giger ML (2012) Breast image feature learning with adaptive deconvolutional networks. In: Medical imaging 2012: computer-aided diagnosis, vol 8315. International Society for Optics and Photonics, p 831506
Jenifer S, Parasuraman S, Kadirvelu A (2016) Contrast enhancement and brightness preserving of digital mammograms using fuzzy clipped contrast-limited adaptive histogram equalization algorithm. Appl Soft Comput 42:167–177
Jothilakshmi G, Sharmila P, Raaza A (2016) Mammogram segmentation using region based method with split and merge technique. Indian Journal of Science and Technology 9(40)
Krizhevsky A, Sutskever I, Hinton GE (2012) Imagenet classification with deep convolutional neural networks. In: Advances in neural information processing systems, pp 1097–1105
Lau TK, Bischof WF (1991) Automated detection of breast tumors using the asymmetry approach. Comput Biomed Res 24(3):273–295
LeCun Y, Bengio Y, Hinton G (2015) Deep learning. Nature 521(7553):436
Mahersia H, Boulehmi H, Hamrouni K (2016) Development of intelligent systems based on bayesian regularization network and neuro-fuzzy models for mass detection in mammograms: a comparative analysis. Comput Meth Prog Bio 126:46–62
Maier A, Syben C, Lasser T, Riess C (2019) A gentle introduction to deep learning in medical image processing. Zeitschrift für Medizinische Physik 29(2):86–101
Mavroforakis ME, Georgiou HV, Dimitropoulos N, Cavouras D, Theodoridis S (2006) Mammographic masses characterization based on localized texture and dataset fractal analysis using linear, neural and support vector machine classifiers. Artif Intell Med 37(2):145–162
McCart Reed AE, Kalita-De Croft P, Kutasovic JR, Saunus JM, Lakhani SR (2019) Recent advances in breast cancer research impacting clinical diagnostic practice. J Pathol 247(5):552– 562
McGuire A, Brown J, Malone C, McLaughlin R, Kerin M (2015) Effects of age on the detection and management of breast cancer. Cancers 7(2):908–929
Oliver A, Freixenet J, Marti J, Perez E, Pont J, Denton ER, Zwiggelaar R (2010) A review of automatic mass detection and segmentation in mammographic images. Med Image Anal 14(2):87–110
Peng W, Mayorga RV, Hussein EM (2016) An automated confirmatory system for analysis of mammograms. Comput Meth Prog Bio 125:134–144
Rampun A, Scotney BW, Morrow PJ, Wang H (2018) Breast mass classification in mammograms using ensemble convolutional neural networks. In: 2018 IEEE 20Th international conference on e-health networking, applications and services healthcom, IEEE, pp 1–6
Rangayyan RM, El-Faramawy NM, Desautels JL, Alim OA (1997) Measures of acutance and shape for classification of breast tumors. IEEE Trans Med Imaging 16(6):799–810
Rebecca L, Siegel KDM, Jemal A (2018) Cancer statistics, 2018. CA: A Cancer J Clin 68(1):7–30
Rojas-Domínguez A, Nandi AK (2009) Development of tolerant features for characterization of masses in mammograms. Comput Biol Med 39(8):678–688
Samulski M, Hupse R, Boetes C, Mus RD, den Heeten GJ, Karssemeijer N (2010) Using computer-aided detection in mammography as a decision support. Eur Radiol 20(10):2323–2330
Simonyan K, Zisserman A (2014) Very Deep Convolutional Networks for Large-scale image Recognition. arXiv:1409.1556
Suk HI, Lee SW, Shen D, Initiative ADN, et al. (2015) Latent feature representation with stacked auto-encoder for ad/mci diagnosis. Brain Struct Funct 220(2):841–859
Sun W, Tseng TLB, Zhang J, Qian W (2017) Enhancing deep convolutional neural network scheme for breast cancer diagnosis with unlabeled data. Comput Med Imaging Graph 57:4–9
Szegedy C, Ioffe S, Vanhoucke V, Alemi AA (2017) Inception-v4, inception-resnet and the impact of residual connections on learning. In: Thirty-first AAAI Conference on Artificial Intelligence
Szegedy C, Liu W, Jia Y, Sermanet P, Reed S, Anguelov D, Erhan D, Vanhoucke V, Rabinovich A (2015) Going deeper with convolutions. In: Proceedings of the IEEE conference on computer vision and pattern recognition, pp 1–9
Tang J, Rangayyan RM, Xu J, El Naqa I, Yang Y (2009) Computer-aided detection and diagnosis of breast cancer with mammography: recent advances. IEEE Transactions on Information Technology in Biomedicine 13(2):236–251
Tsochatzidis L, Costaridou L, Pratikakis I (2019) Deep learning for breast cancer diagnosis from mammograms—a comparative study. J Imaging 5(3):37
Tsochatzidis L, Zagoris K, Arikidis N, Karahaliou A, Costaridou L, Pratikakis I (2017) Computer-aided diagnosis of mammographic masses based on a supervised content-based image retrieval approach. Pattern Recogn 71:106–117
Wang D, Khosla A, Gargeya R, Irshad H, Beck AH (2016) Deep learning for identifying metastatic breast cancer. arXiv:1606.05718
Yi D, Sawyer RL, Cohn III D, Dunnmon J, Lam C, Xiao X, Rubin D (2017) Optimizing and visualizing deep learning for benign/malignant classification in breast tumors. arXiv:1705.06362
Young T, Hazarika D, Poria S, Cambria E (2018) Recent trends in deep learning based natural language processing. IEEE Comput Intell Mag 13(3):55–75
Zhang W, Li R, Deng H, Wang L, Lin W, Ji S, Shen D (2015) Deep convolutional neural networks for multi-modality isointense infant brain image segmentation. NeuroImage 108:214–224
Zhu F, Liang Z, Jia X, Zhang L, Yu Y (2019) A benchmark for edge-preserving image smoothing. IEEE Trans Image Process 28(7):3556–3570
Acknowledgments
The authors would like to extend their thanks to the Indian Institute of Technology Roorkee, Uttarakhand, INDIA for their continuous support.
Funding
This research is being funded by the Ministry of Human Resource Development (MHRD), Government of India, INDIA (grant number OH-31-23-200-428).
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interests
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.
Electronic supplementary material
Rights and permissions
About this article
Cite this article
Arora, R., Rai, P.K. & Raman, B. Deep feature–based automatic classification of mammograms. Med Biol Eng Comput 58, 1199–1211 (2020). https://doi.org/10.1007/s11517-020-02150-8
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11517-020-02150-8