Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Does Your ViT Still Need U-Net for Segmentation?

About

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

Xin Li, Wenhui Zhu, Xuanzhao Dong, Xiwen Chen, Yanxi Chen, Yujian Xiong, Hao Wang, Oana M. Dumitrascu, Yalin Wang• 2026

Related benchmarks

TaskDatasetResultRank
Medical Image SegmentationGLAS
Dice93.27
134
Multi-organ SegmentationSynapse multi-organ CT (test)
DSC (Aorta)90.8
108
Polyp SegmentationKvasir-Seg
mIoU0.8535
54
Skin Lesion SegmentationISIC 2016
Dice Score (D)93.22
49
Skin Lesion SegmentationISIC 2017
Dice Score87.4
43
Medical Image SegmentationMoNuSeg
Dice80.51
37
Cardiac SegmentationACDC
Dice (LV)95.85
33
Showing 7 of 7 rows

Other info

Follow for update