Abstract
Objective
Despite the critical role of Magnetic Resonance Imaging (MRI) in the diagnosis of brain tumours, there are still many pitfalls in the exact grading of them, in particular, gliomas. In this regard, it was aimed to examine the potential of Transfer Learning (TL) and Machine Learning (ML) algorithms in the accurate grading of gliomas on MRI images.
Materials and methods
Dataset has included four types of axial MRI images of glioma brain tumours with grades I–IV: T1-weighted, T2-weighted, FLAIR, and T1-weighted Contrast-Enhanced (T1-CE). Images were resized, normalized, and randomly split into training, validation, and test sets. ImageNet pre-trained Convolutional Neural Networks (CNNs) were utilized for feature extraction and classification, using Adam and SGD optimizers. Logistic Regression (LR) and Support Vector Machine (SVM) methods were also implemented for classification instead of Fully Connected (FC) layers taking advantage of features extracted by each CNN.
Results
Evaluation metrics were computed to find the model with the best performance, and the highest overall accuracy of 99.38% was achieved for the model containing an SVM classifier and features extracted by pre-trained VGG-16.
Discussion
It was demonstrated that developing Computer-aided Diagnosis (CAD) systems using pre-trained CNNs and classification algorithms is a functional approach to automatically specify the grade of glioma brain tumours in MRI images. Using these models is an excellent alternative to invasive methods and helps doctors diagnose more accurately before treatment.
Similar content being viewed by others
References
Britannica, T.E.o.E. neuroglia; Available from: https://www.britannica.com/science/neuroglia. Accessed 4 Feb 2022
Britannica, T.E.o.E. glioma; Available from: https://www.britannica.com/science/glioma. Accessed 4 Feb 2022
Recht L (2019) Brain and spinal cord tumors. Cancer: prevention, early detection, treatment and recovery, pp 395–414
Goodenberger ML, Jenkins RB (2012) Genetics of adult glioma. Cancer Genet 205(12):613–621
Louis DN et al (2016) The 2016 World Health Organization classification of tumors of the central nervous system: a summary. Acta Neuropathol 131(6):803–820
Dequidt P et al (2021) Exploring radiologic criteria for glioma grade classification on the BraTS dataset. IRBM 42(6):407–414
Sun P et al (2019) Comparison of feature selection methods and machine learning classifiers for radiomics analysis in glioma grading. IEEE Access 7:102010–102020
Wen PY, Huse JT (2017) 2016 World Health Organization classification of central nervous system tumors. CONTINUUM: Lifelong Learn Neurol 23(6):1531–1547
Malone H et al (2015) Complications following stereotactic needle biopsy of intracranial tumors. World Neurosurg 84(4):1084–1089
Copeland BJ (2022) Artificial intelligence; Available from: https://www.britannica.com/technology/artificial-intelligence. Accessed 4 Feb 2022
Esteva A et al (2021) Deep learning-enabled medical computer vision. NPJ digital medicine 4(1):1–9
Géron A (2019) Hands-on machine learning with Scikit-Learn, Keras, and TensorFlow: concepts, tools, and techniques to build intelligent systems. O'Reilly Media, Inc
Alzubaidi L et al (2021) Review of deep learning: concepts, CNN architectures, challenges, applications, future directions. J Big Data 8(1):1–74
Yamashita R et al (2018) Convolutional neural networks: an overview and application in radiology. Insights Imaging 9(4):611–629
Ribani R, Marengoni M (2019) A survey of transfer learning for convolutional neural networks. In: 2019 32nd SIBGRAPI conference on graphics, patterns and images tutorials (SIBGRAPI-T). IEEE
Lopes U, Valiati JF (2017) Pre-trained convolutional neural networks as feature extractors for tuberculosis detection. Comput Biol Med 89:135–143
Tajbakhsh N et al (2016) Convolutional neural networks for medical image analysis: full training or fine tuning? IEEE Trans Med Imaging 35(5):1299–1312
Khawaldeh S et al (2017) Noninvasive grading of glioma tumor using magnetic resonance imaging with convolutional neural networks. Appl Sci 8(1):27
Hsieh KL-C, Lo C-M, Hsiao C-J (2017) Computer-aided grading of gliomas based on local and global MRI features. Comput Methods Programs Biomed 139:31–38
Yang Y et al (2018) Glioma grading on conventional MR images: a deep learning study with transfer learning. Front Neurosci 12:804
Sultan HH, Salem NM, Al-Atabany W (2019) Multi-classification of brain tumor images using deep neural network. IEEE Access 7:69215–69225
Ma L et al (2020) Game theoretic interpretability for learning based preoperative gliomas grading. Futur Gener Comput Syst 112:1–10
Gutta S et al (2021) Improved glioma grading using deep convolutional neural networks. Am J Neuroradiol 42(2):233–239
Varma DR (2012) Managing DICOM images: tips and tricks for the radiologist. Indian J Radiol Imaging 22(01):4–13
Ali PJM et al (2014) Data normalization and standardization: a technical report. Mach Learn Tech Rep 1(1):1–6
Ruder S (2016) An overview of gradient descent optimization algorithms. arXiv preprint arXiv:1609.04747
Koidl K (2013) Loss functions in classification tasks. School of Computer Science and Statistic Trinity College, Dublin
Abadi M et al. (2016) Tensorflow: large-scale machine learning on heterogeneous distributed systems. arXiv preprint arXiv:1603.04467
Pedregosa F et al (2011) Scikit-learn: machine learning in python. J Mach Learn Res 12:2825–2830
Huang G et al. (2017) Densely connected convolutional networks. In: Proceedings of the IEEE conference on computer vision and pattern recognition
Howard AG et al. (2017) Mobilenets: efficient convolutional neural networks for mobile vision applications. arXiv preprint arXiv:1704.04861
Szegedy C et al. (2016) Rethinking the inception architecture for computer vision. In: Proceedings of the IEEE conference on computer vision and pattern recognition
Chollet F (2017) Xception: Deep learning with depthwise separable convolutions. In: Proceedings of the IEEE conference on computer vision and pattern recognition
Simonyan K, Zisserman A (2014) Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556
Hossin M, Sulaiman MN (2015) A review on evaluation metrics for data classification evaluations. Int J Data Min Knowl Manag Process 5(2):1
Anaya-Isaza A, Zequera-Diaz M (2022) Fourier transform-based data augmentation in deep learning for diabetic foot thermograph classification. Biocybern Biomed Eng 42(2):437–452
Filipe V, Teixeira P, Teixeira A (2022) Automatic classification of foot thermograms using machine learning techniques. Algorithms 15(7):236
Shi Z et al (2019) A deep CNN based transfer learning method for false positive reduction. Multimed Tools Appl 78(1):1017–1033
Janghel R, Rathore Y (2021) Deep convolution neural network based system for early diagnosis of Alzheimer’s disease. Irbm 42(4):258–267
Sharma S, Mehra R (2020) Conventional machine learning and deep learning approach for multi-classification of breast cancer histopathology images—a comparative insight. J Digit Imaging 33(3):632–654
Priya KM, Kavitha S, Bharathi B (2016) Brain tumor types and grades classification based on statistical feature set using support vector machine. In: 2016 10th International Conference on Intelligent Systems and Control (ISCO). IEEE
Wasule V, Sonar P (2017) Classification of brain MRI using SVM and KNN classifier. In: 2017 Third International Conference on Sensing, Signal Processing and Security (ICSSS). IEEE
Decuyper M, Bonte S, Holen RV (2018) Binary glioma grading: radiomics versus pre-trained CNN features. In: International conference on medical image computing and computer-assisted intervention. Springer
Suja S, George N, George A (2018) Classification of grades of Astrocytoma images from MRI using Deep neural network. In: 2018 2nd International Conference on Trends in Electronics and Informatics (ICOEI). IEEE
George N, Manuel M (2019) A four grade brain tumor classification system using deep neural network. In: 2019 2nd International Conference on Signal Processing and Communication (ICSPC). IEEE
Funding
No funding was received to assist with the preparation of this manuscript.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
The authors declare that there is no conflict of interest.
Ethical statement
This article does not contain any studies with human participants or animals performed by any of the authors.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Fasihi Shirehjini, O., Babapour Mofrad, F., Shahmohammadi, M. et al. Grading of gliomas using transfer learning on MRI images. Magn Reson Mater Phy 36, 43–53 (2023). https://doi.org/10.1007/s10334-022-01046-y
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10334-022-01046-y