Early detection of diseases is a crucial step towards patient healthcare and recovery, especially, for low survival rate diseases like brain cancer. In recent years, deep learning models such as convolutional neural networks (CNNs) have been widely utilized for medical image classification owing to their proficiency in extracting local features. However, accurate classification of complex medical images necessitates capturing local and global features. This paper focuses on the classification of the three most prominent kinds of brain tumors, namely, glioma, meningioma and pituitary tumor, with the proposed CNN based on the structure of GoogLeNet. The proposed model uses an attention-based inception module (ABIM) embedded with attention mechanism of spatial attention and efficient channel attention. Attention helps CNN focus on features with more distinguishing power and ignores irrelevant features while the inception approach aids in extracting multiscale local features. ABIM incorporates attention with inception to generate better feature representation. The model training and testing are done on a publicly accessible Figshare dataset with 3064 brain MRI samples. The dataset comprises 1426, 708, and 930 MRI samples of glioma, meningioma, and pituitary tumors, respectively. In a five-fold cross-validation experiment, the proposed model achieved an average accuracy of 97.97% for glioma, 94.76% for meningioma, and 99.22% for pituitary tumors. The proposed CNN model surpasses existing models with an overall accuracy of 97.62%, while utilizing fewer parameters than its base model, GoogLeNet. The model is also evaluated on datasets from the Kaggle and Harvard medical repositories, and showed promising results in distinguishing between different tumor classes, and normal brain MRIs. These results demonstrate that our model not only outperforms other existing models in terms of accuracy but also has comparatively lower computational complexity, providing a robust tool to support healthcare professionals.
Read full abstract