More complex encoder is not all you need
Not on the map. Why?
Abstract (arXiv)
U-Net and its variants have been widely used in medical image segmentation. However, most current U-Net variants confine their improvement strategies to building more complex encoder, while leaving the decoder unchanged or adopting a simple symmetric structure. These approaches overlook the true functionality of the decoder: receiving low-resolution feature maps from the encoder and restoring feature map resolution and lost information through upsampling. As a result, the decoder, especially its upsampling component, plays a crucial role in enhancing segmentation outcomes. However, in 3D medical image segmentation, the commonly used transposed convolution can result in visual artifacts. This issue stems from the absence of direct relationship between adjacent pixels in the output feature map. Furthermore, plain encoder has already possessed sufficient feature extraction capability because downsampling operation leads to the gradual expansion of the receptive field, but the loss of information during downsampling process is unignorable. To address the gap in relevant research, we extend our focus beyond the encoder and introduce neU-Net (i.e., not complex encoder U-Net), which incorporates a novel Sub-pixel Convolution for upsampling to construct a powerful decoder. Additionally, we introduce multi-scale wavelet inputs module on the encoder side to provide additional information. Our model design achieves excellent results, surpassing other state-of-the-art methods on both the Synapse and ACDC datasets.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 26 of its 49 references that are in the map. First posted on arXiv 20 Sep 2023.
Builds on
26 of 49 references.
References · 26
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2016Feature Pyramid Networks for Object Detectioncited by 5,036
- 2016DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFscited by 3,241
- 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,558
- 20163D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotationcited by 1,476
- 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,435
- 2018ImageNet-trained CNNs are biased towards texture; increasing shape bias improves accuracy and robustnesscited by 1,432
- 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,222
- 2016Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Networkcited by 1,204
- 2016Instance Normalization: The Missing Ingredient for Fast Stylizationcited by 1,198
14 more
- 2015Multi-Scale Context Aggregation by Dilated Convolutionscited by 1,133
- 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 959
- 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 830
- 2019UNet++: Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentationcited by 586
- 2017Deep Learning for Medical Image Analysiscited by 401
- 2020UNet 3+: A Full-Scale Connected UNet for Medical Image Segmentationcited by 332
- 2016A guide to convolution arithmetic for deep learningcited by 185
- 2016EnhanceNet: Single Image Super-Resolution Through Automated Texture Synthesiscited by 178
- 2018A Novel Focal Tversky loss function with improved Attention U-Net for lesion segmentationcited by 172
- 2020U-Net and its variants for medical image segmentation: theory and applicationscited by 169
- 2017Deep Learning for Medical Image Processing: Overview, Challenges and Futurecited by 104
- 2021MISSFormer: An Effective Medical Image Segmentation Transformercited by 103
- 2022Y-Net: A Spatiospectral Dual-Encoder Networkfor Medical Image Segmentationcited by 19
- 2016Is the deconvolution layer the same as a convolutional layer?cited by 16
Built on by
Not cited yet.