The invention discloses a medical image
automatic segmentation method of a U-shaped network based on channel mixing and a self-attention mechanism. According to the method, channel grouping and a disturbance strategy are introduced into a self-attention mechanism, so that the problems of channel redundancy, high operation cost and the like of a traditional self-attention method are effectively relieved; and the capability of extracting
local space information and global channel information by the convolutional layer and the capability of extracting long-
distance space dependence and modeling local channel information by the self-attention layer are combined more closely, so that the method has more comprehensive modeling capability for features. According to a network used in the method, firstly, an input medical
nuclear magnetic resonance image or a
computed tomography image passes through a down-sampling layer to preliminarily extract feature information and reduce the spatial dimension of the image, then a feature map is sent into a multi-level symmetric
encoder-decoder network, and each level comprises a designed module; the module structure comprises a patch embedding module formed by depth separable
convolution, a self-attention layer based on channel grouping and channel disturbance, and a
bottleneck convolution layer formed by
convolution layers of different dimensions. Wherein the patch embedding module is used for extracting position information, and the middle self-attention layer is used for extracting a long-distance dependency relationship and extracting local channel information through channel grouping and disturbance. The
bottleneck convolution layer is used for exploring inductive bias and aggregating global channel information. Besides, in the jump connection process of the decoder, an efficient fusion module of channel and space cross attention is constructed, and the
recovery capability of the feature information is further enhanced. The provided model is evaluated on a plurality of common data sets, and experimental results show that the method makes full use of channel information and spatial information between features, obtains a better segmentation result compared with an existing method, can be used for improving the performance of medical
image segmentation, and has important application prospects and practical significance.