An Improved EfficientNet-B0 Approach for Brain Tumor MRI Image Classification
For the task of brain tumor classification in MRI images, this paper proposes a classification model based on an improved EfficientNet-B0 network. Due to the complex pathological features and diversity of medical imaging data, higher requirements are placed on the feature extraction capability and generalization performance of classification models. To address this, we optimized the original EfficientNet-B0 network by introducing the MS-CAM attention mechanism and improving the MBConv module, enabling the model to capture multi-scale feature information more comprehensively, thereby enhancing classification performance and generalization ability. Experimental results show that the improved model achieved a classification accuracy of 99.31% on the validation set, outperforming the original EfficientNet-B0 and other lightweight models. This model can provide clinicians with an efficient and accurate auxiliary diagnostic tool, potentially advancing the development of automated brain tumor diagnosis.
Authors
- Linkun He (ORCID: https://orcid.org/0000-0003-4865-3089)
- Xiang Chen (ORCID: https://orcid.org/0000-0002-0557-0905)
- Xinyuan Li (ORCID: https://orcid.org/0009-0009-9595-7123)
- Yi Wu (ORCID: https://orcid.org/0000-0001-9057-5817)
- He Wen (ORCID: https://orcid.org/0000-0001-8024-1899)
- Lei Du (ORCID: https://orcid.org/0000-0002-2578-5452)
Institutions
- Chengdu Medical College (CN)
Publication Details
- Published
- 2025-03-14
- DOI
- https://doi.org/10.1109/itoec63606.2025.10968077
- Citations
- 2
- Primary Topic
- Brain Tumor Detection and Classification
- Type
- article
- Field-Weighted Citation Impact
- 1.68