江汉大学学报（自然科学版）2024，Vol.52Issue(3)：51-61,11.DOI:10.16389/j.cnki.cn42-1737/n.2024.03.006
基于改进通道多头注意力机制的U-Net3+医学图像分割算法研究
U-Net3+Medical Image Segmentation Algorithm Based on Improved Channel Multi-head Attention Mechanism
摘要
Abstract
Medical image segmentation is one of the current research hotspots,and the segmentation accuracy significantly impacts the subsequent medical diagnosis.In this paper,we proposed an improved U-Net3+image segmentation algorithm that incorporated a channel attention mechanism to address the shortcomings of most current medical image segmentation techniques that can not fully utilize and fuse multi-scale feature information.Based on the global jump connection structure of U-Net3+,a new channel attention mechanism was designed and embedded into the decoding path of the U-Net3+network to help the segmentation network adjust the training weights of important information when stitching the global feature map to fuse the global feature information efficiently.Finally,the model was compared and evaluated on two classical medical image segmentation datasets,and the average Dice coefficients reached 74.31%and 77.16%,respectively,were 3.01%and 2.98%higher than the original U-Net3+Dice coefficients.The experimental results show that the improved network model effectively improves the segmentation accuracy of medical images.关键词
U-Net3+/图像分割/注意力机制Key words
U-Net3+/image segmentation/attention mechanism分类
信息技术与安全科学引用本文复制引用
张全鑫,叶曦,杨志红,向青..基于改进通道多头注意力机制的U-Net3+医学图像分割算法研究[J].江汉大学学报（自然科学版）,2024,52(3):51-61,11.基金项目
工业烟尘污染控制湖北省重点实验室开放课题(HBIK2022-08) （HBIK2022-08）
