Using The Combined Model Between MobileNetV2 and EfficientNetB0 to Classify Brain Tumors Based on MRI Images

Nhon Nguyen Thien, Kheo Chau Mui, Hoang-Tu Vo, Phuc Pham Tien, Huan Lam Le

Abstract


Brain tumors are extremely dangerous to one's health. If unchecked cell proliferation is not identified and treated promptly, it can lead to mortality, raise intracranial pressure, and endanger lifespan. To remove the tumor and lengthen the patient's life, early illness identification and drug administration are essential. In this research paper, we aim to improve the effectiveness of magnetic resonance imaging (MRI) equipment to identify cancerous brain tumour cells. It helps experts identify diseases faster. We classify brain tumour cells based on an image set of 3264 images with effective classification models such as ResNet50, InceptionV3, VGG19, EfficientNetB7, DenseNet201, MobileNetV2, Xception, etc. Besides, we also proposed two combined models: pooling (Xception + ResNet50) and pooling (MobileNetV2 + EfficientNetB0) to evaluate the effectiveness and found that the pooling model (MobileNetV2 + EfficientNetB0) gives the highest result, with 100% for the training set, 98% for the valid set, and 78% for the test set. We continued to improve the model by randomly re-dividing the data set with a Train-Valid-Test ratio of 60:20:20 and obtained an increased F1-score of 97%. We continued to improve the model again using the data augmentation techniques to create a larger data set, and the results far exceeded expectations with an F1-score of almost 100% for all classes. Based on the results, we found that combining MobileNetV2 with EfficientNetB0 is suitable for detecting brain tumour cancer cells. Aids in the early detection of dangerous cancers before they spread and endanger human health.

Keywords


Classification of Brain tumors;improve the model;fine-tuning;image classification; EfficientNetB0; MobileNetV2; Deep learning

References


Soni, Vaibhav & Singh, Nikhil & Singh, Rishi & Tomar, Deepak. (2023). Multiencoder‐based federated intelligent deep learning model for brain tumor segmentation. International Journal of Imaging Systems and Technology. n/an/a. 10.1002/ima.22981.

R. Young, J. Knopp, and A. Edmond, “Brain MRI: tumor evaluation,” Journal of Magnetic Resonance Imaging: An Official Journal of the International Society for Magnetic Resonance in Medicine, vol. 24, no. 4, pp. 709–724, 2006.

T. Logeswari and M. Karnan, “An improved implementation of brain tumor detection using segmentation based on hierarchical self organizing map,” International Journal of Computer Theory Engineering, vol. 2, no. 4, pp. 591– 595, 2010.

R. L. Siegel, K. D. Miller, and A. Jemal, “Cancer statistics, 2015,” CA: a Cancer Journal for Clinicians, vol. 65, no. 1, pp. 5–29, 2015.

R. L. Siegel, K. D. Miller, and A. Jemal, “Cancer statistics, 2019,” CA: a Cancer Journal for Clinicians, vol. 69, no. 1, pp. 7–34, 2019.

R. L. Siegel, K. D. Miller, H. E. Fuchs, and A. Jemal, “Cancer statistics, 2021,” CA: a Cancer Journal for Clinicians, vol. 71, no. 1, pp. 7–33, 2021.

H. T. Zaw, N. Maneerat and K. Y. Win, "Brain tumor detection based on Naïve Bayes Classification," 2019 5th International Conference on Engineering, Applied Sciences and Technology (ICEAST), Luang Prabang, Laos, 2019, pp. 1-4, doi: 10.1109/ICEAST.2019.8802562.

A. Pandey and V. K. Pandey, "Deep Transfer Learning Models for Brain Tumor Classification Using Magnetic Resonance Images," 2023 IEEE 12th International Conference on Communication Systems and Network Technologies (CSNT), Bhopal, India, 2023, pp. 398-403, doi: 10.1109/CSNT57126.2023.10134745.

