Ensemble Deep Learning with Attention Mechanism and Explainable AI for Enhanced Brain Tumor Classification from MRI Images
TL;DR
The findings demonstrate the potential of combining ensemble learning, attention, and visual explanation for brain tumor classification and represent internal validation on a single partition of aggregated public datasets.
Abstract
Brain tumors represent a significant clinical challenge, with accurate classification being essential for treatment planning. Current deep learning approaches face two critical limitations: insufficient robust-ness across diverse imaging conditions and lack of clinical interpretability. Here we present an ensemble deep learning framework integrating three complementary architectures (MobileNetV2, EfficientNetB3, and DenseNet121) with spatial attention mechanisms and explainability features. Using 7,023 MRI images across four diagnostic categories (glioma, meningioma, pituitary tumor, and tumor-absent), our approach achieved 98.47% classification accuracy with balanced performance across all classes (F1-scores: 98.12% for glioma, 97.89% for meningioma, 98.76% for pituitary, 99.21% for tumor-absent cases). The ensemble demonstrated statistically significant improvement over individual models (p < 0.01, McNemar’s test), with gains of 1.24-1.75 percentage points. Integration of Gradient-weighted Class Activation Map-ping provided interpretable visual explanations with activation patterns consistently focusing on tumor regions. The findings demonstrate the potential of combining ensemble learning, attention, and visual explanation for brain tumor classification. However, the results represent internal validation on a single partition of aggregated public datasets and require confirmation through repeated validation and independent clinical evaluation.