MRI-Based Brain Tumor Classification Using Ensemble of Deep Features and Machine Learning Classifiers.

Jaeyong Kang, Zahid Ullah, Jeonghwan Gwak
Author Information
  1. Jaeyong Kang: Department of Software, Korea National University of Transportation, Chungju 27469, Korea. ORCID
  2. Zahid Ullah: Department of Software, Korea National University of Transportation, Chungju 27469, Korea.
  3. Jeonghwan Gwak: Department of Software, Korea National University of Transportation, Chungju 27469, Korea.

Abstract

Brain tumor classification plays an important role in clinical diagnosis and effective treatment. In this work, we propose a method for brain tumor classification using an ensemble of deep features and machine learning classifiers. In our proposed framework, we adopt the concept of transfer learning and uses several pre-trained deep convolutional neural networks to extract deep features from brain magnetic resonance (MR) images. The extracted deep features are then evaluated by several machine learning classifiers. The top three deep features which perform well on several machine learning classifiers are selected and concatenated as an ensemble of deep features which is then fed into several machine learning classifiers to predict the final output. To evaluate the different kinds of pre-trained models as a deep feature extractor, machine learning classifiers, and the effectiveness of an ensemble of deep feature for brain tumor classification, we use three different brain magnetic resonance imaging (MRI) datasets that are openly accessible from the web. Experimental results demonstrate that an ensemble of deep features can help improving performance significantly, and in most cases, support vector machine (SVM) with radial basis function (RBF) kernel outperforms other machine learning classifiers, especially for large datasets.

Keywords

References

  1. Cancers (Basel). 2019 Jan 18;11(1): [PMID: 30669406]
  2. Comput Biol Med. 2019 Aug;111:103345 [PMID: 31279167]
  3. Sci Rep. 2019 Apr 23;9(1):6381 [PMID: 31011155]
  4. IEEE Trans Med Imaging. 2016 May;35(5):1299-1312 [PMID: 26978662]
  5. Med Hypotheses. 2020 Jun;139:109684 [PMID: 32240877]
  6. Healthcare (Basel). 2021 Feb 02;9(2): [PMID: 33540873]
  7. Med Hypotheses. 2020 Jun;139:109696 [PMID: 32234609]
  8. Med Image Anal. 2017 Jan;35:18-31 [PMID: 27310171]
  9. IEEE Trans Med Imaging. 2011 Mar;30(3):733-46 [PMID: 21118769]
  10. Med Hypotheses. 2020 Oct;143:109922 [PMID: 32682214]
  11. Acta Neuropathol. 2016 Jun;131(6):803-20 [PMID: 27157931]
  12. Brain Inform. 2018 Mar;5(1):23-30 [PMID: 29313301]
  13. J Med Syst. 2019 Jul 12;43(8):282 [PMID: 31300899]
  14. IEEE J Biomed Health Inform. 2017 Jan;21(1):76-84 [PMID: 28114048]
  15. Neuroimage. 2016 Apr 1;129:460-469 [PMID: 26808333]
  16. Acad Radiol. 2003 Dec;10(12):1341-8 [PMID: 14697002]
  17. IEEE Trans Med Imaging. 2016 May;35(5):1240-1251 [PMID: 26960222]
  18. Math Biosci Eng. 2020 Sep 15;17(5):6203-6216 [PMID: 33120595]
  19. Int J Comput Assist Radiol Surg. 2012 Jul;7(4):493-506 [PMID: 21833491]
  20. J Digit Imaging. 2012 Dec;25(6):708-19 [PMID: 22692772]
  21. PLoS One. 2016 Jun 06;11(6):e0157112 [PMID: 27273091]

Grants

  1. NRF-2019M3C7A1020406; NRF-2020R1I1A3074141/National Research Foundation of Korea

MeSH Term

Brain Neoplasms
Humans
Machine Learning
Magnetic Resonance Imaging
Neural Networks, Computer
Support Vector Machine