首页|基于改进通道多头注意力机制的U-Net3+医学图像分割算法研究

基于改进通道多头注意力机制的U-Net3+医学图像分割算法研究

扫码查看
医学图像分割作为当前的研究热点之一,分割精度对于后续的医学诊断影响巨大.针对目前大多数医学图像分割技术无法充分利用并融合多尺度特征信息的缺陷,提出了一种融合通道注意力机制的改进U-Net3+图像分割算法.在U-Net3+的全局跳跃连接结构的基础上,设计一种新的通道注意力机制并将它嵌入到U-Net3+网络的解码路径中,帮助分割网络在拼接全局特征图时调整重要信息的训练权重从而高效融合全局特征信息.最后,在两种经典的医学图像分割数据集上将该模型进行对比评估,平均Dice系数分别达到了 74.31%和 77.16%,相比原本U-Net3+的Dice系数分别提高了 3.01%和 2.98%.实验结果表明改进后的网络模型有效提高了医学图像的分割精度.
U-Net3+Medical Image Segmentation Algorithm Based on Improved Channel Multi-head Attention Mechanism
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+image segmentationattention mechanism

张全鑫、叶曦、杨志红、向青

展开 >

江汉大学 智能制造学院,湖北 武汉 430056

江汉大学 工业烟尘污染控制湖北省重点实验室,湖北 武汉 430056

U-Net3+ 图像分割 注意力机制

工业烟尘污染控制湖北省重点实验室开放基金

HBIK2022-08

2024

江汉大学学报(自然科学版)
江汉大学

江汉大学学报(自然科学版)

影响因子:0.413
ISSN:1673-0143
年,卷(期):2024.52(3)