RAG⁴-Unet: An Approach for Recognition and Segmentation of Brain Tumor in MRI Scans
Ameer Hamza, Robertas Damaševičius
DOI: http://dx.doi.org/10.15439/2025F7399
Citation: Position Papers of the 20th Conference on Computer Science and Intelligence Systems, M. Bolanowski, M. Ganzha, L. Maciaszek, M. Paprzycki, D. Ślęzak (eds). ACSIS, Vol. 44, pages 41–48 (2025)
Abstract. We propose a novel U-net architecture, RAG${}^4$-Unet, based on residual attention gated for brain tumor segmentation, Swin transformer for classification task, and Yolo11 for tumor detection. For the experiments, the Figshare dataset is employed and the proposed architecture achieved 91.37\% Dice for tumor segmentation task, and Swin transformer achieved 91.74\% classification accuracy. The Yolo11 gained 89.6\% of detection precision. Comparative evaluation with the SOTA techniques reveals that the proposed architecture outperformed the existing methods and Yolo11. The proposed architecture improved the tumor boundary detection, making it a promising solution for brain tumor recognition and segmentation.
References
- T. Rahman, M. S. Islam, and J. Uddin, “Mri-based brain tumor classification using a dilated parallel deep convolutional neural network,” Digital, vol. 4, no. 3, pp. 529–554, 2024.
- H. A. Munira and M. S. Islam, “Hybrid deep learning models for multi-classification of tumour from brain mri,” J Inf Syst Eng Bus Intell, vol. 8, pp. 162–74, 2022.
- N. Elazab, W. A. Gab-Allah, and M. Elmogy, “A multi-class brain tumor grading system based on histopathological images using a hybrid yolo and resnet networks,” Scientific Reports, vol. 14, no. 1, p. 4584, 2024.
- P. Kuppler, P. Strenge, B. Lange, S. Spahr-Hess, W. Draxinger, C. Hagel, D. Theisen-Kunde, R. Brinkmann, R. Huber, V. Tronnier, et al., “Microscope-integrated optical coherence tomography for in vivo human brain tumor detection with artificial intelligence,” Journal of Neurosurgery, vol. 1, no. aop, pp. 1–9, 2024.
- Z. Rasheed, Y.-K. Ma, I. Ullah, M. Al-Khasawneh, S. S. Almutairi, and M. Abohashrh, “Integrating convolutional neural networks with attention mechanisms for magnetic resonance imaging-based classification of brain tumors,” Bioengineering, vol. 11, no. 7, p. 701, 2024.
- S. Saket, Y. Nilipour, R. Taherian, and N. F. Marnaanni, “Evaluation of radiographic, neuropathological, and demographic findings in children aged 1 to 18 years with brain tumor,” Novelty in Biomedicine, vol. 12, no. 2, pp. 55–59, 2024.
- M. S. I. Khan, A. Rahman, T. Debnath, M. R. Karim, M. K. Nasir, S. S. Band, A. Mosavi, and I. Dehzangi, “Accurate brain tumor detection using deep convolutional neural network,” Computational and Structural Biotechnology Journal, vol. 20, pp. 4733–4745, 2022.
- D. Reyes and J. Sánchez, “Performance of convolutional neural networks for the classification of brain tumors using magnetic resonance imaging,” Heliyon, vol. 10, no. 3, 2024.
- K. Singh, A. Kaur, and P. Kaur, “Computer aided detection of brain tumors using convolutional neural network based analysis of mri data,” 2023.
- Y. Zhang, H. C. Ngo, Y. Zhang, N. F. A. Yusof, and X. Wang, “Imaging segmentation of brain tumors based on the modified u-net method,” Information Technology and Control, vol. 53, no. 4, p. 1074 – 1087, 2024.
- R. Ahsan, I. Shahzadi, F. Najeeb, and H. Omer, “Brain tumor detection and segmentation using deep learning,” Magnetic Resonance Materials in Physics, Biology and Medicine, pp. 1–10, 2024.
- T. Arumaiththurai and B. Mayurathan, “The effect of deep learning and machine learning approaches for brain tumor recognition,” in 2021 10th International Conference on Information and Automation for Sustainability (ICIAfS), pp. 185–190, IEEE, 2021.
- J. Alyami, A. Rehman, F. Almutairi, A. M. Fayyaz, S. Roy, T. Saba, and A. Alkhurim, “Tumor localization and classification from mri of brain using deep convolution neural network and salp swarm algorithm,” Cognitive Computation, vol. 16, no. 4, pp. 2036–2046, 2024.
- A. A. Asiri, A. Shaf, T. Ali, M. Aamir, M. Irfan, and S. Alqahtani, “Enhancing brain tumor diagnosis: an optimized cnn hyperparameter model for improved accuracy and reliability,” PeerJ Computer Science, vol. 10, p. e1878, 2024.
- I. S. Rajput, A. Gupta, V. Jain, and S. Tyagi, “A transfer learning-based brain tumor classification using magnetic resonance images,” Multimedia Tools and Applications, vol. 83, no. 7, pp. 20487–20506, 2024.
- K. Shiny, “Brain tumor segmentation and classification using optimized u-net,” The Imaging Science Journal, vol. 72, no. 2, pp. 204–219, 2024.