Farhana Alam, Farhana Chowdhury Tisha, Sara Anisa Rahman, Samia Sultana, Md. Ahied Mahi Chowdhury, Ahmed Wasif Reza and Mohammad Shamsul Arefin, “Automated Brain Disease Classification using Transfer Learning based Deep Learning Models” International Journal of Advanced Computer Science and Applications(IJACSA), 13(9), 2022. http://dx.doi.org/10.14569/IJACSA.2022.01309109

American Cancer Society. Cancer Facts & Figures 2019. https://www.cancer.org/research/cancer-facts-statistics/all-cancer-facts-figures.html

SEER Cancer Statistics Review 1975–2015. National Cancer Institute. https://seer.cancer.gov

American Cancer Society. Cancer Facts & Figures 2025 (projections). https://acsjournals.onlinelibrary.wiley.com/

The Brain Tumour Charity (UK). https://www.thebraintumourcharity.org

Cancer Research UK. Brain Tumour Statistics. https://www.cancerresearchuk.org/health professional/cancerstatistics/statistics-by-cancer- type/brain-tumours

Khan, Muhammad & Khan, Awais & Alhaisoni, Majed & Alqahtani, Abdullah & Alsubai, Shtwai & Alharbi, Meshal & Malik, Nazir & Damaševičius, Robertas. (2022). Multimodal Brain Tumor Detection and Classification using Deep Saliency Map and Improved Dragonfly Optimization Algorithm. International Journal of Imaging Systems and Technology. 33. 10.1002/ima.22831.

M. Michael and W. Smith, “WebMD cancer center: types of brain cancer,”2020, https://www.webmd.com/cancer/brain-cancer/brain-tumor-types.

P. C. Wesseling and D. Capper, “WHO 2016 classification of gliomas,” Neuropathology Applied Neurobiology, vol. 44, no. 2, pp. 139–150, 2018.

M. R. Gupta, V. Rajagopalan, and B. V. V. S. N. P. Rao, “Glioma grade classification using wavelet transform-local binary pattern based statistical texture features and geometric measures extracted from MRI,” Journal of Experimental Theoretical Artificial Intelligence, vol. 31, no. 1, pp. 57–76, 2019.

M. Y. Talo, O. Yildirim, U. B. Baloglu, G. Aydin, and U. R. Acharya, “Convolutional neural networks for multiclass brain disease detection using MRI images,” Computerized Medical Imaging and Graphics, vol. 78, article 101673, 2019.

Sandler, M., Howard, A., Zhu, M., Zhmoginov, A., & Chen, L. C. (2018). MobileNetV2: Inverted Residuals and Linear Bottlenecks. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (pp. 4510-4520)

He, K.; Zhang, X.; Ren, S.; Sun, J. Deep residual learning for image recognition. In Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR 2016), Las Vegas, NV, USA, 27–30 June 2016; pp. 770–778.

Mark Sandler, Andrew Howard, Menglong Zhu, Andrey Zhmoginov, and Liang-Chieh Chen. Mobilenetv2: Inverted residuals and linear bottlenecks. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 4510–4520, 2018.

Mingxing Tan and Quoc Le. Efficientnet: Rethinking model scaling for convolutional neural networks. In International conference on machine learning, pages 6105–6114. PMLR, 2019.

H. Asri, H. Mousannif, H. Moatassime, & T. Noël. Using Machine Learning Algorithms for Breast Cancer Risk Prediction and Diagnosis. ANT/SEIT, 2017.

N. Ponraj, E. Jenifer., P. Poongodi, S. Manoharan. "Morphological operations for the mammogram image to increase the contrast for the efficient detection of breast cancer", European Journal of Scientific Research, (ISSN) 1450-216X, vol. 68, no.4, pp. 494-505, 2021.

D. R. Pavithra, S. Preethi, & A. K. SriRakshitha. Breast Cancer Classification using the Supervised Learning Algorithms. 2021 5th International Conference on Intelligent Computing and Control Systems (ICICCS), pp. 1492-1498, 2021

