Related Experiment Video
Updated: Jul 6, 2025

14:08
Automated Midline Shift and Intracranial Pressure Estimation based on Brain CT Images
Published on: April 13, 2013
42.6K
Detection and classification of brain tumor using hybrid deep learning models.
Baiju Babu Vimala1, Saravanan Srinivasan2, Sandeep Kumar Mathivanan3
1School of Computer Science and Engineering, Vellore Institute of Technology, Vellore, 632014, Tamil Nadu, India.
Scientific Reports
|December 28, 2023
Summary
This study fine-tuned EfficientNets using transfer learning for brain tumor classification. EfficientNetB2 achieved 99.06% accuracy in identifying glioma, meningioma, and pituitary tumors from MRI scans.
Area of Science:
- Medical Imaging Analysis
- Artificial Intelligence in Oncology
- Deep Learning for Diagnostics
Background:
- Accurate brain tumor classification is vital for effective treatment and patient outcomes.
- Magnetic Resonance Imaging (MRI) is a key non-invasive tool for brain tumor diagnosis.
- Deep learning, particularly Convolutional Neural Networks (CNNs), has enhanced medical image interpretation.
Purpose of the Study:
- To classify brain tumors (glioma, meningioma, pituitary) using a transfer learning approach with EfficientNets.
- To evaluate the performance of fine-tuned EfficientNet models (B0-B4) on the CE-MRI Figshare dataset.
- To analyze the impact of data augmentation and visualize model decisions using Grad-CAM.
Main Methods:
- Transfer learning-based fine-tuning of pre-trained EfficientNet models (B0-B4) initialized with ImageNet weights.
- Addition of custom layers for brain tumor classification task.
- Utilized the CE-MRI Figshare dataset and incorporated data augmentation techniques.
- Employed Grad-CAM for visualizing model attention and highlighting tumor regions.
Main Results:
- EfficientNetB2 demonstrated superior performance among the tested EfficientNet models.
- Achieved high performance metrics: 99.06% overall test accuracy, 98.73% precision, 99.13% recall, and 98.79% F1-score.
- Grad-CAM visualization effectively identified tumor locations, aiding in understanding model predictions.
Conclusions:
- Fine-tuned EfficientNetB2 offers a highly accurate and robust solution for classifying common brain tumor types from MRI.
- The transfer learning approach with EfficientNets shows significant promise for improving automated brain tumor diagnosis.
- Data augmentation and visualization techniques enhance model reliability and interpretability in medical imaging applications.

