Advanced Brain Tumor Classification in MR Images Using Transfer Learning and Pre-Trained Deep CNN Models
Karadeniz Technical University · Norwegian University of Science and Technology
Abstract
A publicly available Brain Tumor MRI dataset containing 7023 images was used in this research. The study employs state-of-the-art pre-trained models, including Xception, MobileNetV2, InceptionV3, ResNet50, VGG16, and DenseNet121, which are fine-tuned using transfer learning, in combination with advanced preprocessing and data augmentation techniques. Transfer learning was applied to fine-tune the models and optimize classification accuracy while minimizing computational requirements, ensuring efficiency in real-world applications.
Among the tested models, Xception emerged as the top performer, achieving a weighted accuracy of 98.73% and a weighted F1 score of 95.29%, demonstrating exceptional generalization capabilities. These models proved particularly effective in addressing class imbalances and delivering consistent performance across various evaluation metrics, thus demonstrating their suitability for clinical adoption. However, challenges persist in improving recall for the Glioma and Meningioma categories, and the black-box nature of deep learning models requires further attention to enhance interpretability and trust in medical settings.
Citation impact
- FWCI
- 47.54
- Percentile
- 100%
- References
- 68
Authors
3Topics & keywords
- Computer science
- Deep learning
- Artificial intelligence
- Interpretability
- Transfer of learning
- Machine learning
- Brain tumor
- Preprocessor