Improving Unet Networks for Medical Image Segmentation by adding Attention Mechanism Layers

Document Type : Research Paper

Authors

1 Computer Science Engineer, Shahid Bahonar University , Kerman , Iran

2 Department of Computer Engineering Shahid Bahonar University of Kerman, Kerman,Iran.

3 Department of Computer Engineering, Shahid Bahonr University of Kerman, Kerman, Iran.

Abstract

Medical image segmentation is one of the most important steps in medical image analysis to improve diagnosis and findings. One of the most common segmentation methods in deep learning is the use of Unet networks. The presence of overlapping layers in the Unet decoder does not allow extracting information from deeper layers. Also, due to the limited range of the received field of convolution cores, long-range information and dependencies are not considered well. In this article, our goal is to place a structure in the area between encoder and decoder in the Unet model in order to fill the semantic gap between the encoder and decoder area and better extract features by paying attention to local and global features. This model makes the target region more prominent in different medical datasets. We have conducted our experiment on 6 medical data sets, and the results obtained in two evaluation criteria, Dice and Iou, show that our proposed model has better results than Unet and based methods.

Keywords

Journal of Machine Vision and Image Processing
Volume 10, Issue 4
January 2024
Pages 49-59

Files

Share

How to cite

Statistics