Performance Comparison of ResNet50, VGG16, and MobileNetV2 for Brain Tumor Classification on MRI Images

Muhammad Bayu Kurniawan, Ema Utami

Abstract


Brain tumor classification using MRI images is a significant challenge in medical diagnosis, requiring models with high accuracy and efficient training. This study aims to compare the performance of three Convolutional Neural Network (CNN) models—ResNet50, VGG16, and MobileNetV2—for brain tumor classification based on MRI images. The dataset consists of four brain tumor categories: glioma, meningioma, pituitary, and no tumor, with data split into training, validation, and testing sets. Each model was evaluated using metrics including accuracy, precision, recall, F1-score, specificity, and training time to assess their effectiveness in predicting brain tumors with optimal accuracy and efficiency. Experimental results indicate that VGG16 achieved the best overall performance, with an accuracy of 94.93%, precision of 94.68%, and specificity of 98.33%, while also having the shortest training time of 47.15 minutes. MobileNetV2 demonstrated strong performance with a recall of 94.08% but required a longer training time of 79.53 minutes. ResNet50 recorded the lowest accuracy (91.67%) despite excelling in precision (91.79%), but it underperformed in recall (91.25%) and specificity (97.2%). Overall, this study confirms that VGG16 is the most efficient and effective model for MRI-based brain tumor classification.

Keywords


ResNet50; VGG16; MobileNetV2; Brain Tumor Classification; MRI Image

Full Text:

PDF

References


H. Mehnatkesh, S. M. J. Jalali, A. Khosravi, and S. Nahavandi, “An Intelligent Driven Deep Residual Learning Framework for Brain Tumor Classification using MRI Images,” Expert Syst Appl, vol. 213, 2023, doi: 10.1016/j.eswa.2022.119087.

S. Lapointe, A. Perry, and N. A. Butowski, “Primary Brain Tumours in Adults,” The Lancet, vol. 392, no. 10145, pp. 432–446, 2018, doi: 10.1016/S0140-6736(18)30990-5.

D. V. Gore and V. Deshpande, “Comparative Study of Various Techniques using Deep Learning for Brain Tumor Detection,” 2020 International Conference for Emerging Technology, INCET 2020, no. June 2020, 2020, doi: 10.1109/INCET49848.2020.9154030.

À. Rovira and C. Auger, “Beyond McDonald: Updated Perspectives on MRI Diagnosis of Multiple Sclerosis,” Expert Rev Neurother, vol. 21, no. 8, pp. 895–911, 2021, doi: 10.1080/14737175.2021.1957832.

M. Havaei et al., “Brain Tumor Segmentation with Deep Neural Networks,” Med Image Anal, vol. 35, pp. 18–31, 2017, doi: 10.1016/j.media.2016.05.004.

M. Binth, T. Noor, N. Z. Zenia, M. S. Kaiser, S. Al Mamun, and M. Mahmud, “Application of Deep Learning in Detecting Neurological Disorders from Magnetic Resonance Images : A Survey on the Detection of Alzheimer’s Disease , Parkinson’s Disease and Schizophrenia,” Brain Inform, 2020, doi: 10.1186/s40708-020-00112-2.

T. D. Nguyen, D. T. Le, J. Bum, S. Kim, S. J. Song, and H. Choo, “Retinal Disease Diagnosis using Deep Learning on Ultra-Wide-Field Fundus Images,” Diagnostics, vol. 14, no. 1, pp. 1–19, 2024, doi: 10.3390/diagnostics14010105.

S. Natha, U. Laila, I. A. Gashim, K. Mahboob, M. N. Saeed, and K. M. Noaman, “Automated Brain Tumor Identification in Biomedical Radiology Images: A Multi-Model Ensemble Deep Learning Approach,” Applied Sciences, vol. 14, no. 5, p. 2210, 2024, doi: 10.3390/app14052210.

M. B. Kurniawan and E. Utami, “Comparative Analysis of Contrast Enhancement Methods for Classification of Pekalongan Batik Motifs using Convolutional Neural Network,” Jurnal Teknik Informatika (JUTIF), vol. 5, no. 6, pp. 1779–1787, 2024, doi: 10.52436/1.jutif.2024.5.6.2621.

A. Younis, L. Qiang, C. O. Nyatega, M. J. Adamu, and H. B. Kawuwa, “Brain Tumor Analysis using Deep Learning and VGG-16 Ensembling Learning Approaches,” Applied Sciences (Switzerland), vol. 12, no. 14, 2022, doi: 10.3390/app12147282.

L. Xu and M. Mohammadi, “Brain Tumor Diagnosis from MRI based on Mobilenetv2 Optimized by Contracted fox Optimization Algorithm,” Heliyon, vol. 10, no. 1, 2024, doi: 10.1016/j.heliyon.2023.e23866.

