Comparative Evaluation of CNN and ResNet18 Architectures for MRI-Based Brain Tumor Classification Using Deep Learning
Main Article Content
Abstract
Accurate and automated classification of brain tumors from magnetic resonance imaging (MRI) scans is essential for improving diagnostic precision and supporting clinical decision-making. This study presents a deep learning-based framework that employs two convolutional neural network architectures a custom-designed CNN and a pretrained ResNet18 model for multi-class classification of brain tumors using the publicly available Kaggle MRI dataset. The dataset was preprocessed through normalization, augmentation, and resizing to ensure consistency and model generalization. Both models were trained and evaluated using an 80:20 data split, and their performance was assessed based on accuracy, precision, recall, and F1-score metrics. Experimental results demonstrate that the ResNet18 model outperforms the baseline CNN, achieving a classification accuracy of 99.7%, precision of 99.5%, and F1-score of 99.6%. These results highlight the effectiveness of transfer learning and residual connections in improving feature representation and convergence speed. These findings underscore the effectiveness of transfer learning for medical image analysis and demonstrate the potential of deep learning"based methods for reliable, automated brain tumor diagnosis. Future research should focus on extending this work to 3D MRI volumes and integrating explainable AI techniques to enhance interpretability and clinical trust.
Accurate classification of brain tumors from Magnetic Resonance Imaging (MRI) is crucial for early diagnosis, treatment planning, and improving patient outcomes. However, manual interpretation of MRI scans is time-consuming and susceptible to diagnostic inconsistencies. This study presents a comparative evaluation of a custom Convolutional Neural Network (CNN) and a transfer learning"based ResNet18 model for automated brain tumor classification using the Kaggle Brain Tumor MRI Dataset, which includes four diagnostic categories: glioma, meningioma, pituitary tumor, and no tumor. Both models were trained and validated under identical preprocessing and experimental conditions to ensure fair comparison.
Comprehensive preprocessing, including normalization, augmentation, and stratified splitting (70% training, 20% validation, 10% testing), was applied to enhance data uniformity and generalization. The CNN model was trained from scratch, whereas the ResNet18 model is fine-tuned using pretrained ImageNet weights to leverage transfer learning. Performance was evaluated using accuracy, precision, recall, F1-score, and AUC metrics, supplemented by visual diagnostics such as confusion matrices, accuracy"loss curves, and F1-confidence plots.
The ResNet18 model achieved superior performance, with a test accuracy of 99.54%, precision of 0.98, recall of 0.99, F1-score of 0.99, and AUC of 0.992, outperforming the custom CNN, which attained 97.84% accuracy, precision of 0.94, recall of 0.95, F1-score of 0.94, and AUC of 0.975. Confusion matrix analysis indicated that both models accurately classified all tumor types, though minor misclassifications were observed between pituitary and no-tumor categories. ResNet18 exhibited faster convergence, smoother loss reduction, and greater robustness to intensity variations due to its residual connections and pretrained feature representations.
Article Details
The Medical Research Archives grants authors the right to publish and reproduce the unrevised contribution in whole or in part at any time and in any form for any scholarly non-commercial purpose with the condition that all publications of the contribution include a full citation to the journal as published by the Medical Research Archives.
References
2. Sajjanar, R., U. D. Dixit, and V. K. Vagga. 2024. “Advancements in Hybrid Approaches for Brain Tumor Segmentation.” Multimedia Tools and Applications 83 (10): 30505–30539. https://doi.org/10.1007/s11042-023-16654-6.
3. Rasool, N., and J. I. Bhat. 2025. “Brain Tumour Detection Using Machine and Deep Learning.” Multimedia Tools and Applications 84 (13): 11551–11604. https://doi.org/10.1007/s11042-024-19333-2.
4. Golkarieh, A., et al. 2025. “Breakthroughs in Brain Tumor Detection.” Computer and Decision Making 2: 708–722. https://doi.org/10.1016/j.cdm.2025.708722.
5. Umarani, C. M., et al. 2024. “Advancements in Deep Learning Techniques for Brain Tumor Segmentation: A Survey.” Informatics in Medicine Unlocked 50: 101576. https://doi.org/10.1016/j.imu.2024.101576.
6. Hassan, F., et al. 2025. “A Systematic Review of Deep Learning-Based Segmentation Techniques.” Digital Health 11: 20552076251380645. https://doi.org/10.1177/20552076251380645.
7. Jin, C., T. F. Ng, and H. Ibrahim. 2025. “Advancements in Semi-Supervised Deep Learning for Brain Tumor Segmentation.” AI 6 (7): 153. https://doi.org/10.3390/ai6070153.
8. Bouhafra, S., and H. El Bahi. 2025. “Deep Learning Approaches for Brain Tumor Detection and Classification.” Journal of Imaging Informatics in Medicine 38 (3): 1403–1433. https://doi.org/10.1007/s10278-024-01283-8.
9. Bhalodiya, J. M., et al. 2022. “Magnetic Resonance Image-Based Brain Tumour Segmentation Methods.” Digital Health 8: 20552076221074122. https://doi.org/10.1177/20552076221074122.
10. Kamnitsas, K., et al. 2017. “Efficient Multi-Scale 3D CNN with Fully Connected CRF for Brain Lesion Segmentation.” Medical Image Analysis 36: 61–78. https://doi.org/10.1016/j.media.2016.10.004.
11. Isensee, F., et al. 2021. “nnU-Net: A Self-Adapting Framework for U-Net-Based Medical Image Segmentation.” Nature Methods 18 (2): 203–211. https://doi.org/10.1038/s41592-020-01008-z.
12. Bakas, S., et al. 2018. “Identifying the Best Machine Learning Algorithms for Brain Tumor Segmentation.” Frontiers in Neuroscience 12: 155. https://doi.org/10.3389/fnins.2018.00155.
13. Zhou, T., et al. 2021. “Glioma Grading Using a Cross-Modal 3D CNN on Multisequence MRI.” IEEE Journal of Biomedical and Health Informatics 25 (7): 2767–2778. https://doi.org/10.1109/JBHI.2020.3040643.
14. Afshar, P., et al. 2019. “Capsule Networks for Brain Tumor Classification Based on MRI Images.” Journal of Medical Systems 43: 1–13. https://doi.org/10.1007/s10916-019-1230-3.
15. Azeez, O., and A. Abdulazeez. 2025. “Classification of Brain Tumor Based on Machine Learning Algorithms: A Review.” Journal of Applied Science and Technology Trends 6 (1): 115. https://doi.org/10.38094/jastt60101.
16. Ghadimi, D. J., et al. 2025. “Deep Learning-Based Techniques in Glioma Brain Tumor Segmentation.” Journal of Magnetic Resonance Imaging 61 (3): 1094–1109. https://doi.org/10.1002/jmri.29543.
17. Menze, B. H., et al. 2015. “The Multimodal Brain Tumor Image Segmentation Benchmark (BRATS).” IEEE Transactions on Medical Imaging 34 (10): 1993–2024. https://doi.org/10.1109/TMI.2014.2377694.
18. Ronneberger, O., Fischer, P., & Brox, T. (2015). U-Net: Convolutional networks for biomedical image segmentation. In Lecture Notes in Computer Science (Vol. 9351, pp. 234–241). MICCAI. https://doi.org/10.1007/978-3-319-24574-4_28
19. Kamnitsas, K., et al. (2017). Efficient multi-scale 3D CNN with fully connected CRF for accurate brain lesion segmentation. Medical Image Analysis, 36, 61–78. https://doi.org/10.1016/j.media.2016.10.004
20. Isensee, F., et al. (2021). nnU-Net: A self-adapting framework for U-Net-based medical image segmentation. Nature Methods, 18(2), 203–211. https://doi.org/10.1038/s41592-020-01008-z
21. Havaei, M., Davy, A., Warde-Farley, D., et al. (2017). Brain tumor segmentation with deep neural networks. Medical Image Analysis, 35, 18–31. https://doi.org/10.1016/j.media.2016.05.004
22. Selvaraju, R. R., Cogswell, M., Das, A., et al. (2019). Grad-CAM: Visual explanations from deep networks via gradient-based localization. International Journal of Computer Vision, 128, 336–359. https://doi.org/10.1007/s11263-019-01228-7
23. Zhou, Z., Siddiquee, M. M. R., Tajbakhsh, N., & Liang, J. (2019). UNet++: A nested U-Net architecture for medical image segmentation. arXiv preprint arXiv:1807.10165. https://arxiv.org/abs/1807.10165
24. Zeineldin, R. A., et al. (2020). DeepSeg: A deep neural network framework for automatic brain lesion detection and segmentation. International Journal of Computer Assisted Radiology and Surgery, 15, 1187–1198. https://doi.org/10.1007/s11548-020-02186-z
25. Bhandari, A., et al. (2020). Convolutional neural networks for brain tumour segmentation: A review. Insights into Imaging, 11, 121. https://doi.org/10.1186/s13244-020-00869-4
26. Litjens, G., et al. (2017). A survey on deep learning in medical image analysis. Medical Image Analysis, 42, 60–88. https://doi.org/10.1016/j.media.2017.07.005
27. Fan, Q., G.W. Zhang, and B. Peng. “Tumor Imaging of a Novel Ho3+-Based Biocompatible NIR Fluorescent Fluoride Nanoparticle.” Journal of Luminescence 235 (2021). https://doi.org/10.1016/j.jlumin.2021.118007
28. Gandhi, Deep B, Anurag Gottipati, Wenxin Tu, Ariana Familiar, Shuvanjan Haldar, Neda Khalili, Paarth Jain, et al. “Img-09. A Deep Learning-Based Approach for Brain Tissue Extraction Using Multi- and Single-Parametric Mri in Pediatrics.” Neuro-Oncology 26, no. Suppl 4 (2024): 0. https://doi.org/10.1093/neuonc/noae064.346.
29. Kothadiya, Deep, Amjad Rehman, Bayan AlGhofaily, Chintan Bhatt, Noor Ayesha, and Tanzila Saba. “VGX: VGG19-Based Gradient Explainer Interpretable Architecture for Brain Tumor Detection in Microscopy Magnetic Resonance Imaging (MMRI).” Microscopy Research and Technique 88, no. 5 (2025): 1544–54. https://doi.org/10.1002/jemt.24809.
30. Yao, Guoyan, Yan Huang, Xiaojing Shang, Lei Guo, Jieping Feng, Zhongqiu Lai, Weikang Huang, Jianye Lu, Lijun Chen, and Minan zheng. “Prediction of Intraductal Cancer Microinfiltration Based on the Hierarchical Fusion of Peri-Tumor Imaging Histology and Dual View Deep Learning.” BMC Cancer 25 (2025): 1–13. https://doi.org/10.1186/s12885-025-15054-3.
31. Afridi, Muhammad, Abhi Jain, Mariam Aboian, and Seyedmehdi Payabvash. “Brain Tumor Imaging: Applications of Artificial Intelligence.” Seminars in Ultrasound, CT, and MRI 43, no. 2 (2022): 153–69. https://doi.org/10.1053/j.sult.2022.02.005.
32. Shen, Lei, Bo Dai, Shewei Dou, Fengshan Yan, Tianyun Yang, and Yaping Wu. “Estimation of TP53 Mutations for Endometrial Cancer Based on Diffusion-Weighted Imaging Deep Learning and Radiomics Features.” BMC Cancer 25 (2025): 1–12. https://doi.org/10.1186/s12885-025-13424-5.
33. Goswami, Mayank. “Deep Learning Models for Benign and Malign Ocular Tumor Growth Estimation.” Computerized Medical Imaging and Graphics 93 (2021). https://doi.org/10.1016/j.compmedimag.2021.101986.
http://orcid.org/0000-0001-6255-3848