Efficient Feature Extraction Using MobileNetV2 and EfficientNetB0 for Multi-Class Brain Tumor Classification
DOI:
https://doi.org/10.30871/jaic.v9i6.11354Keywords:
EfficientNetB0, Machine Learning, Lightweight CNN, Brain Tumor ClassificationAbstract
Brain tumor classification in MRI is complicated by the similarity of imaging features across multiple tumor classes. This study evaluates the use of lightweight convolutional neural network (CNN) architectures as feature extractors combined with machine learning classifiers for multi-class classification. MobileNetV2 and EfficientNetB0 were used to extract fixed-length feature representations, which were then classified using Support Vector Machine (SVM), Logistic Regression, Random Forest, and K-Nearest Neighbors. The evaluation used stratified five-fold cross-validation, and performance was measured with accuracy, F1-score, and Matthews Correlation Coefficient (MCC). Results show that EfficientNetB0 features paired with SVM achieved the highest test accuracy (98.5%), while Logistic Regression also yielded competitive performance (97.1%). Class-wise analysis indicated strong results for pituitary and non-tumor cases. This work shows that lightweight CNN-based feature extraction may serve as a practical direction for improving multi-class brain tumor MRI classification, with potential benefits for applications in resource-limited environments.
Downloads
References
[1] S. Kim et al., “Global burden of brain and central nervous system cancer in 185 countries, and projections up to 2050: a population-based systematic analysis of GLOBOCAN 2022,” J Neurooncol, vol. 175, no. 2, pp. 673–685, Nov. 2025, doi: 10.1007/s11060-025-05164-0.
[2] X. Zhao, M. He, R. Yang, N. Geng, X. Zhu, and N. Tang, “The global, regional, and national brain and central nervous system cancer burden and trends from 1990 to 2021: an analysis based on the Global Burden of Disease Study 2021,” Front Neurol, vol. 16, Jun. 2025, doi: 10.3389/fneur.2025.1574614.
[3] S. Bouhafra and H. El Bahi, “Deep Learning Approaches for Brain Tumor Detection and Classification Using MRI Images (2020 to 2024): A Systematic Review,” Journal of Imaging Informatics in Medicine, vol. 38, no. 3, pp. 1403–1433, Sep. 2024, doi: 10.1007/s10278-024-01283-8.
[4] A. S, V. M. Gayathri, and R. Pitchai, “Brain Tumor Segmentation and Survival Prediction using Multimodal MRI Scans with Deep learning Algorithms,” in 2022 International Conference on Innovative Computing, Intelligent Communication and Smart Electrical Systems (ICSES), IEEE, Jul. 2022, pp. 1–5. doi: 10.1109/ICSES55317.2022.9914152.
[5] N. Bhardwaj, M. Sood, and S. S. Gill, “Design and Development of Hypertuned Deep learning Frameworks for Detection and Severity Grading of Brain Tumor using Medical Brain MR images,” Current Medical Imaging Formerly Current Medical Imaging Reviews, vol. 20, Apr. 2024, doi: 10.2174/0115734056288248240309044616.
[6] Y. Dogan, C. Ozdemir, and Y. Kaya, “Enhancing CNN model classification performance through RGB angle rotation method,” Neural Comput Appl, vol. 36, no. 32, pp. 20259–20276, Nov. 2024, doi: 10.1007/s00521-024-10232-z.
[7] I. D. Mienye and T. G. Swart, “A Comprehensive Review of Deep Learning: Architectures, Recent Advances, and Applications,” Information, vol. 15, no. 12, p. 755, Nov. 2024, doi: 10.3390/info15120755.
[8] S. Benyahia, B. Meftah, and O. Lézoray, “Multi-features extraction based on deep learning for skin lesion classification,” Tissue Cell, vol. 74, p. 101701, Feb. 2022, doi: 10.1016/j.tice.2021.101701.
[9] S. Deepak and P. M. Ameer, “Automated Categorization of Brain Tumor from MRI Using CNN features and SVM,” J Ambient Intell Humaniz Comput, vol. 12, no. 8, pp. 8357–8369, Aug. 2021, doi: 10.1007/s12652-020-02568-w.
[10] K. N. Rao et al., “An efficient brain tumor detection and classification using pre-trained convolutional neural network models,” Heliyon, vol. 10, no. 17, p. e36773, Sep. 2024, doi: 10.1016/j.heliyon.2024.e36773.
[11] E. Yagis et al., “Effect of data leakage in brain MRI classification using 2D convolutional neural networks,” Sci Rep, vol. 11, no. 1, p. 22544, Nov. 2021, doi: 10.1038/s41598-021-01681-w.
[12] M. Benaouali, M. Bentoumi, M. Abed, M. Mimi, and A. T. Ahmed, “A Study on CNN-Based and Handcrafted Extraction Methods with Machine Learning for Automated Classification of Breast Tumors from Ultrasound Images,” Electronic Letters on Computer Vision and Image Analysis, vol. 23, no. 2, pp. 85–104, 2024, doi: 10.5565/REV/ELCVIA.1887.
[13] M. A. Gómez-Guzmán et al., “Classifying Brain Tumors on Magnetic Resonance Imaging by Using Convolutional Neural Networks,” Electronics (Basel), vol. 12, no. 4, p. 955, Feb. 2023, doi: 10.3390/electronics12040955.
[14] Masoud Nickparvar, “Brain Tumor MRI Dataset.” Accessed: Aug. 18, 2025. [Online]. Available: https://doi.org/10.34740/kaggle/dsv/2645886
[15] G. Brookshire et al., “Data leakage in deep learning studies of translational EEG,” Front Neurosci, vol. 18, May 2024, doi: 10.3389/fnins.2024.1373515.
[16] M. Sandler, A. Howard, M. Zhu, A. Zhmoginov, and L.-C. Chen, “MobileNetV2: Inverted Residuals and Linear Bottlenecks,” Mar. 2019.
[17] M. Tan and Q. V. Le, “EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks,” Sep. 2020.
[18] J. Aftab, M. A. Khan, S. Arshad, S. ur Rehman, D. A. AlHammadi, and Y. Nam, “Artificial intelligence based classification and prediction of medical imaging using a novel framework of inverted and self-attention deep neural network architecture,” Sci Rep, vol. 15, no. 1, p. 8724, Mar. 2025, doi: 10.1038/s41598-025-93718-7.
[19] B. Bischl et al., “Hyperparameter optimization: Foundations, algorithms, best practices, and open challenges,” WIREs Data Mining and Knowledge Discovery, vol. 13, no. 2, Mar. 2023, doi: 10.1002/widm.1484.
[20] M. Anshori, M. S. Haris, and A. Wahyudi, “Logistic Regression’s Effectiveness in Feature Selection with Information Gain in Predicting Heart Failure Patients,” Journal of Enhanced Studies in Informatics and Computer Applications, vol. 1, no. 2, pp. 35–39, Jul. 2024, doi: 10.47794/jesica.v1i2.8.
[21] J. Tamura, Y. Itaya, K. Hayashi, and K. Yamamoto, “Statistical Inference of the Matthews Correlation Coefficient for Multiclass Classification,” Mar. 2025.
[22] Y. Itaya, J. Tamura, K. Hayashi, and K. Yamamoto, “Asymptotic Properties of Matthews Correlation Coefficient,” Jun. 2024.
[23] A. Muis, S. Sunardi, and A. Yudhana, “Medical image classification of brain tumor using convolutional neural network algorithm,” JURNAL INFOTEL, vol. 15, no. 3, Aug. 2023, doi: 10.20895/infotel.v15i3.964.
[24] K. Puspita, F. Ernawan, Y. Alkhalifi, S. Kasim, and A. Erianda, “Brain Tumor Classification based on Convolutional Neural Networks with an Ensemble Learning Approach through Soft Voting,” JOIV : International Journal on Informatics Visualization, vol. 9, no. 5, p. 1964, Sep. 2025, doi: 10.62527/joiv.9.5.4609.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2025 Hemas Anggita Amelia, Majid Rahardi
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution License (Attribution-ShareAlike 4.0 International (CC BY-SA 4.0) ) that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See The Effect of Open Access).
