用空间提示引导扩散模型,实现无需标注的精准实例分割。
Prompting Diffusion Models for Zero-Shot Instance Segmentation

- 引入空间提示分支,让扩散模型直接响应用户输入
- 在7个数据集上零样本泛化,优于基线模型
- 适合需要交互式分割但无大量标注的场景
近年来,计算机视觉中涌现出基础模型与生成模型的突破性进展,后者凭借丰富的先验知识在场景理解任务中表现优异。然而,现有方法在可提示分割任务中仍存在定位不准、误检和过分割问题。早期工作仅在后处理阶段使用提示,忽略了用户意图。本文提出Prompt2Seg,一种基于扩散模型的空间条件框架,通过2D高斯或置信图作为显式输入信号,使冻结的扩散分割模型能直接响应用户意图。在从Hypersim和Virtual KITTI 2中选取的有限类别上微调后,该方法可在多种未见物体类型和视觉域(包括绘画、第一视角、X射线图像)上实现零样本泛化。在七个数据集上的评估表明,Prompt2Seg始终优于底层扩散分割基线模型。结果表明,生成预训练中的丰富先验与合理空间条件结合,为无需大规模掩码监督的通用交互式分割提供了有效路径。
原文摘要 · Abstract (English)
Several disruptive research directions have recently emerged in computer vision, including foundation models achieving previously unseen zero-shot performance in scene understanding, even interactively, and generative models that synthesize extremely realistic images. The latter have also been shown to be highly effective in scene understanding tasks thanks to their rich priors. However, for promptable segmentation, foundation models struggle with accurately segmenting an object's region, leading to false positives and over-segmentation. Notably, early attempts that leverage generative priors use prompts only during post-processing, yielding suboptimal segments because the process is agnostic to the user input. In this paper, we target these limitations with Prompt2Seg, a spatial conditioning framework for diffusion-based segmentation. Prompt2Seg augments a frozen diffusion segmentation model with a conditioning branch. Our approach takes spatial prompts, represented as 2D Gaussians or confidence maps, as explicit input signals, training the model to respond directly to user intent. Fine-tuned on a deliberately constrained set of object categories drawn from Hypersim and Virtual KITTI 2, Prompt2Seg generalizes zero-shot to a wide range of unseen object types and visual domains. We evaluate on seven datasets ranging from standard benchmarks to more challenging domains, including paintings, egocentric views, and X-ray data. Furthermore, we demonstrate that Prompt2Seg consistently outperforms the underlying diffusion segmentation backbone across all benchmarks. Our results suggest that the rich priors encoded in generative pretraining, combined with principled spatial conditioning, offer a compelling path toward broadly generalizing interactive segmentation without large-scale mask supervision.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。