Improved EfficientNet Architecture for Multi-Grade Brain Tumor Detection

Abstract

Accurate detection and diagnosis of brain tumors at early stages is significant for effective treatment. While numerous methods have been developed for tumor detection and classification, several rely on traditional techniques, often resulting in suboptimal performance. In contrast, AI-based deep learning techniques have shown promising results, consistently achieving high accuracy across various tumor types while maintaining model interpretability. Inspired by these advancements, this paper introduces an improved variant of EfficientNet for multi-grade brain tumor detection and classification, addressing the gap between performance and explainability. Our approach extends the capabilities of EfficientNet to classify four tumor types: glioma, meningioma, pituitary tumor, and non-tumor. For enhanced explainability, we incorporate gradient-weighted class activation mapping (Grad-CAM) to improve model interpretability. The input MRI images undergo data augmentation before being passed through the feature extraction phase, where the underlying tumor patterns are learned. Our model achieves an average accuracy of 98.6%, surpassing other state-of-the-art methods on standard datasets while maintaining a substantially reduced parameter count. Furthermore, the explainable AI (XAI) analysis demonstrates the model’s ability to focus on relevant tumor regions, enhancing its interpretability. This accurate and interpretable model for brain tumor classification has the potential to significantly aid clinical decision-making in neuro-oncology.