Application of Deep Learning for the Classification of Brain Tumor Magnetic Resonance Imaging Images
DOI:
https://doi.org/10.57152/ijatis.v3i2.3089Keywords:
Brain Tumor, Convolutional Neural Network, DenseNet121, EfficientNet-B2, MRI, ResNet-50Abstract
Accurate and timely classification of brain tumors from Magnetic Resonance Imaging (MRI) is essential for supporting clinical diagnosis and treatment planning. This study presents a comparative evaluation of transfer learning-based Convolutional Neural Network (CNN) architectures, including ResNet-50, DenseNet121, and EfficientNet-B2, for binary brain tumor MRI classification (Glioma and Meningioma). The dataset was augmented to improve model generalization, and the data were split into training and test sets using an 80:20 hold-out split. The models were trained using Adam and RMSProp optimizers with different learning rates, and their performance was evaluated using accuracy, precision, recall, and F1-score. Experimental results demonstrate that data augmentation significantly enhances classification performance across all evaluated architectures. Among the tested models, ResNet-50 with the RMSProp optimizer and a learning rate of 0.001 achieved the best performance, yielding the lowest training and validation losses and perfect classification results on the test set, with accuracy, precision, recall, and F1-score approaching 100%. DenseNet121 and EfficientNet-B2 also achieved excellent performance but were slightly inferior to ResNet-50. These findings indicate that transfer learning with ResNet-50 and RMSProp provides an effective and reliable solution for automated brain tumor MRI image classification
References
M. M. Badža and M. C. Barjaktarovi?, “Classification of brain tumors from mri images using a convolutional neural network,” Applied Sciences (Switzerland), vol. 10, no. 6, 2020, doi: 10.3390/app10061999.
D. Prasetya, A. Layyinah, A. Maftuchan, S. Putri, E. Rosita, and A. I. Nurjanah, “Studi Kasus Penderita Kanker di Ibu Kota Jakarta 2023,” Konsekuensi Finanasial Pengobatan Kanker Di Indonesia, p. 101, 2023.
Oneonco, “Tumor Otak.” [Online]. Available: https://oneonco.co.id/blog/tumor-otak
P. Afshar, A. Mohammadi, and K. N. Plataniotis, “Concordia Institute for Information Systems Engineering , Concordia University , Montreal , QC , Canada Department of Electrical and Computer Engineering , University of Toronto , Toronto , ON , Canada Emails?: { p afs , arashmoh } @ encs . concordia . ca,” 2018 25th IEEE International Conference on Image Processing (ICIP), pp. 3129–3133, 2018.
H. E. Kim, A. Cosa-Linan, N. Santhanam, M. Jannesari, M. E. Maros, and T. Ganslandt, “Transfer learning for medical image classification: a literature review,” BMC Med. Imaging, vol. 22, no. 1, pp. 1–13, 2022, doi: 10.1186/s12880-022-00793-7.
M. C. Daniel and L. M. Ruxandra, “Brain Tumor Classification Using Pretrained Convolutional Neural Networks,” 2021 16th International Conference on Engineering of Modern Electric Systems, EMES 2021 - Proceedings, vol. 11, no. September, pp. 1457–1461, 2021, doi: 10.1109/EMES52337.2021.9484102.
R. Haridas and J. R L, “Convolutional Neural Networks: A Comprehensive Survey,” International Journal of Applied Engineering Research, vol. 14, no. 3, p. 780, 2019, doi: 10.37622/ijaer/14.3.2019.780-789.
A. Çinar and M. Yildirim, “Detection of tumors on brain MRI images using the hybrid convolutional neural network architecture,” Med. Hypotheses, vol. 139, no. February, p. 109684, 2020, doi: 10.1016/j.mehy.2020.109684.
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, doi: 10.1007/s40998-021-00426-9.
A. Khan, A. Sohail, U. Zahoora, and A. S. Qureshi, “A survey of the recent architectures of deep convolutional neural networks,” Artif. Intell. Rev., vol. 53, no. 8, pp. 5455–5516, 2020, doi: 10.1007/s10462-020-09825-6.
W. Islam, M. Jones, R. Faiz, N. Sadeghipour, Y. Qiu, and B. Zheng, “Improving Performance of Breast Lesion Classification Using a ResNet50 Model Optimized with a Novel Attention Mechanism,” Tomography, vol. 8, no. 5, pp. 2411–2425, 2022, doi: 10.3390/tomography8050200.
A. D. Saputra, D. Hindarto, and H. Santoso, “Disease Classification on Rice Leaves using DenseNet121, DenseNet169, DenseNet201,” Sinkron, vol. 8, no. 1, pp. 48–55, 2023, doi: 10.33395/sinkron.v8i1.11906.
F. Salim, F. Saeed, S. Basurra, S. N. Qasem, and T. Al-Hadhrami, “DenseNet-201 and Xception Pre-Trained Deep Learning Models for Fruit Recognition,” Electronics (Switzerland), vol. 12, no. 14, 2023, doi: 10.3390/electronics12143132.
H. Farman, J. Ahmad, B. Jan, Y. Shahzad, M. Abdullah, and A. Ullah, “Efficientnet-based robust recognition of peach plant diseases in field images,” Computers, Materials and Continua, vol. 71, no. 1, pp. 2073–2089, 2022, doi: 10.32604/cmc.2022.018961.
V. Ravi, V. Acharya, and M. Alazab, “A multichannel EfficientNet deep learning-based stacking ensemble approach for lung disease detection using chest X-ray images,” Cluster Comput., vol. 26, no. 2, pp. 1181–1203, 2023, doi: 10.1007/s10586-022-03664-6.
F. Mehmood, S. Ahmad, and T. K. Whangbo, “An Efficient Optimization Technique for Training Deep Neural Networks,” Mathematics, vol. 11, no. 6, 2023, doi: 10.3390/math11061360.
R. Zaheer and H. Shaziya, “2019 Third International Conference on Inventive Systems and Control (ICISC).,” 2019 Third International Conference on Inventive Systems and Control (ICISC), no. Icisc, pp. 536–539, 2019.
