Brain Tumor Classification Using Fine-Tuned Transfer Learning
Mustafa Suban Dut1, Meriç Çetin2*
1Department of Computer Engineering, Pamukkale University, Denizli, Turkiye
2Department of Computer Engineering, Pamukkale University, Denizli, Turkiye
* Corresponding author: mcetin@pau.edu.tr
Presented at the International Symposium on AI-Driven Engineering Systems (ISADES2025), Tokat, Turkiye, Jun 19, 2025
SETSCI Conference Proceedings, 2025, 22, Page (s): 39-42 , https://doi.org/10.36287/setsci.22.10.001
Published Date: 10 July 2025
Brain tumor is a progressive disease that significantly affects human life. These diseases can reduce the quality of life by causing losses in motor and cognitive functions. Since the chance of treatment decreases when not diagnosed early, advanced technologies are needed for diagnosis and prognosis. In this study, the applicability of transfer learning-based approaches for the classification of brain tumors was investigated. For this purpose, the classification performance of a proposed convolutional neural network model was compared with the performances of pre-trained transfer learning models such as VGG16, ResNet50 and InceptionV3. The designed models were trained on two different brain tumor datasets with two and four classes. The obtained results were analyzed by comparing the performance parameters of each model on different datasets. As a result, pre-trained transfer learning models supported by fine-tuning applications showed high performance in both two-class and four-class brain tumor classification tasks.
Keywords - Brain tumor classification, deep learning, transfer learning, fine tuning
[1] S. M. Rasa, M. M. Islam, M. A. Talukder, M. A. Uddin, M. Khalid, M. Kazi and M. Z. Kazi, “Brain tumor classification using fine-tuned transfer learning models on magnetic resonance imaging (MRI) images,” Digital Health, 10, 20552076241286140, 2024.
[2] J. Amin, M. Sharif, M. Raza, T. Saba, R. Sial and S.A. Shad, “Brain tumor detection: A long short-term memory (lstm)-based learning model,” Neural Computing and Applications, 32(20), pp: 15965-15973, 2020.
[3] A. Saleh, R. Sukaik and S. S. Abu-Naser, “Brain tumor classification using deep learning,” International Conference on Assistive and Rehabilitation Technologies, pp: 131-136, 2020.
[4] H.A. Khan, W. Jue, M. Mushtaq and M. U. Mushtaq, “Brain tumor classification in mri image using convolutional neural network,” Math. Biosci. Eng, 17, pp: 6203, 2020.
[5] F. D. E. Baran and M. Cetin, “Evrişimli sinir ağları kullanarak beyin tümörü sınıflandırması,” Researcher, 4(02), 76-83, 2024.
[6] Y. Sun, S. Zhu, K. Ma, W. Liu, Y. Yue, G. Hu, H. Lu and W. Chen, “Identification of 12 cancer types through genome deep learning,” Scientific Reports, 9(1), 17256, 2019.
[7] P. Saxena, A. Maheshwari and S. Maheshwari, “Predictive modeling of brain tumor: A deep learning approach,” Innovations in Computational Intelligence and Computer Vision, Springer, pp: 275-285, 2021.
[8] H. Mohsen, E. S. A. El-Dahshan, E. S. M. El-Horbaty and A. B. M. Salem, “Classification using deep learning neural networks for brain tumors,” Future Computing and Informatics Journal, 3(1), pp: 68-71, 2018.
[9] S. Deepak and P. Ameer, “Brain tumor classification using deep cnn features via transfer learning,” Computers in Biology and Medicine, 111, pp: 103345, 2019.
[10] A. Rehman, S. Naz, M. I. Razzak, F. Akram and M. Imran, “A deep learning-based framework for automatic brain tumors classification using transfer learning,” Circuits, Systems, and Signal Processing, 39(2), pp: 757-775, 2020.
[11] V. Romeo, R. Cuocolo, C. Ricciardi, L. Ugga, S. Cocozza, F. Verde, A. Stanzione, V. Napolitano, D. Russo, G. Improta et al. “Prediction of tumor grade and nodal status in oropharyngeal and oral cavity squamous-cell carcinoma using a radiomic approach,” Anticancer Research, 40(1), pp: 271-280, 2020.
[12] S. Saeedi, S. Rezayi, H. Keshavarz, and S. Niakan Kalhori, “MRI-based brain tumor detection using convolutional deep learning methods and chosen machine learning techniques,” BMC Medical Informatics and Decision Making, 23(1), 16, 2023.
[13] S. Hossain, A. Chakrabarty, T. R. Gadekallu, M. Alazab and M. J. Piran, “Vision transformers, ensemble model, and transfer learning leveraging explainable AI for brain tumor detection and classification,” IEEE Journal of Biomedical and Health Informatics, 28(3), 1261-1272, 2023.
[14] Z. N. K. Swati, Q. Zhao, M. Kabir, F. Ali, Z. Ali, S. Ahmed and J. Lu, “Content-based brain tumor retrieval for MR images using transfer learning,” IEEE Access, 7, 17809-17822, 2019.
[15] R. Chelghoum, A. Ikhlef, A. Hameurlaine and S. Jacquir, “Transfer learning using convolutional neural network architectures for brain tumor classification from MRI images,” In IFIP International Conference on Artificial Intelligence Applications and Innovations (pp. 189-200). Cham: Springer International Publishing, 2020.
[16] H. A. Khan, W. Jue, M. Mushtaq and M. U. Mushtaq, “Brain tumor classification in MRI image using convolutional neural network,” Mathematical Biosciences and Engineering, 2021.
[17] J. S. Paul, A. J. Plassard, B. A. Landman and D. Fabbri, “Deep learning for brain tumor classification,” In Medical Imaging 2017: Biomedical Applications in Molecular, Structural, and Functional Imaging (Vol. 10137, pp. 253-268), 2017.
[18] G. Litjens, T. Kooi, B. E. Bejnordi, A. A. A. Setio, F. Ciompi, M. Ghafoorian, and C. I. Sánchez, “A survey on deep learning in medical image analysis,” Medical Image Analysis, 42, 60-88, 2017.
[19] (2024) A. Hamada. Br35H: Brain Tumor Detection 2020. [Online]. Available: https://www.kaggle.com/datasets/ahmedhamada0/brain-tumor-detection
[20] S. Bhuvaji, Brain tumor classification (mri). [Online]. Available: https://www.kaggle.com/datasets/sartajbhuvaji/brain-tumor-classification-mri
|
This is an Open Access article distributed under the terms of the Creative Commons Attribution License 4.0, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited. |
