Skip to main content

Classification of Brain Tumor MRIs Using Deep Learning and Data Augmentation

  • Conference paper
  • First Online:
Progress in Advanced Computing and Intelligent Engineering

Part of the book series: Advances in Intelligent Systems and Computing ((AISC,volume 1299))

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.

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

Access this chapter

Chapter
USD 29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever
eBook
USD 169.00
Price excludes VAT (USA)
  • Available as EPUB and PDF
  • Read on any device
  • Instant download
  • Own it forever
Softcover Book
USD 219.99
Price excludes VAT (USA)
  • Compact, lightweight edition
  • Dispatched in 3 to 5 business days
  • Free shipping worldwide - see info

Tax calculation will be finalised at checkout

Purchases are for personal use only

Institutional subscriptions

Similar content being viewed by others

References

  1. 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)

    Google Scholar 

  2. 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

  3. 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)

    Article  Google Scholar 

  4. Cheng, J.: Brain tumor dataset. Figshare (2017)

    Google Scholar 

  5. 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)

    Google Scholar 

  6. Deepak, S., Ameer, P.M.: Brain tumor classification using deep CNN features via transfer learning. Comput. Biol. Med. 111, 103345 (2019)

    Article  Google Scholar 

  7. Esteva, A., et al.: Dermatologist-level classification of skin cancer with deep neural networks. Nature 542(7639), 115–118 (2017)

    Article  Google Scholar 

  8. 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)

    Article  Google Scholar 

  9. 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)

    Google Scholar 

  10. Horie, Y., et al.: Diagnostic outcomes of esophageal cancer by artificial intelligence using convolutional neural networks. Gastrointest. Endosc. 89(1), 25–32 (2019)

    Article  Google Scholar 

  11. Howard, A.G., et al.: MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications (2017)

    Google Scholar 

  12. Huang, G., et al.: Densely connected convolutional networks. In: 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 2261–2269. IEEE (2017)

    Google Scholar 

  13. 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)

    Article  Google Scholar 

  14. Krizhevsky, A., et al.: ImageNet Classification with Deep Convolutional Neural Networks (Semantic Scholar. Undefined) (2012)

    Google Scholar 

  15. 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)

    Google Scholar 

  16. LeCun, Y., et al.: Deep learning. Nature 521(7553), 436–444 (2015)

    Article  MathSciNet  Google Scholar 

  17. Malone, I.B., et al.: MIRIAD—public release of a multiple time point Alzheimer’s MR imaging dataset. Neuroimage 70, 33–36 (2013)

    Article  Google Scholar 

  18. Pattanayak, S.: Pro Deep Learning with TensorFlow: A Mathematical Approach to Advanced Artificial Intelligence in Python. Apress (2017)

    Google Scholar 

  19. Rawat, W., Wang, Z.: Deep convolutional neural networks for image classification: a comprehensive review. Neural Comput. 29(9), 2352–2449 (2017)

    Article  MathSciNet  Google Scholar 

  20. Sajjad, M., et al.: Multi-grade brain tumor classification using deep CNN with extensive data augmentation. J. Comput. Sci. 30, 174–182 (2019)

    Article  Google Scholar 

  21. 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)

    Article  Google Scholar 

  22. 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)

    Google Scholar 

  23. Saxena, P. et al.: Predictive modeling of brain tumor: a deep learning approach (2020) (Unpublished)

    Google Scholar 

  24. Simonyan, K., Zisserman, A.: Very Deep Convolutional Networks for Large-Scale Image Recognition (2014)

    Google Scholar 

  25. 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)

    Google Scholar 

  26. Swati, Z.N.K., et al.: Content-based brain tumor retrieval for MR images using transfer learning. IEEE Access. 7, 17809–17822 (2019)

    Article  Google Scholar 

  27. 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)

    Google Scholar 

  28. 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)

    Google Scholar 

  29. 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)

    Chapter  Google Scholar 

  30. 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)

    Article  Google Scholar 

  31. 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)

    Article  Google Scholar 

  32. Zhou, L., et al.: A deep learning-based radiomics model for differentiating benign and malignant renal tumors. Transl. Oncol. 12(2), 292–300 (2019)

    Article  Google Scholar 

  33. Access the Data—The Cancer Imaging Archive (TCIA), https://www.cancerimagingarchive.net/access-data/

  34. Brain MRI Images for Brain Tumor Detection, Kaggle, https://www.kaggle.com/navoneel/brain-mri-images-for-brain-tumor-detection

  35. Brain Tumor Information, National Brain Tumor Society, https://braintumor.org/brain-tumor-information/

  36. IXI Dataset—Brain Development, https://brain-development.org/ixi-dataset/

  37. 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

  38. NFBS Skull-Stripped Repository, https://preprocessed-connectomes-project.org/NFB_skullstripped/

Download references

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Zahra Mungloo-Dilmohamud .

Editor information

Editors and Affiliations

Rights and permissions

Reprints and permissions

Copyright information

© 2021 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.

About this paper

Check for updates. Verify currency and authenticity via CrossMark

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)

Publish with us

Policies and ethics