PC-UNet: a pure convolutional UNet with channel shuffle average for medical image segmentation

Abstract

In this study, a pure convolutional UNet with channel shuffle average, abbreviated as PC-UNet, has been proposed for medical image segmentation. Notably, the proposed PC-UNet is suitable for extracting context features, which is useful for model improvement. PC-UNet operates as an encoder-decoder network, where both the encoder and decoder are stacked with the proposed Pure Convolution (PC) modules. The PC module, containing a Channel Shuffle Average (CSA) component, is efficient in capturing context features without significant computational overhead. The CSA component transfers feature information from the channel dimension to the spatial dimension, enabling efficient computation. The effectiveness of the proposed PC-UNet has been rigorously validated on four widely used datasets, which are ISIC 2018, BUSI, GlaS, and Kvasir-SEG. Experimental results demonstrate that PC-UNet yields outstanding performance without imposing a significant computational load or increasing floating-point operations (FLOPs). When compared with eight mainstream models across all datasets, PC-UNet achieves the highest scores in both Dice and IoU metrics. The source code is available at: https://github.com/lwwant2sleep/PC-UNet.