Journal of Jianghan University (Natural Science Edition) ›› 2024, Vol. 52 ›› Issue (3): 51-61.doi: 10.16389/j.cnki.cn42-1737/n.2024.03.006

Previous Articles    

U-Net3+ Medical Image Segmentation Algorithm Based on Improved Channel Multi-head Attention Mechanism

ZHANG Quanxina,YE Xi*a,b,YANG Zhihonga,XIANG Qinga   

  1. a. School of Intelligent Manufacturing;b. Hubei Key Laboratory of Industrial Fume and Dust Pollution Control, Jianghan University,Wuhan 430056,Hubei,China
  • Published:2024-06-17
  • Contact: YE Xi

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.

Key words: U-Net3+, image segmentation, attention mechanism

CLC Number: