Segmentation of Medical MRI Images Using Nested U-Net with Attention Mechanism and Fuzzy Pooling

Section: Research Paper
Issue
Vol. 30 No. 2 (2025): Volume 30 Issue 2
Published
Sep 1, 2025
Pages
104-115

Abstract

Segmentation of medical images is a crucial step in the diagnosis and treatment of various diseases, particularly when analyzing magnetic resonance imaging (MRI) scans. Despite significant progress, accurate segmentation remains challenging due to the complexity of anatomical structures, ambiguous boundaries, and variations in image quality. This paper proposes an enhanced Nested U-Net structure that incorporates an attention mechanism and fuzzy pooling to improve retail performance. Nested U-Net benefits from extensive skip connections to capture multi-level contextual information, while the attention mechanism enhances the model's ability to focus on relevant features and noise suppression. In addition, we replace the traditional extreme pooling layers with fuzzy pooling, which enables the network to handle spatial ambiguity more effectively and produce more accurate and refined retail maps. Experimental results on publicly available MRI datasets show that the proposed model achieves remarkable improvements, with a 4.97% increase in the dice coefficient, a 2.76% improvement in accuracy, and a 4.25% increase in recall compared to enhanced U-Net structures. Furthermore, significant improvements have been observed in Hausdorff's distance measures.

