arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.20105cs.CV

FreeSeg-Diff:基于扩散模型的无训练开放词汇分割

FreeSeg-Diff: Training-Free Open-Vocabulary Segmentation with Diffusion Models

DIISM University of Siena(DIISM锡耶纳大学);CNRS, ISIR Sorbonne University(CNRS,ISIR索邦大学);Inria, ARCHES Sorbonne University(Inria,ARCHES索邦大学);Valeo.ai Sorbonne University(Valeo.ai索邦大学)

Barbara Toniella Corradini, Mustafa Shukor, Paul Couairon, Guillaume Couairon, Franco Scarselli, Matthieu Cord

AI总结:

该研究提出FreeSeg-Diff方法,利用BLIP、Stable Diffusion Model和CLIP等开源基础模型,以无训练方式实现零样本开放词汇分割,在Pascal VOC和COCO数据集上优于多数训练基方法,且扩散模型特征表现更优。

摘要:

基础模型在处理多个领域和任务时展现出前所未有的能力。CLIP等模型目前被广泛用于桥接跨模态表示,而文本到图像扩散模型可以说是真实图像生成领域的领先模型。图像生成模型在大规模数据集上训练,使其具备强大的内部空间表示能力。在这项工作中,我们探索了这类表示在图像生成之外的潜在益处,特别是在密集视觉预测任务中的应用。我们聚焦于图像分割任务,该任务传统上通过在封闭词汇数据集上用像素级标注训练模型来解决。为避免标注成本或训练大型扩散模型,我们将设置限制为零样本且无训练。简而言之,我们的 pipeline 利用不同且相对小型的开源基础模型进行零样本开放词汇分割。pipeline 流程如下:将图像同时输入到 captioner 模型(即 BLIP)和扩散模型(即 Stable Diffusion Model),分别生成文本描述和视觉表示;对特征进行聚类和二值化,以获得每个对象的类无关掩码;然后使用 CLIP 模型将这些掩码映射到文本类别,以支持开放词汇;最后,我们添加一个细化步骤,以获得更精确的分割掩码。我们的方法(称为 FreeSeg-Diff)不依赖任何训练,在 Pascal VOC 和 COCO 数据集上优于许多基于训练的方法。此外,与最近的弱监督分割方法相比,我们展示了极具竞争力的结果。我们提供了全面的实验,表明扩散模型特征优于其他预训练模型。项目页面:https://bcorrad.github.io/freesegdiff/

英文摘要:

Foundation models have exhibited unprecedented capabilities in tackling many domains and tasks. Models such as CLIP are currently widely used to bridge cross-modal representations, and text-to-image diffusion models are arguably the leading models in terms of realistic image generation. Image generative models are trained on massive datasets that provide them with powerful internal spatial representations. In this work, we explore the potential benefits of such representations, beyond image generation, in particular, for dense visual prediction tasks. We focus on the task of image segmentation, which is traditionally solved by training models on closed-vocabulary datasets, with pixel-level annotations. To avoid the annotation cost or training large diffusion models, we constraint our setup to be zero-shot and training-free. In a nutshell, our pipeline leverages different and relatively small-sized, open-source foundation models for zero-shot open-vocabulary segmentation. The pipeline is as follows: the image is passed to both a captioner model (i.e. BLIP) and a diffusion model (i.e., Stable Diffusion Model) to generate a text description and visual representation, respectively. The features are clustered and binarized to obtain class agnostic masks for each object. These masks are then mapped to a textual class, using the CLIP model to support open-vocabulary. Finally, we add a refinement step that allows to obtain a more precise segmentation mask. Our approach (dubbed FreeSeg-Diff), which does not rely on any training, outperforms many training-based approaches on both Pascal VOC and COCO datasets. In addition, we show very competitive results compared to the recent weakly-supervised segmentation approaches. We provide comprehensive experiments showing the superiority of diffusion model features compared to other pretrained models. Project page: https://bcorrad.github.io/freesegdiff/

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