Brain tumors pose a significant global health challenge, impacting individuals across all age groups. Early and precise diagnosis is critical for effective treatment planning and enhancing patient outcomes. The variability in tumor appearance and the complexity of brain structures make manual examination prone to errors. The availability of advanced datasets and the opportunity to contribute to healthcare and interdisciplinary research inspired this project.
- π Enhance the efficiency and accuracy of deep learning (DL) models for brain tumor classification.
- π Compare multiple DL models to determine the optimal approach for real-world application.
The dataset, sourced from Kaggle. It comprises 3,264 MRI scans of brain tumors categorized as:
- Glioma Tumors (926 images)
- Meningioma Tumors (937 images)
- Pituitary Tumors (901 images)
- No Tumors (500 images)
Steps taken to prepare the data for model training:
- βοΈ Cropping: Removed irrelevant regions to focus on tumor areas
- π Resizing: Standardized image size to 225Γ225Γ3
- βοΈ Normalization: Scaled pixel values to a range of [0, 1]
To boost dataset diversity and reduce overfitting:
- π Rotation (Β±30Β°)
- πͺ Shear (0.2)
- π Zoom (0.05)
βοΈ Horizontal Flip
Two models were employed:
-
Custom CNN:
- 5 convolutional layers with increasing filters: 16 β 32 β 64 β 128 β 512
- MaxPooling2D, ReLU activations, and dropout layers (rates: 0.2 and 0.5)
- Fully connected layer with 512 units and softmax output for classification
-
EfficientNet-B0:
- Advanced scalable architecture for efficient training and high accuracy
- Batch size: 64
- Epochs: 30
- Loss function: Categorical Cross-Entropy
- Optimizer: Adam
EfficientNet-B0 outperformed the custom CNN with:
- β Validation Accuracy: 97.55%
- π Validation Loss: 0.10
- π§ͺ Training Accuracy: 99.74%
It provided a superior balance between performance and computational efficiency.
This project highlights the potential of deep learning, especially EfficientNet-B0, in automating brain tumor classification from MRI scans. Such systems can assist radiologists by:
- π©Ί Increasing diagnostic accuracy
- π§ββοΈ Reducing workload
- π Accelerating treatment timelines
- Alishba Zulfiqar
- Bushra Tanveer
- Zahra Batool