Ensemble of EfficientNet B1 and ResNet 101 with Attention Mechanism for Brain Tumor Classification in MRI Images

Zare Kordkheili, Amirhossein; Zare Kordkheili, Amirreza; Asadi Amiri, Sekine

doi:10.22044/jadm.2026.17084.2845

Document Type : Original/Review Paper

Authors

¹ University of Mazandaran

² Department of Computer Engineering, University of Tehran, Tehran, Iran

https://doi.org/10.22044/jadm.2026.17084.2845

Abstract

Brain tumor detection is a critical task in medical imaging, requiring accurate and reliable methods. Recent advancements in deep learning have shown great potential in this field. In this article, we present a novel method for brain tumor detection based on a Convolutional Block Attention Module (CBAM) enhanced attention ensemble of deep learning networks. Initially, image augmentation is applied to increase data diversity. We utilize two deep neural network models, EfficientNet-B1 and ResNet-101, for tumor detection. First, we enhance the performance of these models by integrating the CBAM attention module into their architectures. Then, we ensemble the two networks using a soft voting strategy to achieve higher detection accuracy. The proposed method is evaluated on the three-class Figshare dataset, achieving an accuracy of 99.09% in detecting tumors in MRI images, which outperforms existing methods. This approach leverages the strengths of an ensemble of models, offering a promising solution for improving the accuracy and reliability of brain tumor detection in medical imaging.

Keywords

Main Subjects

H.5. Image Processing and Computer Vision

References

[1] L. Arokia Jesu Prabhu and A. Jayachandran, "Mixture model segmentation system for parasagittal meningioma brain tumor classification based on hybrid feature vector," Journal of medical systems, vol. 42, no. 12, p. 251, 2018.

[2] A. Akter et al., "Robust clinical applicable CNN and U-Net based algorithm for MRI classification and segmentation for brain tumor," Expert Systems with Applications, vol. 238, p. 122347, 2024.

[3] E. Irmak, "Multi-classification of brain tumor MRI images using deep convolutional neural network with fully optimized framework," Iranian Journal of Science and Technology, Transactions of Electrical Engineering, vol. 45, no. 3, pp. 1015–1036, 2021.

[4] E.-S. A. El-Dahshan, H. M. Mohsen, K. Revett, and A.-B. M. Salem, "Computer-aided diagnosis of human brain tumor through MRI: A survey and a new algorithm," Expert systems with Applications, vol. 41, no. 11, pp. 5526–5545, 2014.

[5] M. Nawaz et al., "Analysis of brain MRI images using improved cornernet approach," Diagnostics, vol. 11, no. 10, p. 1856, 2021.

[6] K. M. Iftekharuddin, J. Zheng, M. A. Islam, and R. J. Ogg, "Fractal-based brain tumor detection in multimodal MRI," Applied Mathematics and Computation, vol. 207, no. 1, pp. 23–41, 2009.

[7] M. M. Zahoor et al., "Brain tumor MRI classification using a novel deep residual and regional CNN," Biomedicines, vol. 12, no. 7, p. 1395, 2024.

[8] P. M. Shakeel, T. E. E. Tobely, H. Al-Feel, G. Manogaran, and S. Baskar, "Neural network based brain tumor detection using wireless infrared imaging sensor," IEEE Access, vol. 7, pp. 5577–5588, 2019.

[9] A. Kharrat, K. Gasmi, M. B. Messaoud, N. Benamrane, and M. Abid, "A hybrid approach for automatic classification of brain MRI using genetic algorithm and support vector machine," Leonardo journal of sciences, vol. 17, no. 1, pp. 71–82, 2010.

[10] J. Cheng, "brain tumor dataset," ed: figshare, 2017.

[11] M. R. Ismael and I. Abdel-Qader, "Brain tumor classification via statistical features and back-propagation neural network," in 2018 IEEE international conference on electro/information technology (EIT), 2018: IEEE, pp. 0252–0257.

[12] J. S. Paul, A. J. Plassard, B. A. Landman, and D. Fabbri, "Deep learning for brain tumor classification," in Medical Imaging 2017: Biomedical Applications in Molecular, Structural, and Functional Imaging, 2017, vol. 10137: SPIE, pp. 253–268.

[13] P. Afshar, A. Mohammadi, and K. Plataniotis, "Brain tumor type classification via capsule networks. In2018 25th IEEE international conference on image processing (ICIP) 2018 Oct 7 (pp. 3129-3133)," arXiv preprint arXiv:1801.09597.

