Abstract
Brain tumor identification and classification is crucial in everyday life. This paper focuses on a four-class classification problem to differentiate between three prominent types of brain tumor namely glioma, meningioma, pituitary tumors and no tumor. The proposed system uses deep transfer learning and two pre-trained and one custom model to classify these brain MRI images. The empirical work is performed using a custom dataset made from existing public datasets. The proposed system registers a classification accuracy of up to 99%. Performance measures such as precision, recall and F-score have also been calculated. Moreover, since the dataset is not so big, the results show that transfer learning is a useful technique when the availability of medical images is limited.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Balasooriya, N.M., Nawarathna, R.D.: A sophisticated convolutional neural network model for brain tumor classification. In: 2017 IEEE International Conference on Industrial and Information Systems (ICIIS), pp. 1–5. IEEE (2017)
BraTS: Multimodal Brain Tumor Segmentation Challenge 2019, CBICA, Perelman School of Medicine at the University of Pennsylvania, https://www.med.upenn.edu/cbica/brats2019/data.html
Charron, O., et al.: Automatic detection and segmentation of brain metastases on multimodal MR images with a deep convolutional neural network. Comput. Biol. Med. 95, 43–54 (2018)
Cheng, J.: Brain tumor dataset. Figshare (2017)
Chollet, F.: Xception: Deep learning with depthwise separable convolutions. In: 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 1800–1807. IEEE (2017)
Deepak, S., Ameer, P.M.: Brain tumor classification using deep CNN features via transfer learning. Comput. Biol. Med. 111, 103345 (2019)
Esteva, A., et al.: Dermatologist-level classification of skin cancer with deep neural networks. Nature 542(7639), 115–118 (2017)
Gu, Y., et al.: Automatic lung nodule detection using a 3D deep convolutional neural network combined with a multi-scale prediction strategy in chest CTs. Comput. Biol. Med. 103, 220–231 (2018)
He, K. et al.: Deep residual learning for image recognition. In: IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 770–778. IEEE (2016)
Horie, Y., et al.: Diagnostic outcomes of esophageal cancer by artificial intelligence using convolutional neural networks. Gastrointest. Endosc. 89(1), 25–32 (2019)
Howard, A.G., et al.: MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications (2017)
Huang, G., et al.: Densely connected convolutional networks. In: 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 2261–2269. IEEE (2017)
Jeyaraj, P.R., Samuel Nadar, E.R.: Computer-assisted medical image classification for early diagnosis of oral cancer employing deep learning algorithm. J. Cancer Res. Clin. Oncol. 145(4), 829–837 (2019)
Krizhevsky, A., et al.: ImageNet Classification with Deep Convolutional Neural Networks (Semantic Scholar. Undefined) (2012)
Kutlu, H., Avcı, E.: A novel method for classifying liver and brain tumors using convolutional neural networks, discrete wavelet transform and long short-term memory networks. Sensors Basel Sensors 19, 9 (2019)
LeCun, Y., et al.: Deep learning. Nature 521(7553), 436–444 (2015)
Malone, I.B., et al.: MIRIAD—public release of a multiple time point Alzheimer’s MR imaging dataset. Neuroimage 70, 33–36 (2013)
Pattanayak, S.: Pro Deep Learning with TensorFlow: A Mathematical Approach to Advanced Artificial Intelligence in Python. Apress (2017)
Rawat, W., Wang, Z.: Deep convolutional neural networks for image classification: a comprehensive review. Neural Comput. 29(9), 2352–2449 (2017)
Sajjad, M., et al.: Multi-grade brain tumor classification using deep CNN with extensive data augmentation. J. Comput. Sci. 30, 174–182 (2019)
Salamon, J., Bello, J.P.: Deep convolutional neural networks and data augmentation for environmental sound classification. IEEE Signal Process. Lett. 24(3), 279–283 (2017)
Sandler, M. et al.: Mobilenetv2: inverted residuals and linear bottlenecks. In: 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 4510–4520. IEEE (2018)
Saxena, P. et al.: Predictive modeling of brain tumor: a deep learning approach (2020) (Unpublished)
Simonyan, K., Zisserman, A.: Very Deep Convolutional Networks for Large-Scale Image Recognition (2014)
Simonyan, K., Zisserman, A.: Very deep convolutional networks for large-scale image recognition. In: Presented at the 3rd International Conference on Learning Representations, ICLR (2015)
Swati, Z.N.K., et al.: Content-based brain tumor retrieval for MR images using transfer learning. IEEE Access. 7, 17809–17822 (2019)
Szegedy, C. et al.: Going deeper with convolutions. In: Proceedings of 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 1–9. IEEE (2015)
Szegedy, C. et al.: Rethinking the inception architecture for computer vision. In: Proceedings of 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 2818–2826. IEEE (2016)
Tan, C., et al.: A survey on deep transfer learning. In: Kůrková, V. (ed.) Artificial Neural Networks and Machine Learning—ICANN 2018: 27th International Conference on Artificial Neural Networks, Rhodes, Greece, Oct 4–7, 2018, Proceedings, Part III, pp. 270–279. Springer International Publishing, Cham (2018)
Toğaçar, M., et al.: BrainMRNet: brain tumor detection using magnetic resonance images with a novel convolutional neural network model. Med. Hypotheses. 134, 109531 (2020)
Yousefi, M., et al.: Mass detection in digital breast tomosynthesis data using convolutional neural networks and multiple instance learning. Comput. Biol. Med. 96, 283–293 (2018)
Zhou, L., et al.: A deep learning-based radiomics model for differentiating benign and malignant renal tumors. Transl. Oncol. 12(2), 292–300 (2019)
Access the Data—The Cancer Imaging Archive (TCIA), https://www.cancerimagingarchive.net/access-data/
Brain MRI Images for Brain Tumor Detection, Kaggle, https://www.kaggle.com/navoneel/brain-mri-images-for-brain-tumor-detection
Brain Tumor Information, National Brain Tumor Society, https://braintumor.org/brain-tumor-information/
IXI Dataset—Brain Development, https://brain-development.org/ixi-dataset/
Minimal Interval Resonance Imaging in Alzheimer’s Disease (MIRIAD), Dementia Research Centre—UCL—London’s Global University, https://www.ucl.ac.uk/drc/research/methods/minimal-interval-resonance-imaging-alzheimers-disease-miriad
NFBS Skull-Stripped Repository, https://preprocessed-connectomes-project.org/NFB_skullstripped/
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2021 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Bhagbut, G., Mungloo-Dilmohamud, Z. (2021). Classification of Brain Tumor MRIs Using Deep Learning and Data Augmentation. In: Panigrahi, C.R., Pati, B., Pattanayak, B.K., Amic, S., Li, KC. (eds) Progress in Advanced Computing and Intelligent Engineering. Advances in Intelligent Systems and Computing, vol 1299. Springer, Singapore. https://doi.org/10.1007/978-981-33-4299-6_6
Download citation
DOI: https://doi.org/10.1007/978-981-33-4299-6_6
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-33-4298-9
Online ISBN: 978-981-33-4299-6
eBook Packages: EngineeringEngineering (R0)