References

  1. Y. M. A. Mohammed, S. El Garouani, and I. Jellouli, A survey of methods for brain tumor segmentation-based MRI images, J. Comput. Des. Eng., vol. 10, no. 1, pp. 266293, 2023, doi: 10.1093/jcde/qwac141.
  2. S. R. Gunasekara, H. N. T. K. Kaldera, and M. B. Dissanayake, A systematic approach for MRI brain tumor localization and segmentation using deep learning and active contouring, J. Healthc. Eng., vol. 2021, pp. 113, 2021, doi: 10.1155/2021/6695108.
  3. M. A. Ottom, H. A. Rahman, and I. D. Dinov, Znet: Deep learning approach for 2D MRI brain tumor segmentation, IEEE J. Transl. Eng. Health Med., vol. 10, no. 1, pp. 18, 2022, doi: 10.1109/JTEHM.2022.3176737.
  4. X. Guan et al., 3D AGSE-VNet: an automatic brain tumor MRI data segmentation framework, BMC Med. Imaging, vol. 22, no. 1, pp. 118, 2022, doi: 10.1186/s12880-021-00728-8.
  5. A. Al-Saegh, A. Daood, and M. H. Ismail, Dual Optimization of Deep CNN for Motor Imagery EEG Tasks Classification, Diyala J. Eng. Sci., vol. 17, no. 4, pp. 7591, 2024, doi: 10.24237/djes.2024.17405.
  6. B. Chen, Y. Liu, Z. Zhang, G. Lu, and A. W. K. Kong, TransAttUnet: Multi-Level Attention-Guided U-Net with Transformer for Medical Image Segmentation, IEEE Trans. Emerg. Top. Comput. Intell., vol. 8, no. 1, pp. 5568, 2024, doi: 10.1109/TETCI.2023.3309626.
  7. S. Sinha, M. Bhatt, A. Anand, A. S. Areeckal, and V. Aparna, EnigmaNet: A Novel Attention-Enhanced Segmentation Framework for Ischemic Stroke Lesion Detection in Brain MRI, IEEE Access, vol. 12, no. June, pp. 9148091498, 2024, doi: 10.1109/ACCESS.2024.3422025.
  8. G. Han et al., Improved U-Net based insulator image segmentation method based on attention mechanism, Energy Reports, vol. 7, pp. 210217, 2021, doi: 10.1016/j.egyr.2021.10.037.
  9. L. Shen, Q. Wang, W. Wang, R. Wang, Y. Yu, and C. Zhang, Deep attention enhanced networks for medical image segmentation, vol. 13167, no. Rsmip, p. 126, 2024, doi: 10.1117/12.3029812.
  10. Y. Zhou, Q. Kong, Y. Zhu, and Z. Su, MCFA-UNet: Multiscale Cascaded Feature Attention U-Net for Liver Segmentation, Irbm, vol. 44, no. 4, p. 100789, 2023, doi: 10.1016/j.irbm.2023.100789.
  11. M. T. Rostami and A. Motamedi, IRA-Unet: Inception Residual Attention U-Net in adversarial network for cardiac MRI segmentation, TechRxiv, preprint, pp. 110, Aug. 2023, doi: 10.36227/techrxiv.23896641.
  12. A. Kumar, Study and analysis of different segmentation methods for brain tumor MRI application, Multimed. Tools Appl., vol. 82, no. 5, pp. 71177139, 2023, doi: 10.1007/s11042-022-13636-y.
  13. S. Kuang et al., MSCDA: Multi-level semantic-guided contrast improves unsupervised domain adaptation for breast MRI segmentation in small datasets, Neural Networks, vol. 165, pp. 119134, 2023, doi: 10.1016/j.neunet.2023.05.014.
  14. F. Zhang et al., Deep learning based segmentation of brain tissue from diffusion MRI, Neuroimage, vol. 233, no. July 2020, p. 117934, 2021, doi: 10.1016/j.neuroimage.2021.117934.
  15. Y. Gu, Z. Piao, and S. J. Yoo, STHarDNet: Swin Transformer with HarDNet for MRI Segmentation, Appl. Sci., vol. 12, no. 1, 2022, doi: 10.3390/app12010468.
  16. C. Zhang, A. Achuthan, and G. M. S. Himel, State-of-the-Art and Challenges in Pancreatic CT Segmentation: A Systematic Review of U-Net and Its Variants, IEEE Access, vol. 12, no. June, pp. 7872678742, 2024, doi: 10.1109/ACCESS.2024.3392595.
  17. H. Wang, S. Qiu, B. Zhang, and L. Xiao, Multilevel Attention Unet Segmentation Algorithm for Lung Cancer Based on CT Images, Comput. Mater. Contin., vol. 78, no. 2, pp. 15691589, 2024, doi: 10.32604/cmc.2023.046821.
  18. K. Wisaeng, U-Net++DSM: Improved U-Net++ for Brain Tumor Segmentation With Deep Supervision Mechanism, IEEE Access, vol. 11, no. October, pp. 132268132285, 2023, doi: 10.1109/ACCESS.2023.3331025.
  19. T. Zhou, X. Ye, H. Lu, X. Zheng, S. Qiu, and Y. Liu, Dense convolutional network and its application in medical image analysis, Biomed. Res. Int., vol. 2022, pp. 111, 2022, doi: 10.1155/2022/2384830.
  20. Z. Mahmoud and M. A. M. Abdullah, Comparison of Deep Learning and Machine Learning Techniques for Automated Diagnosis of Acute Lymphoblastic Leukemia, Al-Rafidain Eng. J., vol. 30, no. 1, pp. 175185, 2025, doi: 10.33899/arej.2024.152478.1366.
  21. Z. Li, H. Zhang, Z. Li, and Z. Ren, Residual-attention UNet++: A nested residual-attention U-Net for medical image segmentation, Appl. Sci., vol. 12, no. 14, pp. 114, 2022, doi: 10.3390/app12147149.
  22. Z. Al-Mokhtar, F. Ibraheem, and H. Al-Layla, A Review of Digital Image Fusion and its Application, Al-Rafidain Eng. J., vol. 26, no. 2, pp. 309322, 2021, doi: 10.33899/rengj.2021.127928.1055.
  23. N. Abdullah and S. Al-Kazzaz, Evaluation of Physiotherapy Exercise by Motion Capturing Based on Artificial Intelligence: A Review, Al-Rafidain Engineering Journal (AREJ), 28, no. 2, pp. 237251, 2023, doi: 10.33899/rengj.2023.138562.1235.
  24. M. Cui, K. Li, J. Chen, and W. Yu, CM-Unet: A Novel Remote Sensing Image Segmentation Method Based on Improved U-Net, IEEE Access, vol. 11, no. June, pp. 5699457005, 2023, doi: 10.1109/ACCESS.2023.3282778.
  25. A. Al-Saegh, Identifying a Suitable Signal Processing Technique for MI EEG Data, Tikrit J. Eng. Sci., vol. 30, no. 3, pp. 140147, 2023, doi: 10.25130/tjes.30.3.14.
  26. N. S. Punn and S. Agarwal, Modality specific U-Net variants for biomedical image segmentation: A survey, Artif. Intell. Rev., vol. 55, no. 7, pp. 58455889, 2022, doi: 10.1007/s10462-022-10152-1.
  27. M. Ali, S. O. Gilani, A. Waris, K. Zafar, and M. Jamil, Brain Tumour Image Segmentation Using Deep Networks, IEEE Access, vol. 8, pp. 153589153598, 2020, doi: 10.1109/ACCESS.2020.3018160.
  28. O. Oktay et al., Attention U-Net: Learning where to look for the pancreas, arXiv preprint, 2018. [Online]. Available: http://arxiv.org/abs/1804.03999
  29. A. Dosovitskiy et al., An image is worth 1616 words: Transformers for image recognition at scale, in Proc. Int. Conf. Learn. Represent. (ICLR), 2021. [Online]. Available: http://arxiv.org/abs/2010.11929
  30. J. Chen et al., TransUNet: Transformers make strong encoders for medical image segmentation, arXiv preprint, pp. 113, 2021. [Online]. Available: http://arxiv.org/abs/2102.04306
  31. H. Cao et al., Swin-Unet: Unet-Like Pure Transformer forMedical Image Segmentation, Lect. Notes Comput. Sci. (including Subser. Lect. Notes Artif. Intell. Lect. Notes Bioinformatics), vol. 13803 LNCS, pp. 205218, 2023, doi: 10.1007/978-3-031-25066-8_9.
  32. G. Xu, X. Zhang, X. He, and X. Wu, LeViT-UNet: Make Faster Encoders withTransformer forMedical Image Segmentation, Lect. Notes Comput. Sci. (including Subser. Lect. Notes Artif. Intell. Lect. Notes Bioinformatics), vol. 14432 LNCS, pp. 4253, 2024, doi: 10.1007/978-981-99-8543-2_4.
  33. A. Tragakis, C. Kaul, R. Murray-Smith, and D. Husmeier, The Fully Convolutional Transformer for Medical Image Segmentation, Proc. - 2023 IEEE Winter Conf. Appl. Comput. Vision, WACV 2023, pp. 36493658, 2023, doi: 10.1109/WACV56688.2023.00365.
  34. R. Yousef, G. Gupta, N. Yousef, and M. Khari, A holistic overview of deep learning approach in medical imaging, vol. 28, no. 3. Springer Berlin Heidelberg, 2022. doi: 10.1007/s00530-021-00884-5.
  35. H.-Y. Zhou, J. Guo, Y. Zhang, L. Yu, L. Wang, and Y. Yu, nnFormer: Interleaved Transformer for volumetric segmentation, arXiv preprint, pp. 110, 2021. [Online]. Available: http://arxiv.org/abs/2109.03201
  36. M. Z. Alom, C. Yakopcic, M. Hasan, T. M. Taha, and V. K. Asari, Recurrent residual U-Net for medical image segmentation, Journal of Medical Imaging, vol. 6, no. 1, pp. 014006-1014006-13, 2019, doi: 10.1117/1.jmi.6.1.014006.
  37. C. Li et al., ANU-Net: Attention-based nested U-Net to exploit full resolution features for medical image segmentation, Comput. Graph., vol. 90, pp. 1120, 2020, doi: 10.1016/j.cag.2020.05.003.
  38. H. Cui, H. Pan, and K. Zhang, SCU-Net++: A Nested U-Net Based on Sharpening Filter and Channel Attention Mechanism, Wirel. Commun. Mob. Comput., vol. 2022, 2022, doi: 10.1155/2022/2848365.
  39. R. A. Moyassar and M. A. M. Abdullah, Advancing EMG Finger Movement Classification with Feature Extraction and Machine Learning, Al-Rafidain Eng. J., vol. 30, no. 1, pp. 150163, 2025, doi: 10.33899/arej.2024.152539.1368.
  40. A. Al-Saegh, S. A. Dawwd, and J. M. Abdul-Jabbar, Towards Efficient Motor Imagery EEG-based BCI Systems using DCNN, 2024 Arab ICT Conf. AICTC 2024, pp. 5966, 2024, doi: 10.1109/AICTC58357.2024.10735028.
  41. H. Peng and S. Yu, A Systematic IoU-Related Method: Beyond Simplified Regression for Better Localization, IEEE Trans. Image Process., vol. 30, no. i, pp. 50325044, 2021, doi: 10.1109/TIP.2021.3077144.
  42. Z. Fu, H. Chen, M. He, and L. Liu, An Enhanced U-Network by Combining PPM and CBAM for Medical Image Segmentation, IEEE Access, vol. 12, no. June, pp. 107098107112, 2024, doi: 10.1109/ACCESS.2024.3426518.

Authors

Noor M. Basheer

Computer Engineering Department, College of Engineering, University of Mosul, Mosul, Iraq

Ali Al-Saegh

Computer Engineering Department, College of Engineering, University of Mosul, Mosul, Iraq

ORCID

Identifiers

https://doi.org/10.33899/arej.2025.159398.1409
Download this PDF file
PDF

Statistics

How to Cite

[1]
N. M. Basheer and A. Al-Saegh, “Segmentation of Medical MRI Images Using Nested U-Net with Attention Mechanism and Fuzzy Pooling”, AREJ, vol. 30, no. 2, pp. 104–115, Sep. 2025.
Crossref
Scopus
Google Scholar
Europe PMC