H. Asri, H. Mousannif, H. Moatassime, & T. Noël. Using Machine Learning Algorithms for Breast Cancer Risk Prediction and Diagnosis. ANT/SEIT, 2017.

B. Gayathri, C. Sumathi, T. Santhanam, "Breast cancer diagnosis using machine learning algorithm a survey”. International Journal of Distributed and Parallel Systems, vol. 4, no. 3, pp. 39-50, 2017.

M. F. Ak. A Comparative Analysis of Breast Cancer Detection and Diagnosis Using Data Visualization and Machine Learning Applications. Healthcare, vol. 8, no. 2, pp.1-11, 2020.

H. Mousannif, H. Asri, H. A. Moatassime, T. Noel. Using Machine Learning Algorithms for Breast Cancer Risk Prediction and Diagnosis. Procedia Computer Science, vol. 83, pp. 1064–1069, 2017.

U. K. Kumar, M. B. S. Nikhil and K. Sumangali, "Prediction of breast cancer using voting classifier technique,"

IEEE International Conference on Smart Technologies and Management for Computing, Communication, Controls, Energy and Materials (ICSTM), 2017, pp. 108-114.

Hoang-Tu Vo, Nhon Nguyen Thien and Kheo Chau Mui, “Tomato Disease Recognition: Advancing Accuracy Through Xception and Bilinear Pooling Fusion” International Journal of Advanced Computer Science and Applications(IJACSA), 14(8), 2023. http://dx.doi.org/10.14569/IJACSA.2023.01408113

Cengıl, E., Çinar, A. (2019). Multiple classification of flower images using transfer learning. In 2019 International

Artificial Intelligence and Data Processing Symposium (IDAP), Malatya, Turkey, pp. 1-6. https://doi.org/10.1109/IDAP.2019.8875953

D. M. Joshi, N. K. Rana and V. M. Misra, "Classification of Brain Cancer using Artificial Neural Network," 2010 2nd International Conference on Electronic Computer Technology, Kuala Lumpur, Malaysia, 2010, pp. 112-116, doi: 10.1109/ICECTECH.2010.5479975.

M. Thachayani and S. Kurian, "AI Based Classification Framework For Cancer Detection Using Brain MRI Images," 2021 International Conference on System, Computation, Automation and Networking (ICSCAN), Puducherry, India, 2021, pp. 1-4, doi: 10.1109/ICSCAN53069.2021.9526456.

B. Erkal, S. Başak, A. Çiloğlu and D. D. Şener, "Multiclass Classification of Brain Cancer with Machine Learning Algorithms," 2020 Medical Technologies Congress (TIPTEKNO), Antalya, Turkey, 2020, pp. 1-4, doi: 10.1109/TIPTEKNO50054.2020.9299233.

V. G. Hamsaveni, S. P. Maniraj, R. Krishnaswamy, K. Radhika, V. Venkataramanan and S. Renukadevi, "Intelligent

Classification of Brain Cancers by Deep Learning with Inception Network," 2022 International Conference on Computer, Power and Communications (ICCPC), Chennai, India, 2022, pp. 681-684, doi: 10.1109/ICCPC55978.2022.10072278.

Shenbagarajan Anantharajan, Shenbagalakshmi Gunasekaran, Thavasi Subramanian, Venkatesh R, MRI brain tumor detection using deep learning and machine learning approaches, Measurement: Sensors, Volume 31, 2024, 101026, ISSN 2665-9174,https://doi.org/10.1016/j.measen.2024.101026.


Full Text: PDF

Refbacks

  • There are currently no refbacks.


 

Indonesian Journal of Electrical Engineering and Informatics (IJEEI)
ISSN 2089-3272

Creative Commons Licence

This work is licensed under a Creative Commons Attribution 4.0 International License.

web analytics
View IJEEI Stats