Examining Transfer Learning Models to Classify Brain Tumors from MRI Images: A Comparative Analysis
摘要
This research focuses on the significant challenge of classifying brain tumors using deep transfer learning methods, a crucial area of study due to the limited progress in improving brain cancer survival rates despite extensive research. The study concentrates on the identification of brain tumors from MRI images and employs four well-known deep transfer learning models: InceptionResNet-V2, MobileNet, ResNet50, and VGG16. A dataset of 7023 Brain Tumor MRI images from Kaggle was utilized, divided into training, validation, and testing sets, for thorough model training and evaluation. The performance assessment of these models involved standard metrics such as Accuracy, F1-score, Recall, and Precision. Results show that all models performed competitively in classifying brain tumors. Notably, InceptionResNet-V2 achieved an impressive 98.78% accuracy on the test set, with VGG16 displaying 96.26% accuracy. Both ResNet50 and InceptionResNet-V2 showed similar accuracies, highlighting the efficacy of transfer learning, particularly in situations with limited datasets. This study stresses the significance of early tumor detection and classification for enhancing patient outcomes and highlights the potential of these models in aiding early diagnosis. In conclusion, this research sets the foundation for creating adaptable models capable of addressing various brain tumor types, potentially transforming the approach to cancer diagnosis and treatment in the medical field.