arXiv:2607.00223cs.CV2026-07

用现代ViT做医学图像分割,无需U-Net解码器,性能更优。

Does Your ViT Still Need U-Net for Segmentation?

论文配图:Does Your ViT Still Need U-Net for Segmentation?
图 1 · 摘自论文原文
  • 基于多层级查询与可学习块融合,设计纯编码器分割框架
  • 在7个数据集上实现85.50%~93.27%的mDice,超越传统U-Net结构
  • 适合追求轻量化、高性能医学图像分割的研究者与应用开发者

医学图像分割长期依赖U-Net类编码器-解码器架构。视觉变换器(ViTs)通过自注意力机制克服卷积网络感受野有限的问题,能建模长程依赖。早期基于ViT的分割方法仍保留U-Net式解码器,因预训练ViT表示能力不足以支持精确密集预测。近期大规模预训练进展显著提升了ViT的表征能力,降低了对U-Net式解码器的依赖。这引发两个问题:U-Net范式在医学图像分割中是否仍必要?若否,如何设计纯编码器分割框架?针对此,我们探索现代ViT骨干下的纯编码器分割关键架构选择,提出基于查询的纯编码器设计(EoSeg),包含多层级查询建模与可学习块融合。在涵盖CT、MRI、组织病理学、内窥镜和皮肤镜的七个基准数据集上进行广泛实验,验证了该设计在多种医学影像模态下的有效性,包括在Synapse数据集上mDice达85.50%,ACDC为91.73%,GlaS为93.27%。结果表明,现代ViT骨干下无需U-Net式解码器,且EoSeg提供了一种高效的纯编码器分割方案。代码已开源。

原文摘要 · Abstract (English)

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

医学图像分割Vision Transformer纯编码器EoSeg

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。