Skip to main content

Advertisement

SpringerLink
Log in

Automated Categorization of Brain Tumor from MRI Using CNN features and SVM

  • Original Research
  • Published:
Journal of Ambient Intelligence and Humanized Computing Aims and scope Submit manuscript

Abstract

Automated tumor characterization has a prominent role in the computer-aided diagnosis (CAD) system for the human brain. Despite being a well-studied topic, CAD of brain tumors poses severe challenges in some specific aspects. One such challenging problem is the category-based classification of brain tumors among glioma, meningioma, and pituitary tumors using magnetic resonance imaging (MRI) images. The emergence of deep learning and machine learning algorithms have addressed image classification tasks with promising results. But an associated limitation with the medical image classification is the small sizes of medical image databases. This limitation, in turn, limits the availability of medical images for training deep neural networks. To mitigate this challenge, we adopt a combination of convolutional neural network (CNN) features with support vector machine (SVM) for classification of the medical images. The fully automated system is evaluated using Figshare open dataset containing MRI images for the three types of brain tumors. CNN is designed to extract features from brain MRI images. For enhanced performance, a multiclass SVM is used with CNN features. Testing and evaluation of the integrated system followed a fivefold cross-validation procedure. The proposed model attained an overall classification accuracy of 95.82%, better than the state-of-the-art method. Extensive experiments are performed on other MRI datasets for the brain to ascertain the improved performance of the proposed system. When the amount of available training data is small, the SVM classifier is observed to perform better than the softmax classifier for the CNN features. Compared to transfer learning-based classification, the adopted strategy of CNN-SVM has lesser computations and memory requirements.

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

Access this article

Log in via an institution

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7

Similar content being viewed by others

