arXiv:2604.24575cs.CV2026-04被引 3

用扩散模型做通用分割,无需改架构就能识别任意文本描述的物体。

Diffusion Model as a Generalist Segmentation Learner

论文配图:Diffusion Model as a Generalist Segmentation Learner
图 1 · 摘自论文原文
  • 将图像和标签图编码到隐空间,用文本引导扩散U-Net生成分割掩码。
  • 在多个标准数据集上达到顶尖性能,且跨医疗、遥感等场景泛化能力强。
  • 无需定制模型结构,即可实现开放词汇分割,适合多领域应用。

扩散模型虽主要用于图像生成,但其去噪过程蕴含丰富的空间对齐视觉先验。本文证明这些先验可用于文本条件下的语义与开放词汇分割,并可推广至多种下游任务,构建通用扩散分割框架。我们提出DiGSeg(Diffusion Models as a Generalist Segmentation Learner),将预训练扩散模型重用于统一分割框架。该方法将输入图像与真实标签图编码至隐空间并拼接为扩散U-Net的条件信号,同时通过并行的CLIP对齐文本路径,在多尺度注入语言特征,使模型能对齐文本查询与动态视觉表示。这一设计将现成扩散主干网络转变为通用接口,可根据外观和任意文本提示生成结构化分割掩码。大量实验表明,该方法在标准语义分割基准上达到顶尖水平,具备强大开放词汇泛化能力,并可跨域迁移至医疗、遥感、农业场景,无需特定领域架构调整。结果表明,现代扩散主干可作为通用分割学习者,而非仅限于生成任务,缩小了视觉生成与理解之间的差距。

原文摘要 · Abstract (English)

Diffusion models are primarily trained for image synthesis, yet their denoising trajectories encode rich, spatially aligned visual priors. In this paper, we demonstrate that these priors can be utilized for text-conditioned semantic and open-vocabulary segmentation, and this approach can be generalized to various downstream tasks to make a general-purpose diffusion segmentation framework. Concretely, we introduce DiGSeg (Diffusion Models as a Generalist Segmentation Learner), which repurposes a pretrained diffusion model into a unified segmentation framework. Our approach encodes the input image and ground-truth mask into the latent space and concatenates them as conditioning signals for the diffusion U-Net. A parallel CLIP-aligned text pathway injects language features across multiple scales, enabling the model to align textual queries with evolving visual representations. This design transforms an off-the-shelf diffusion backbone into a universal interface that produces structured segmentation masks conditioned on both appearance and arbitrary text prompts. Extensive experiments demonstrate state-of-the-art performance on standard semantic segmentation benchmarks, as well as strong open-vocabulary generalization and cross-domain transfer to medical, remote sensing, and agricultural scenarios-without domain-specific architectural customization. These results indicate that modern diffusion backbones can serve as generalist segmentation learners rather than pure generators, narrowing the gap between visual generation and visual understanding.

扩散模型分割开放词汇通用模型

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