P. S. L.Veeranki, G. L. Banavath, and P. R. Devi, “Detection and Classification of Brain Tumors using Convolutional Neural Network,” 7th International Conference on Trends in Electronics and Informatics, ICOEI 2023 - Proceedings, pp. 780–786, 2023, doi: 10.1109/ICOEI56765.2023.10125652.

M. A. B. Siddiaue, S. Sakib, M. M. R. Khan, A. K. Tanzeem, M. Chowdhury, and N. Yasmin, “Deep Convolutional Neural Networks Model-based Brain Tumor Detection in Brain MRI Images,” Proceedings of the 4th International Conference on IoT in Social, Mobile, Analytics and Cloud, ISMAC 2020, pp. 909–914, 2020, doi: 10.1109/I-SMAC49090.2020.9243461.

R. L. Kumar, J. Kakarla, B. V. Isunuri, and M. Singh, “Multi-Class Brain Tumor Classification using Residual Network and Global Average Pooling,” Multimed Tools Appl, vol. 80, no. 9, pp. 13429–13438, 2021, doi: 10.1007/s11042-020-10335-4.

Z. N. K. Swati et al., “Brain Tumor Classification for Mr Images using Transfer Learning and Fine-Tuning,” Computerized Medical Imaging and Graphics, vol. 75, pp. 34–46, 2019, doi: 10.1016/j.compmedimag.2019.05.001.

A. E. Minarno, M. H. C.Mandiri, Y. Munarko, and H. Hariyady, “Convolutional Neural Network with Hyperparameter Tuning for Brain Tumor Classification,” Kinetik: Game Technology, Information System, Computer Network, Computing, Electronics, and Control, vol. 4, 2021, doi: 10.22219/kinetik.v6i2.1219.

P. Rajak, A. S. Jangde, and G. P. Gupta, “Towards Design of Brain Tumor Detection Framework using Deep Transfer Learning Techniques,” IGI Global, pp. 90–103, 2023, doi: 10.4018/978-1-6684-5264-6.ch004.

N. Ullah et al., “An Effective Approach to Detect and Identify Brain Tumors using Transfer Learning,” Applied Sciences (Switzerland), vol. 12, no. 11, 2022, doi: 10.3390/app12115645.

G. S. Tandel, A. Tiwari, and O. G. Kakde, “Performance Optimisation of Deep Learning Models using Majority Voting Algorithm for Brain Tumour Classification,” Comput Biol Med, vol. 135, no. June, p. 104564, 2021, doi: 10.1016/j.compbiomed.2021.104564.

A. Rehman, S. Naz, M. I. Razzak, F. Akram, and M. Imran, “A Deep Learning-based Framework for Automatic Brain Tumors Classification using Transfer Learning,” Circuits Syst Signal Process, vol. 39, no. 2, pp. 757–775, 2020, doi: 10.1007/s00034-019-01246-3.

M. A. Hamza et al., “Optimal and Efficient Deep Learning Model for Brain Tumor Magnetic Resonance Imaging Classification and Analysis,” Applied Sciences (Switzerland), vol. 12, no. 15, 2022, doi: 10.3390/app12157953.

M. Wu, Q. Liu, C. Yan, and G. Sen, “Multi-Classification of Brain Tumors on Magnetic Resonance Images using an Ensemble of Pre-Trained Convolutional Neural Networks,” Curr Med Imaging, vol. 19, pp. 65–76, 2023, doi: 10.2174/1573405618666220415122843.

A. Kazemi, M. E. Shiri, A. Sheikhahmadi, and M. Khodamoradi, “Classifying Tumor Brain Images using Parallel Deep Learning Algorithms,” Comput Biol Med, vol. 148, p. 105775, 2022, doi: 10.1016/j.compbiomed.2022.105775.

N. Noreen, S. Palaniappan, A. Qayyum, I. Ahmad, and M. O. Alassafi, “Brain Tumor Classification based on Fine-Tuned Models and the Ensemble Method,” Computers, Materials and Continua, vol. 67, no. 3, pp. 3967–3982, 2021, doi: 10.32604/cmc.2021.014158.

M. I. Mahmud, M. Mamun, and A. Abdelgawad, “A Deep Analysis of Brain Tumor Detection from MR Images using Deep Learning Networks,” Algorithms, vol. 16, no. 4, pp. 1–19, 2023, doi: 10.3390/a16040176.

N. Shamshad et al., “Enhancing Brain Tumor Classification by a Comprehensive Study on Transfer Learning Techniques and Model Efficiency using MRI Datasets,” IEEE Access, vol. 12, no. May, pp. 100407–100418, 2024, doi: 10.1109/ACCESS.2024.3430109.

N. Masoud, “Brain Tumor MRI Dataset,” Kaggle. Accessed: Sep. 16, 2024. [Online]. Available: https://www.kaggle.com/datasets/masoudnickparvar/brain-tumor-mri-dataset




DOI: https://doi.org/10.32520/stmsi.v14i2.5054

Article Metrics

Abstract view : 158 times
PDF - 34 times

Refbacks

  • There are currently no refbacks.


Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.