[14] P. Afshar, K. N. Plataniotis, and A. Mohammadi, "Capsule networks for brain tumor classification based on MRI images and coarse tumor boundaries," in ICASSP 2019-2019 IEEE international conference on acoustics, speech and signal processing (ICASSP), 2019: IEEE, pp. 1368–1372.

[15] A. K. Anaraki, M. Ayati, and F. Kazemi, "Magnetic resonance imaging-based brain tumor grades classification and grading via convolutional neural networks and genetic algorithms," biocybernetics and biomedical engineering, vol. 39, no. 1, pp. 63–74, 2019.

[16] N. Ghassemi, A. Shoeibi, and M. Rouhani, "Deep neural network with generative adversarial networks pre-training for brain tumor classification based on MR images," Biomedical Signal Processing and Control, vol. 57, p. 101678, 2020.

[17] Y. Zhou et al., "Holistic brain tumor screening and classification based on densenet and recurrent neural network," in International MICCAI brainlesion workshop, 2018: Springer, pp. 208–217.

[18] M. Aamir et al., "A deep learning approach for brain tumor classification using MRI images," Computers and Electrical Engineering, vol. 101, p. 108105, 2022.

[19] F. Demir and Y. Akbulut, "A new deep technique using R-CNN model and L1NSR feature selection for brain MRI classification," Biomedical Signal Processing and Control, vol. 75, p. 103625, 2022.

[20] A. K. Sartaj Bhuvaji, Prajakta Bhumkar, Sameer Dedge, Swati Kanchan. Brain Tumor Classification (MRI), doi: https://doi.org/10.34740/KAGGLE/DSV/12745533.

[21] X. Liu and Z. Wang, "Deep learning in medical image classification from mri-based brain tumor images," in 2024 IEEE 6th International Conference on Power, Intelligent Computing and Systems (ICPICS), 2024: IEEE, pp. 840–844.

[22] A. Hamada. Br35H :: Brain Tumor Detection 2020. [Online]. Available: https://www.kaggle.com/datasets/ahmedhamada0/brain-tumor-detection

[23] C. K. K. Reddy et al., "A fine-tuned vision transformer based enhanced multi-class brain tumor classification using MRI scan imagery," Frontiers in oncology, vol. 14, p. 1400341, 2024.

[24] J. Manikandan and S. Snega, "TriCANet: Cross-Plane Attention and Swin Transformer Fusion for Multi-View Brain MRI Tumor Classification," in 2025 10th International Conference on Research in Intelligent Computing in Engineering (RICE), 2025: IEEE, pp. 1–6.

[25] S. U. R. Khan, S. Asif, M. Zhao, W. Zou, Y. Li, and C. Xiao, "ShallowMRI: A novel lightweight CNN with novel attention mechanism for Multi brain tumor classification in MRI images," Biomedical Signal Processing and Control, vol. 111, p. 108425, 2026.

[26] S. Woo, J. Park, J.-Y. Lee, and I. S. Kweon, "Cbam: Convolutional block attention module," in Proceedings of the European conference on computer vision (ECCV), 2018, pp. 3–19.

[27] M. Tan and Q. Le, "Efficientnet: Rethinking model scalingfo convolutional neural networks," in International conference on machine learning, 2019: PMLR, pp. 6105–6114.

[28] A G. Howard et al., "Mobilenets: Efficient convolutonal neural networks for mobile vision applicatins," arXiv preprint arXiv:1704.04861, 2017.

[29] K. He, X. Zhang, S. Ren, and J. Sun, "Deep residual learning for image recognition," in Proceedings of the IEEE conference on computer vision and pattern recognition, 2016, pp. 770–778.

[30] Z. Mohammed and D. J. Mussa, "Brain tumour classificaion using BoF-SURF with filter-based feature selection methods," Multimedia Tools and Applications, vol. 83, no. 25, pp. 65833–65855, 2024.

[31] S. Mavaddati, "A Hybrid Approach for Brain Tumor Classification: Enhancing MRI-Based Diagnosis with CNN-Transformer Synergy," Journal of AI and Data Mining, vol. 14, no. 1, pp. 37–49, 2026.

Ensemble of EfficientNet B1 and ResNet 101 with Attention Mechanism for Brain Tumor Classification in MRI Images

References

References

Volume 14, Issue 3July 2026Pages 381-393

Volume 14, Issue 3
July 2026
Pages 381-393