Multi-Class Brain Tumor Segmentation and Classification in MRI Using a U-Net and Machine Learning Model
DOI:
https://doi.org/10.52436/1.jutif.2025.6.5.5369Keywords:
Brain Tumor Classification, Computer Vision, Machine Learning, Medical Image Processing, U-Net SegmentationAbstract
Brain tumor diagnosis remains a critical challenge in medical imaging, as accurate classification and precise localization are essential for effective treatment planning. Traditional diagnostic approaches often rely on manual interpretation of MRI scans, which can be time-consuming, subjective, and prone to variability across radiologists. To address this limitation, this study proposes a two-stage framework that integrates machine learning (ML) based classifiers for tumor type recognition and a U-Net architecture for tumor segmentation. The classifier was trained to distinguish four tumor categories: glioma, meningioma, pituitary, and no tumor, while the U-Net model was employed to delineate tumor regions at the pixel level, enabling volumetric assessment. The novelty of this research lies in its dual focus that combines classification and segmentation within a single framework, which enhances clinical applicability by offering both diagnostic and spatial insights. Experimental results demonstrated that among the evaluated classifiers, XGBoost achieved the highest accuracy of 86 percent, surpassing other models such as Random Forest, SVC, and Logistic Regression, while the U-Net model delivered consistent segmentation performance across tumor types. These findings highlight the potential of hybrid ML and deep learning solutions to improve reliability, efficiency, and objectivity in brain tumor analysis. In real-world practice, the proposed framework can serve as a valuable decision-support tool, assisting radiologists in early detection, reducing diagnostic workload, and supporting personalized treatment strategies.
Downloads
References
G. Calabrese et al., “Carbon Dots: An Innovative Tool for Drug Delivery in Brain Tumors,” Int J Mol Sci, vol. 22, no. 21, p. 11783, Oct. 2021, doi: 10.3390/ijms222111783.
Y. Xie et al., “Convolutional Neural Network Techniques for Brain Tumor Classification (from 2015 to 2022): Review, Challenges, and Future Perspectives,” Diagnostics, vol. 12, no. 8, p. 1850, Jul. 2022, doi: 10.3390/diagnostics12081850.
A. I. R. Maas et al., “Traumatic brain injury: progress and challenges in prevention, clinical care, and research,” Lancet Neurol, vol. 21, no. 11, pp. 1004–1060, Nov. 2022, doi: 10.1016/S1474-4422(22)00309-X.
F. Mo, A. Pellerino, R. Soffietti, and R. Rudà, “Blood-brain barrier in brain tumors: Biology and clinical relevance,” Dec. 01, 2021, MDPI. doi: 10.3390/ijms222312654.
M. Adel Fahmideh and M. E. Scheurer, “Pediatric Brain Tumors: Descriptive Epidemiology, Risk Factors, and Future Directions,” Cancer Epidemiology, Biomarkers & Prevention, vol. 30, no. 5, pp. 813–821, May 2021, doi: 10.1158/1055-9965.EPI-20-1443.
D. Bartusik-Aebisher et al., “The Use of Photodynamic Therapy in the Treatment of Brain Tumors—A Review of the Literature,” Oct. 01, 2022, MDPI. doi: 10.3390/molecules27206847.
M. J. Lakshmi and S. Nagaraja Rao, “Brain tumor magnetic resonance image classification: a deep learning approach,” Soft comput, vol. 26, no. 13, pp. 6245–6253, Jul. 2022, doi: 10.1007/s00500-022-07163-z.
N. R. D. Cahyo, C. A. Sari, E. H. Rachmawanto, C. Jatmoko, R. R. A. Al-Jawry, and M. A. Alkhafaji, “A Comparison of Multi Class Support Vector Machine vs Deep Convolutional Neural Network for Brain Tumor Classification,” in 2023 International Seminar on Application for Technology of Information and Communication (iSemantic), IEEE, Sep. 2023, pp. 358–363. doi: 10.1109/iSemantic59612.2023.10295336.
Y. Anagun, “Smart brain tumor diagnosis system utilizing deep convolutional neural networks,” Multimed Tools Appl, 2023, doi: 10.1007/s11042-023-15422-w.
S. Shanthi, S. Saradha, J. A. Smitha, N. Prasath, and H. Anandakumar, “An efficient automatic brain tumor classification using optimized hybrid deep neural network,” International Journal of Intelligent Networks, vol. 3, pp. 188–196, Jan. 2022, doi: 10.1016/j.ijin.2022.11.003.
B. Arjmand et al., “Machine Learning: A New Prospect in Multi-Omics Data Analysis of Cancer,” Jan. 27, 2022, Frontiers Media S.A. doi: 10.3389/fgene.2022.824451.
A. A. Khan, A. A. Laghari, and S. A. Awan, “Machine Learning in Computer Vision: A Review,” EAI Endorsed Transactions on Scalable Information Systems, vol. 8, no. 32, pp. 1–11, 2021, doi: 10.4108/eai.21-4-2021.169418.
J. Wen, Z. Zhang, Y. Lan, Z. Cui, J. Cai, and W. Zhang, “A survey on federated learning: challenges and applications,” International Journal of Machine Learning and Cybernetics, vol. 14, no. 2, pp. 513–535, Feb. 2023, doi: 10.1007/s13042-022-01647-y.
M. Tsuneki, “Deep learning models in medical image analysis,” J Oral Biosci, vol. 64, no. 3, pp. 312–320, Sep. 2022, doi: 10.1016/j.job.2022.03.003.
J. Gupta, S. Pathak, and G. Kumar, “Deep Learning (CNN) and Transfer Learning: A Review,” in Journal of Physics: Conference Series, Institute of Physics, 2022. doi: 10.1088/1742-6596/2273/1/012029.
Y. Wang et al., “ID-Seg: an infant deep learning-based segmentation framework to improve limbic structure estimates,” Brain Inform, vol. 9, no. 1, p. 12, Dec. 2022, doi: 10.1186/s40708-022-00161-9.
R. Azad et al., “Medical Image Segmentation Review: The Success of U-Net,” IEEE Trans Pattern Anal Mach Intell, vol. 46, no. 12, pp. 10076–10095, Dec. 2024, doi: 10.1109/TPAMI.2024.3435571.
C. Zhang, X. Deng, and S. H. Ling, “Next-Gen Medical Imaging: U-Net Evolution and the Rise of Transformers,” Sensors, vol. 24, no. 14, p. 4668, Jul. 2024, doi: 10.3390/s24144668.
Kavitha Soppari, Sravya Putnala, Sai Srujana Borala, and Sai Kumar Gondhi, “A survey on brain MRI segmentation,” World Journal of Advanced Research and Reviews, vol. 21, no. 3, pp. 1702–1710, Mar. 2024, doi: 10.30574/wjarr.2024.21.3.0813.
T. Trongtirakul, S. Agaian, and A. Oulefki, “Automated tumor segmentation in thermographic breast images,” Mathematical Biosciences and Engineering, vol. 20, no. 9, pp. 16786–16806, 2023, doi: 10.3934/mbe.2023748.
Y. Shin, M. Lee, Y. Lee, K. Kim, and T. Kim, “Artificial Intelligence-Powered Quality Assurance: Transforming Diagnostics, Surgery, and Patient Care—Innovations, Limitations, and Future Directions,” Life, vol. 15, no. 4, p. 654, Apr. 2025, doi: 10.3390/life15040654.
R. R. Ali et al., “Learning Architecture for Brain Tumor Classification Based on Deep Convolutional Neural Network: Classic and ResNet50,” Diagnostics, vol. 15, no. 5, p. 624, Mar. 2025, doi: 10.3390/diagnostics15050624.
H. P. Hadi, E. Faisal, and E. H. Rachmawanto, “Brain tumor segmentation using multi-level Otsu thresholding and Chan-Vese active contour model,” TELKOMNIKA (Telecommunication Computing Electronics and Control), vol. 20, no. 4, p. 825, Aug. 2022, doi: 10.12928/telkomnika.v20i4.21679.
Q. Jia and H. Shu, “BiTr-Unet: A CNN-Transformer Combined Network for MRI Brain Tumor Segmentation,” 2022, pp. 3–14. doi: 10.1007/978-3-031-09002-8_1.
K. Avazov, S. Mirzakhalilov, S. Umirzakova, A. Abdusalomov, and Y. I. Cho, “Dynamic Focus on Tumor Boundaries: A Lightweight U-Net for MRI Brain Tumor Segmentation,” Bioengineering, vol. 11, no. 12, p. 1302, Dec. 2024, doi: 10.3390/bioengineering11121302.
Y. Wong, E. L. M. Su, C. F. Yeong, W. Holderbaum, and C. Yang, “Brain tumor classification using MRI images and deep learning techniques,” PLoS One, vol. 20, no. 5, p. e0322624, May 2025, doi: 10.1371/journal.pone.0322624.
S. E. Nassar, I. Yasser, H. M. Amer, and M. A. Mohamed, “A robust MRI-based brain tumor classification via a hybrid deep learning technique,” J Supercomput, vol. 80, no. 2, pp. 2403–2427, Jan. 2024, doi: 10.1007/s11227-023-05549-w.
Sartaj, “Brain Tumor Classification (MRI).” Accessed: Aug. 13, 2024. [Online]. Available: https://www.kaggle.com/datasets/sartajbhuvaji/brain-tumor-classification-mri/data
I. P. Kamila, C. A. Sari, E. H. Rachmawanto, and N. R. D. Cahyo, “A Good Evaluation Based on Confusion Matrix for Lung Diseases Classification using Convolutional Neural Networks,” Advance Sustainable Science, Engineering and Technology, vol. 6, no. 1, p. 0240102, Dec. 2023, doi: 10.26877/asset.v6i1.17330.
M. M. I. Al-Ghiffary, N. R. D. Cahyo, E. H. Rachmawanto, C. Irawan, and N. Hendriyanto, “Adaptive deep learning based on FaceNet convolutional neural network for facial expression recognition,” Journal of Soft Computing, vol. 05, no. 03, pp. 271–280, 2024, doi: https://doi.org/10.52465/joscex.v5i3.450.
K. Kavin Kumar et al., “Brain Tumor Identification Using Data Augmentation and Transfer Learning Approach,” Computer Systems Science and Engineering, vol. 46, no. 2, pp. 1845–1861, 2023, doi: 10.32604/csse.2023.033927.
F. J. Moreno-Barea, J. M. Jerez, and L. Franco, “Improving classification accuracy using data augmentation on small data sets,” Expert Syst Appl, vol. 161, p. 113696, 2020.
N. P. Sutramiani, N. Suciati, and D. Siahaan, “MAT-AGCA: Multi Augmentation Technique on small dataset for Balinese character recognition using Convolutional Neural Network,” ICT Express, vol. 7, no. 4, pp. 521–529, Dec. 2021, doi: 10.1016/j.icte.2021.04.005.
D. Sheth and M. Shah, “Predicting stock market using machine learning: best and accurate way to know future stock prices,” International Journal of System Assurance Engineering and Management, vol. 14, no. 1, pp. 1–18, Feb. 2023, doi: 10.1007/s13198-022-01811-1.
O. A. Montesinos López, A. Montesinos López, and J. Crossa, Multivariate Statistical Machine Learning Methods for Genomic Prediction. Springer International Publishing, 2022. doi: 10.1007/978-3-030-89010-0.
P. Agrawal, N. Katal, and N. Hooda, “Segmentation and classification of brain tumor using 3D-UNet deep neural networks,” International Journal of Cognitive Computing in Engineering, vol. 3, pp. 199–210, Jun. 2022, doi: 10.1016/j.ijcce.2022.11.001.
U. Latif, A. R. Shahid, B. Raza, S. Ziauddin, and M. A. Khan, “An end‐to‐end brain tumor segmentation system using multi‐inception‐UNET,” Int J Imaging Syst Technol, vol. 31, no. 4, pp. 1803–1816, Dec. 2021, doi: 10.1002/ima.22585.
Additional Files
Published
How to Cite
Issue
Section
License
Copyright (c) 2025 Jackri Hendrik, Octara Pribadi, Hendri, Leony Hoki, Feriani Astuti Tarigan, Edi Wijaya, Rabei Raad Ali
This work is licensed under a Creative Commons Attribution 4.0 International License.
