Deep Learning Technique for MRI-Based Brain Tumor Classification

Zicheng Zhang

doi:10.62051/n521xr37

Authors

Zicheng Zhang

DOI:

https://doi.org/10.62051/n521xr37

Keywords:

Brain Tumor, Convolutional Neural Network, MRI scan, Deep Learning.

Abstract

Brain tumors represent a severe threat to human health due to their potential to disrupt vital functions controlled by the brain, such as movement, sensation, and cognition. This study aims to classify a comprehensive dataset of 7,023 magnetic resonance imaging (MRI) images from Figshare, SARTAJ, and BR35H into four distinct classes (Glioma, Meningioma, Pituitary, and No-Tumor). After evaluating the state-of-the-art models including VGG16, DenseNet201, ResNet50, and V3 Inception, the study proposes a novel Convolutional Neural Network (CNN) model with multiple inception layers. As a result, the model achieved an accuracy of 98.71% and high AUC scores for all four categories. These results suggest the potential of this model in enhancing diagnostic accuracy and reducing radiologist workload in clinical applications. Further improvements could focus on advanced feature extraction and integration of multi-parametric MRI data. The model's robustness and effectiveness underscore its potential utility in real-world clinical settings for accurate tumor classification.

Downloads

Download data is not yet available.

References

[1] Key Statistics for Brain and Spinal Cord Tumors. (n.d.). American Cancer Society. Retrieved on June 18, 2024. Retrieved from: https://www.cancer.org/cancer/types/brain-spinal-cord-tumors-adults/about/key-statistics.html

[2] Cancer of the Brain and Other Nervous System - Cancer Stat Facts. (n.d.). SEER. Retrieved on June 18, 2024. Retrieved from: https://seer.cancer.gov/statfacts/html/brain.html

[3] Key Statistics for Brain and Spinal Cord Tumors. (n.d.-b). American Cancer Society. Retrieved on June 18, 2024. Retrieved from: https://www.cancer.org/cancer/types/brain-spinal-cord-tumors-adults/about/key-statistics.html

[4] Developer. MRI vs. CT scan. Health Images. August 30, 2021. Retrieved on June 18, 2024. Retrieved from: https://www.healthimages.com/mri-vs-ct-scan/

[5] Mohan G, Subashini M M. MRI based medical image analysis: Survey on brain tumor grade classification. Biomed. Signal Processing Control. 2018, 39, 139-161.

[6] Xie Y, Zaccagna F, Rundo L, et al. Convolutional neural network techniques for brain tumor classification (from 2015 to 2022): review, challenges, and future perspectives. Diagnostics, 2022, 12(8), 1850.

[7] Havaei M, Davy A, Warde Farley D, et al. Brain tumor segmentation with deep neural networks. Medical Image Analysis, 2017, 35, 18-31.

[8] Ravinder M, Saluja G, Allabun S, et al. Enhanced brain tumor classification using graph convolutional neural network architecture. Scientific Reports, 2023, 13(1).

[9] ZainEldin H, Gamel S A, El-Kenawy E S M, et al. Brain tumor detection and classification using deep learning and sine-cosine fitness grey wolf optimization. Bioengineering, 2022, 10(1), 18.

[10] Saeedi S, Rezayi S, Keshavarz H, et al. MRI-based brain tumor detection using convolutional deep learning methods and chosen machine learning techniques. BMC Medical Informatics and Decision Making, 2023, 23(1).

[11] Krizhevsky A, Sutskever I, Hinton G E. ImageNet classification with deep convolutional neural networks. Advances in Neural Information Processing Systems, 2012, 25, 1097-1105.

[12] Ranzato M, Boureau Y L, Chopra S, et al. Efficient learning of sparse representations with an energy-based model. Advances in Neural Information Processing Systems, 2017, 20, 1137-1144.

[13] Srivastava N, Hinton G, Krizhevsky A, et al. Dropout: A simple way to prevent neural networks from overfitting. Journal of Machine Learning Research, 2014, 15(1), 1929-1958.

[14] Szegedy C, Liu W, Jia Y, et al. Going deeper with convolutions. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2015, 1-9.

[15] Glorot X, Bengio Y. Understanding the difficulty of training deep feedforward neural networks. Proceedings of the Thirteenth International Conference on Artificial Intelligence and Statistics, 2010, 249-256.

[16] Liu Y, Bai Y, Zhu X. Radiomics and its emerging role in the evaluation and management of brain tumors: A review of current applications and future perspectives. Journal of Magnetic Resonance Imaging, 2019 50(3), 877-890.

[17] Komura D, Ishikawa S. Machine learning methods for histopathological image analysis. Computational and Structural Biotechnology Journal, 2018, 16, 34-42.