References

  • Abbasi S, Tajeripour F (2017) Detection of brain tumor in 3D MRI images using local binary patterns and histogram orientation gradient. Neurocomputing 219:526–535

    Article  Google Scholar 

  • Abiwinanda N, Hanif M, Hesaputra ST, Handayani A, Mengko, TR (2019). Brain tumor classification using convolutional neural network. In: World Congress on Medical Physics and Biomedical Engineering 2018, Springer, Singapore, pp 183-189

  • Afshar P, Mohammadi A, Plataniotis KN (2018) Brain tumor type classification via capsule networks. In: 2018 25th IEEE International Conference on Image Processing (ICIP), IEEE, pp 3129-3133

  • Afshar P, Plataniotis KN, Mohammadi A (2019) Capsule networks for brain tumor classification based on mri images and coarse tumor boundaries. In: ICASSP 2019–2019 IEEE international conference on acoustics, speech and signal processing (ICASSP), IEEE, pp 1368–1372

  • Agarap AF (2017) An architecture combining convolutional neural network (CNN) and support vector machine (SVM) for image classification. arXiv preprint arXiv:1712.03541

  • Agarwal R, Diaz O, Lladó X, Martí R (2018) Mass detection in mammograms using pre-trained deep learning models. In: 14th International workshop on breast imaging (IWBI 2018), vol 10718, International Society for Optics and Photonics, p 107181F

  • Agarwal P, Wang HC, Srinivasan K (2018) Epileptic seizure prediction over EEG data using hybrid CNN-SVM model with edge computing services. In: MATEC Web of Conferences, vol 210, EDP Sciences, p 03016

  • Amin J, Sharif M, Yasmin M, Fernandes SL (2017) A distinctive approach in brain tumor detection and classification using MRI. Pattern Recogn Lett. https://doi.org/10.1016/j.patrec.2017.10.036

    Article  Google Scholar 

  • Amin J, Sharif M, Raza M, Yasmin M (2018) Detection of brain tumor based on features fusion and machine learning. J Ambient Intell Humaniz Comput. https://doi.org/10.1007/s12652-018-1092-9

    Article  Google Scholar 

  • Anthimopoulos M, Christodoulidis S, Ebner L, Christe A, Mougiakakou S (2016) Lung pattern classification for interstitial lung diseases using a deep convolutional neural network. IEEE Trans Med Imaging 35(5):1207–1216

    Article  Google Scholar 

  • Bardou D, Zhang K, Ahmad SM (2018) Classification of breast cancer based on histology images using convolutional neural networks. IEEE Access 6:24680–24693

    Article  Google Scholar 

  • Cao Y, Geddes TA, Yang JYH, Yang P (2020) Ensemble deep learning in bioinformatics. Nat Mach Intell 2:1–9

    Article  Google Scholar 

  • Cheng J, Huang W, Cao S, Yang R, Yang W, Yun Z, Feng Q (2015) Enhanced performance of brain tumor classification via tumor region augmentation and partition. PloS One 10:e014449

    Google Scholar 

  • Cheng J, Yang W, Huang M, Huang W, Jiang J, Zhou Y, Chen W (2016) Retrieval of brain tumors by adaptive spatial pooling and fisher vector representation. PloS One 11:e0157112

    Article  Google Scholar 

  • Cortes C, Vapni V (1995) Support-vector networks. Mach Learn 20(3):273–297

    Google Scholar 

  • Dataset 1 (2018) Figshare brain tumor dataset’. https://do.org/10.6084/m9.figshare.1512427.v5. Accessed Dec 2018

  • Dataset 2 (2019) Radiopaedia dataset. https://radiopaedia.org. Accessed Nov 2019

  • Dataset 3 (2019) Harvard medical dataset. http://med.harvard.edu/AANLIB. Accessed Nov 2019

  • Dong H, Yang G, Liu F, Mo Y, Guo Y (2017). Automatic brain tumor detection and segmentation using u-net based fully convolutional networks. In: Annual conference on medical image understanding and analysis, Springer, Cham, pp 506–517

  • Fang T (2018) A novel computer-aided lung cancer detection method based on transfer learning from GoogLeNet and median intensity projections. In: 2018 IEEE international conference on computer and communication engineering technology (CCET), IEEE, pp 286–290

  • Huang S, Yang J, Fong S, Zhao Q (2019) Mining prognosis index of brain metastases using artificial intelligence. Cancers 11(8):1140

    Article  Google Scholar 

  • Ismael MR, Abdel-Qader I (2018) Brain tumor classification via statistical features and back-propagation neural network. In: 2018 IEEE international conference on electro/information technology (EIT), USA, pp 252–257

  • Kaur T, Saini BS, Gupta S (2017) Quantitative metric for MR brain tumour grade classification using sample space density measure of analytic intrinsic mode function representation. IET Image Process 11(8):620–632

    Article  Google Scholar 

  • Kumar S, Dabas C, Godara S (2017) Classification of brain MRI tumor images: a hybrid approach. Procedia Comput Sci 122:510–517

    Article  Google Scholar 

  • Liu B, Yu X, Zhang P, Yu A, Fu Q, Wei X (2017) Supervised deep feature extraction for hyperspectral image classification. IEEE Trans Geosci Remote Sens 56(4):1909–1921

    Article  Google Scholar 

  • Lu S, Lu Z, Zhang YD (2019) Pathological brain detection based on AlexNet and transfer learning. J Comput Sci 30:41–47

    Article  Google Scholar 

  • Mahesh KM, Renjit JA (2018) Evolutionary intelligence for brain tumor recognition from MRI images: a critical study and review. Evol Intell 11(1–2):19–30

    Article  Google Scholar 

  • Mohan G, Subashini MM (2018) MRI based medical image analysis: Survey on brain tumor grade classification. Biomed Signal Process Control 39:139–161

    Article  Google Scholar 

  • Mohsen H, El-Dahshan ESA, El-Horbaty ESM, Salem ABM (2018) Classification using deep learning neural networks for brain tumors. Future Comput Inform J 3(1):68–71

    Article  Google Scholar 

  • Mzoughi H, Njeh I, Wali A, Slima MB, BenHamida A, Mhiri C, Mahfoudhe KB (2020) Deep multi-scale 3D convolutional neural network (CNN) for MRI gliomas brain tumor classification. J Digit Imaging. https://doi.org/10.1007/s10278-020-00347-9

    Article  Google Scholar 

  • Nabizadeh N, Kubat M (2015) Brain tumors detection and segmentation in MR images: Gabor wavelet vs. statistical features. Comput Electr Eng 45:286–301

    Article  Google Scholar 

  • Nagabushanam P, George ST, Radha S (2019) EEG signal classification using LSTM and improved neural network algorithms. Soft Comput 24:1–23

    Google Scholar 

  • Nanthagopal AP, Sukanesh R (2013) Wavelet statistical texture features-based segmentation and classification of brain computed tomography images. IET Image Process 7(1):25–32

    Article  MathSciNet  Google Scholar 

  • Pashaei A, Sajedi H, Jazayeri N (2018) Brain tumor classification via convolutional neural network and extreme learning machines. In: 2018 8th international conference on computer and knowledge engineering (ICCKE), Mashhad, pp 314–319

  • Paul J (2016) Deep Learning for brain tumor classification. PhD diss, Vanderbilt University

  • Rajagopal R (2019) Glioma brain tumor detection and segmentation using weighting random forest classifier with optimized ant colony features. Int J Imaging Syst Technol 29(3):353–359

    Article  Google Scholar 

  • Rocha A, Goldenstein SK (2013) Multiclass from binary: expanding one-versus-all, one-versus-one and ECOC-based approaches. IEEE Trans Neural Netw Learn Syst 25(2):289–302

    Article  Google Scholar 

  • Sampathkumar A, Vivekanandan P (2019) Gene selection using parallel lion optimization method in microarray data for cancer classification. J Med Imaging Health Inform 9(6):1294–1300

    Article  Google Scholar 

  • Sampathkumar A, Rastogi R, Arukonda S, Shankar A, Kautish S, Sivaram M (2020) An efficient hybrid methodology for detection of cancer-causing gene using CSC for micro array data. J Ambient Intell Human Comput. https://doi.org/10.1007/s12652-020-01731-7

    Article  Google Scholar 

  • Seetha J, Raja SS (2018) Brain tumor classification using convolutional neural networks. Biomed Pharmacol J 11(3):1457

    Article  Google Scholar 

  • Swati ZNK, Zhao Q, Kabir M, Ali F, Ali Z, Ahmed S, Lu J (2019) Brain tumor classification for MR images using transfer learning and fine-tuning. Comput Med Imaging Graph 75:34–46

    Article  Google Scholar 

  • Sze V, Chen YH, Yang TJ, Emer JS (2017) Efficient processing of deep neural networks: A tutorial and survey. Proc IEEE 105(12):2295–2329

    Article  Google Scholar 

  • Tavakoli N, Karimi M, Norouzi A, Karimi N, Samavi S, Soroushmehr SR (2019) Detection of abnormalities in mammograms using deep features. J Ambient Intell Human Comput. https://doi.org/10.1007/s12652-019-01639-x

    Article  Google Scholar 

  • Xue DX, Zhang R, Feng H, Wang YL (2016) CNN-SVM for microvascular morphological type recognition with data augmentation. J Med Biol Eng 36(6):755–764

    Article  Google Scholar 

  • Zhou Y, Li Z, Zhu H, Chen C, Gao M, Xu K, Xu J (2018). Holistic brain tumor screening and classification based on densenet and recurrent neural network. In: International MICCAI Brainlesion workshop, Springer, Cham, pp 208–217

Download references

Author information

Authors and Affiliations

  1. National Institute of Technology, Calicut, India

    S. Deepak & P. M. Ameer

Authors
  1. S. Deepak
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. P. M. Ameer
    View author publications

    You can also search for this author in PubMed Google Scholar

Corresponding author

Correspondence to S. Deepak.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Deepak, S., Ameer, P.M. Automated Categorization of Brain Tumor from MRI Using CNN features and SVM. J Ambient Intell Human Comput 12, 8357–8369 (2021). https://doi.org/10.1007/s12652-020-02568-w

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s12652-020-02568-w

Keywords

Search

Navigation