Swin-unet: Unet-like pure transformer for medical image segmentation

H Cao, Y Wang, J Chen, D Jiang, X Zhang… - European conference on …, 2022 - Springer
European conference on computer vision, 2022Springer
In the past few years, convolutional neural networks (CNNs) have achieved milestones in
medical image analysis. In particular, deep neural networks based on U-shaped architecture
and skip-connections have been widely applied in various medical image tasks. However,
although CNN has achieved excellent performance, it cannot learn global semantic
information interaction well due to the locality of convolution operation. In this paper, we
propose Swin-Unet, which is an Unet-like pure Transformer for medical image …
Abstract
In the past few years, convolutional neural networks (CNNs) have achieved milestones in medical image analysis. In particular, deep neural networks based on U-shaped architecture and skip-connections have been widely applied in various medical image tasks. However, although CNN has achieved excellent performance, it cannot learn global semantic information interaction well due to the locality of convolution operation. In this paper, we propose Swin-Unet, which is an Unet-like pure Transformer for medical image segmentation. The tokenized image patches are fed into the Transformer-based U-shaped Encoder-Decoder architecture with skip-connections for local-global semantic feature learning. Specifically, we use a hierarchical Swin Transformer with shifted windows as the encoder to extract context features. And a symmetric Swin Transformer-based decoder with a patch expanding layer is designed to perform the up-sampling operation to restore the spatial resolution of the feature maps. Under the direct down-sampling and up-sampling of the inputs and outputs by , experiments on multi-organ and cardiac segmentation tasks demonstrate that the pure Transformer-based U-shaped Encoder-Decoder network outperforms those methods with full-convolution or the combination of transformer and convolution. The codes have been publicly available at the link (https://github.com/HuCaoFighting/Swin-Unet).
Springer