Implementation of Deep Learning for Brain Tumor Classification from Magnetic Resonance Imaging
Keywords:
Brain Tumor, Convolution Neutral Network, Magnetic Resonance Imaging, MobileNetV2, ResNet50Abstract
Brain tumours are a medical problem that causes many people to die in the world due to brain cancer. Brain tumours are one of the dangerous types of brain cancer. MRI is well proven in the assessment of brain tumours, although conventional imaging has limitations in evaluating the extent of the tumour. In the field of medicine, there has been an increase in large amounts of data and traditional models cannot manage such data efficiently. So there is a need for medical image analysis that can store and analyse large medical data efficiently. This research will adopt a deep understanding transfer learning approach with four models namely VGG16, VGG19, MobileNetV2 and ResNet50 to classify 2 types of image shapes that detect whether a person has a brain tumour or not using Magnetic Resonance Imaging (MRI) data with Convolution Neutral Network (CNN). The number of datasets used is 4600 MRI images with 2 classes namely Brain Tumour and Health. The hyperparameters used are image size 224x224 pixels, training data ratio 70%, test data 30%, using Adam optimizer, learning rate 0.0001, using batch size 64 and epoch value 50. The best results in this study were obtained by MobileNetV2 architecture with an accuracy of 88.77%.
References
D. R. Nayak, N. Padhy, P. K. Mallick, M. Zymbler, and S. Kumar, “Brain Tumor Classification Using Dense Efficient-Net,” Axioms, vol. 11, no. 1, 2022, doi: 10.3390/axioms11010034.
N. Alturki et al., “Combining CNN Features with Voting Classifiers for Optimizing Performance of Brain Tumor Classification,” Cancers (Basel)., vol. 15, no. 6, pp. 1–15, 2023, doi: 10.3390/cancers15061767.
“Global Cancer Observatory, ‘Global Cancer Observatory in Indonesia,’ 2021.” [Online]. Available: https://gco.iarc.fr/today/en
I. A. El Kader, G. Xu, Z. Shuai, S. Saminu, I. Javaid, and I. S. Ahmad, “Differential deep convolutional neural network model for brain tumor classification,” Brain Sci., vol. 11, no. 3, 2021, doi: 10.3390/brainsci11030352.
Z. Liu et al., “Deep learning based brain tumor segmentation: a survey,” Complex Intell. Syst., vol. 9, no. 1, pp. 1001–1026, 2023, doi: 10.1007/s40747-022-00815-5.
A. A. Akinyelu, F. Zaccagna, J. T. Grist, M. Castelli, and L. Rundo, “Brain Tumor Diagnosis Using Machine Learning, Convolutional Neural Networks, Capsule Neural Networks and Vision Transformers, Applied to MRI: A Survey,” J. Imaging, vol. 8, no. 8, pp. 1–40, 2022, doi: 10.3390/jimaging8080205.
O. Özkaraca et al., “Multiple Brain Tumor Classification with Dense CNN Architecture Using Brain MRI Images,” Life, vol. 13, no. 2, 2023, doi: 10.3390/life13020349.
D. Rastogi, P. Johri, and V. Tiwari, “Augmentation based detection model for brain tumor using VGG 19,” Int. J. Comput. Digit. Syst., vol. 13, no. 1, pp. 1227–1237, 2023, doi: 10.12785/ijcds/1301100.
P. Gayathri, A. Dhavileswarapu, S. Ibrahim, R. Paul, and R. Gupta, “Exploring the Potential of VGG-16 Architecture for Accurate Brain Tumor Detection Using Deep Learning,” J. Comput. Mech. Manag., vol. 2, no. 2, pp. 13–22, 2023, doi: 10.57159/gadl.jcmm.2.2.23056.
A. Mahmoud et al., “Advanced Deep Learning Approaches for Accurate Brain Tumor Classification in Medical Imaging,” Symmetry (Basel)., vol. 15, no. 3, 2023, doi: 10.3390/sym15030571.
Z. Rasheed et al., “Automated Classification of Brain Tumors from Magnetic Resonance Imaging Using Deep Learning,” Brain Sci., vol. 13, no. 4, 2023, doi: 10.3390/brainsci13040602.
T. Tazin et al., “A Robust and Novel Approach for Brain Tumor Classification Using Convolutional Neural Network,” Comput. Intell. Neurosci., vol. 2021, 2021, doi: 10.1155/2021/2392395.
M. Yaqub et al., “State-of-the-art CNN optimizer for brain tumor segmentation in magnetic resonance images,” Brain Sci., vol. 10, no. 7, pp. 1–19, 2020, doi: 10.3390/brainsci10070427.
P. D. Rinanda, D. N. Aini, T. A. Pertiwi, S. Suryani, and A. J. Prakash, “Implementation of Convolutional Neural Network (CNN) for Image Classification of Leaf Disease In Mango Plants Using Deep Learning Approach,” Public Res. J. Eng. Data Technol. Comput. Sci., vol. 1, no. 2, pp. 56–61, 2024, doi: 10.57152/predatecs.v1i2.872.
T. Siddharth, B. S. Kirar, and D. K. Agrawal, “Plant Species Classification Using Transfer Learning by Pre-trained Classifier VGG-19,” IETE J. Res., vol. 68, no. 03772063, pp. 2345–3140, 2022.
W. Wardianto, F. Farikhin, and D. M. Kusumo Nugraheni, “Analisis Sentimen Berbasis Aspek Ulasan Pelanggan Restoran Menggunakan LSTM Dengan Adam Optimizer,” JOINTECS (Journal Inf. Technol. Comput. Sci., vol. 8, no. 2, p. 67, 2023, doi: 10.31328/jointecs.v8i2.4737.
S. C. Toh, S. H. Lai, M. Mirzaei, E. Z. X. Soo, and F. Y. Teo, “Sequential Data Processing for IMERG Satellite Rainfall Comparison and Improvement Using LSTM and ADAM Optimizer,” Appl. Sci., vol. 13, no. 12, 2023, doi: 10.3390/app13127237.
F. Kunstner et al., “Noise Is Not the Main Factor Behind the Gap Between SGD and Adam on Transformers, but Sign Descent Might Be,” J. Curr. Sci. Technol., vol. 13, no. 1, pp. 23–35, 2023, doi: 10.14456/jcst.2023.3.
J. Chen, R. Zhang, and Y. Liu, “An Adam-enhanced Particle Swarm Optimizer for Latent Factor Analysis,” 2023.
A. Nana Hermana, D. Rosmala, and M. Gustiana Husada, “Transfer Learning for Classification of Fruit Ripeness Using VGG16,” ACM Int. Conf. Proceeding Ser., pp. 139–146, 2021, doi: 10.1145/3450588.3450943.
K. Kusrini, S. Suputa, A. Setyanto, I. M. A. Agastya, H. Priantoro, and S. Pariyasto, “A comparative study of mango fruit pest and disease recognition,” TELKOMNIKA (Telecommunication Comput. Electron. Control., vol. 20, no. 6, p. 1264, Dec. 2022, doi: 10.12928/telkomnika.v20i6.21783.
M. Bouni, B. Hssina, K. Douzi, and S. Douzi, “Impact of Pretrained Deep Neural Networks for Tomato Leaf Disease Prediction,” J. Electr. Comput. Eng., vol. 2023, 2023, doi: 10.1155/2023/5051005.
A. Bhola, S. Verma, and P. Kumar, “A comparative analysis of deep learning models for cucumber disease classification using transfer learning,” J. Curr. Sci. Technol., vol. 13, no. 1, pp. 23–35, 2023, doi: 10.14456/jcst.2023.3.
Kamal and H. Ez-zahraouy, “A comparison between the VGG16 , VGG19 and ResNet50 architecture frameworks for classi cation of normal and CLAHE processed medical images,” Res. Sq., pp. 0–16, 2023.
J. Cui et al., “High-Precision Inversion of Shallow Bathymetry under Complex Hydrographic Conditions Using VGG19—A Case Study of the Taiwan Banks,” Remote Sens., vol. 15, no. 5, 2023, doi: 10.3390/rs15051257.
R. Mohan, S. Kadry, V. Rajinikanth, A. Majumdar, and O. Thinnukool, “Automatic Detection of Tuberculosis Using VGG19 with Seagull-Algorithm,” Life, vol. 12, no. 11, 2022, doi: 10.3390/life12111848.
Y. Huang et al., “Mobilenetv2_CA Lightweight Object Detection Network in Autonomous Driving,” Technologies, vol. 11, no. 2, pp. 9489–9510, 2023, doi: 10.3390/diagnostics13101664.
A. Ignatov et al., “PyNet-V2 Mobile: Efficient On-Device Photo Processing With Neural Networks,” Proc. - Int. Conf. Pattern Recognit., vol. 2022-Augus, pp. 677–684, 2022, doi: 10.1109/ICPR56361.2022.9956598.
S. Narduzzi, E. Turetken, J. P. Thiran, and L. A. Dunbar, “Adaptation of MobileNetV2 for Face Detection on Ultra-Low Power Platform,” Proc. - 2022 9th Swiss Conf. Data Sci. SDS 2022, pp. 1–6, 2022, doi: 10.1109/SDS54800.2022.00008.
M. Ragab et al., “COVID-19 Identification System Using Transfer Learning Technique With Mobile-NetV2 and Chest X-Ray Images,” Front. Public Heal., vol. 10, no. March, pp. 1–15, 2022, doi: 10.3389/fpubh.2022.819156.
K. Liu et al., “Prediction of Primary Tumor Sites in Spinal Metastases Using a ResNet-50 Convolutional Neural Network Based on MRI,” Cancers (Basel)., vol. 15, no. 11, 2023, doi: 10.3390/cancers15112974.
E. Suherman, B. Rahman, D. Hindarto, and H. Santoso, “Implementation of ResNet-50 on End-to-End Object Detection (DETR) on Objects,” SinkrOn, vol. 8, no. 2, pp. 1085–1096, 2023, doi: 10.33395/sinkron.v8i2.12378.
Y. Huang, L. Lin, P. Cheng, J. Lyu, R. Tam, and X. Tang, “Identifying the Key Components in ResNet-50 for Diabetic Retinopathy Grading from Fundus Images: A Systematic Investigation,” Diagnostics, vol. 13, no. 10, 2023, doi: 10.3390/diagnostics13101664.
L. Zhang, Y. Bian, P. Jiang, and F. Zhang, “A Transfer Residual Neural Network Based on ResNet-50 for Detection of Steel Surface Defects,” Appl. Sci., vol. 13, no. 9, 2023, doi: 10.3390/app13095260.
D. Hastari, S. Winanda, and A. R. Pratama, “Application of Convolutional Neural Network ResNet-50 V2 on Image Classification of Rice Plant Disease,” vol. 1, no. January, pp. 71–77, 2024.
Published
How to Cite
Issue
Section
License
Copyright (c) 2025 Nur Alfa Husna, Desvita Hendri, Muhammad Farid Wajdi, Ella Silvana Ginting, Chintya Harum Pramesthi
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
Copyright © by Author; Published by Institut Riset dan Publikasi Indonesia (IRPI)
This Public Research Journal of Engineering, Data Technology and Computer Science is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
