Abstract
The timely and precise diagnosis of brain tumors is crucial in reducing mortality rates. Although Magnetic Resonance Imaging (MRI) is commonly used as a detection tool for brain tumors, the presence of unwanted regions in MRI and multi-class brain MRI datasets may pose challenges in accurately classifying tumors. This study proposed a two-phase end-to-end framework comprising DenseNet-121-based deep learning to extract features and an ensemble of machine learning methodologies for precise classification. The deep learning-based feature extraction phase effectively extracted essential and discriminant features that were utilized by multiple machine learning models. Preprocessing MRI images to eliminate unwanted regions enhanced the deep learning model’s feature extraction capabilities. The effectiveness of the proposed framework was evaluated by measuring the classification performance of the ensemble mechanism, which achieved an accuracy of 98.86% and an f1-score of 98.76% without any data augmentation. Notably, the random forest attained the utmost accuracy and f1-score among the machine learning approaches used in the ensemble technique.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
American Society of Clinical Oncology (2023) Brain tumor statistics. https://www.cancer.net/cancer-types/brain-tumor/statistics. Accessed on 6 April 2023
Komninos J, Vlassopoulou V, Protopapa D, Korfias S, Kontogeorgos G, Sakas DE, Thalassinos NC (2004) Tumors metastatic to the pituitary gland: case report and literature review. J Clin Endocrinol Metabol 89(2):574–580
Bauer S, Wiest R, Nolte LP, Reyes M (2013) A survey of MRI-based medical image analysis for brain tumor studies. Phys Med Biol 58(13):R97
Reddy AV, Krishna CP, Mallick PK, Satapathy SK, Tiwari P, Zymbler M, Kumar S (2020) Analyzing MRI scans to detect glioblastoma tumor using hybrid deep belief networks. J Big Data 7:1–17
Khalid S, Khalil T, Nasreen S (2014) A survey of feature selection and feature extraction techniques in machine learning. In: 2014 science and information conference. IEEE, pp 372–378
Huang G, Liu Z, Van Der Maaten L, Weinberger KQ (2017) Densely connected convolutional networks. In: Proceedings of the IEEE conference on computer vision and pattern recognition, pp 4700–4708
Shah HA, Saeed F, Yun S, Park JH, Paul A, Kang JM (2022) A robust approach for brain tumor detection in magnetic resonance images using finetuned efficientnet. IEEE Access 10:65426–65438
Sultan HH, Salem NM, Al-Atabany W (2019) Multi-classification of brain tumor images using deep neural network. IEEE Access 7:69215–69225
Deepak S, Ameer P (2021) Automated categorization of brain tumor from MRI using CNN features and SVM. J Ambient Intell Human Comput 12:8357–8369
Cheng J (2023) Brain tumor dataset (2017). https://doi.org/10.6084/m9.figshare.1512427.v5. https://figshare.com/articles/dataset/brain_tumor_dataset/1512427. Accessed on 6 Mar 2023
Mondal M, Faruk MF, Raihan N, Ahammed P (2021) Deep transfer learning based multi-class brain tumors classification using MRI images. In: 2021 3rd International conference on electrical & electronic engineering (ICEEE). IEEE, pp 73–76
Alanazi MF, Ali MU, Hussain SJ, Zafar A, Mohatram M, Irfan M, AlRuwaili R, Alruwaili M, Ali NH, Albarrak AM (2022) Brain tumor/mass classification framework using magnetic-resonance-imaging-based isolated and developed transfer deep-learning model. Sensors 22(1):372
Ghassemi N, Shoeibi A, Rouhani M (2020) Deep neural network with generative adversarial networks pre-training for brain tumor classification based on MRI images. Biomed Signal Process Control 57:101678
Nickparvar M (2023) Brain tumor MRI dataset (2021). https://doi.org/10.34740/KAGGLE/DSV/2645886. https://www.kaggle.com/dsv/2645886. Accessed on 6 Mar 2023
Bhuvaji S, Kadam A, Bhumkar P, Dedge S, Kanchan S (2020) Brain tumor classification (MRI) (2020). https://doi.org/10.34740/KAGGLE/DSV/1183165. https://www.kaggle.com/dsv/1183165. Accessed on 6 Mar 2023
Hamada A (2020) Br35h:: brain tumor detection 2020. https://www.kaggle.com/datasets/ahmedhamada0/brain-tumor-detection
ImageNet: ImageNet (2020). http://www.image-net.org/. Accessed 19 Jan 2020
Tajbakhsh N, Shin JY, Gurudu SR, Hurst RT, Kendall CB, Gotway MB, Liang J (2016) Convolutional neural networks for medical image analysis: full training or fine tuning? IEEE Trans Med Imag 35(5):1299–1312
John GH, Langley P (2013) Estimating continuous distributions in Bayesian classifiers. arXiv preprint arXiv:1302.4964
Freund Y, Schapire RE (1997) A decision-theoretic generalization of on-line learning and an application to boosting. J Comput Syst Sci 55(1):119–139
Breiman L (2001) Random forests. Mach Learn 45:5–32
Cortes C, Vapnik V (1995) Support-vector networks. Mach Learn 20:273–297
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Shuvo, S.M. et al. (2024). Multi-class Brain Tumor Classification with DenseNet-Based Deep Learning Features and Ensemble of Machine Learning Approaches. In: Arefin, M.S., Kaiser, M.S., Bhuiyan, T., Dey, N., Mahmud, M. (eds) Proceedings of the 2nd International Conference on Big Data, IoT and Machine Learning. BIM 2023. Lecture Notes in Networks and Systems, vol 867. Springer, Singapore. https://doi.org/10.1007/978-981-99-8937-9_38
Download citation
DOI: https://doi.org/10.1007/978-981-99-8937-9_38
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-8936-2
Online ISBN: 978-981-99-8937-9
eBook Packages: EngineeringEngineering (R0)