Does Your ViT Still Need U-Net for Segmentation?
Not on the map. Why?
Abstract (arXiv)
Medical image segmentation is dominated by U-Net-style encoder-decoder architectures. Vision Transformers (ViTs) overcome the limited receptive field of convolutional networks through self-attention, enabling modeling of long-range dependencies. Early ViT-based segmentation methods typically retained U-Net-style decoders because pretrained ViT representations were insufficient to support accurate dense prediction. Recent advances in large-scale pretraining have redefined the representation capability of ViTs, reducing the reliance on U-Net-style decoder architectures in modern vision models. This prompts two questions: Is the U-Net paradigm still necessary for medical image segmentation? If not, how should an encoder-only segmentation framework be designed? Motivated by these questions, we explore key architectural choices for encoder-only medical image segmentation based on modern ViT backbones and establish a query-based encoder-only design with multi-level query modeling and learnable block fusion, realized in Encoder-only Segmentation (EoSeg). Extensive experiments across seven benchmark datasets spanning CT, MRI, histopathology, endoscopy, and dermoscopy validate the effectiveness of the proposed design across diverse medical imaging modalities, including mDice scores of 85.50% on Synapse, 91.73% on ACDC, and 93.27% on GlaS. The results demonstrate that a U-Net-style decoder is no longer necessary for medical image segmentation with modern ViT backbones and further show that EoSeg provides an effective encoder-only design. Code is available at: https://github.com/Retinal-Research/EoSeg
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 31 of its 42 references that are in the map. First posted on arXiv 30 Jun 2026.
Builds on
31 of 42 references.
References · 31
- 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
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2023Segment Anythingcited by 8,234
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2014Fully Convolutional Networks for Semantic Segmentationcited by 4,859
- 2018Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentationcited by 3,223
- 2023Sigmoid Loss for Language Image Pre-Trainingcited by 3,109
- 2021Masked-attention Mask Transformer for Universal Image Segmentationcited by 2,351
- 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,558
19 more
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,525
- 2025DINOv3cited by 1,414
- 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,222
- 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 959
- 2019Kvasir-SEG: A Segmented Polyp Datasetcited by 589
- 2022How Do Vision Transformers Work?cited by 307
- 2019CE-Net: Context Encoder Network for 2D Medical Image Segmentationcited by 220
- 2019ResUNet++: An Advanced Architecture for Medical Image Segmentationcited by 211
- 2016Gland Segmentation in Colon Histology Images: The GlaS Challenge Contestcited by 203
- 2022UNeXt: MLP-based Rapid Medical Image Segmentation Networkcited by 185
- 2020DoubleU-Net: A Deep Convolutional Neural Network for Medical Image Segmentationcited by 125
- 2022HiFormer: Hierarchical Multi-scale Representations Using Transformers for Medical Image Segmentationcited by 95
- 2021LeViT-UNet: Make Faster Encoders with Transformer for Medical Image Segmentationcited by 85
- 2024MTP: Advancing Remote Sensing Foundation Model via Multi-Task Pretrainingcited by 66
- 2025Your ViT is Secretly an Image Segmentation Modelcited by 63
- 2024SelfReg-UNet: Self-Regularized UNet for Medical Image Segmentationcited by 18
- 2021NanoNet: Real-Time Polyp Segmentation in Video Capsule Endoscopy and Colonoscopycited by 16
- 2022BATFormer: Towards Boundary-Aware Lightweight Transformer for Efficient Medical Image Segmentationcited by 14
- 2024EViT-Unet: U-Net Like Efficient Vision Transformer for Medical Image Segmentation on Mobile and Edge Devicescited by 6
Built on by
Not cited yet.