用图文联合提示增强SAM,实现无需人工标注的少样本分割
Vision and Language Reference Prompt into SAM for Few-shot Segmentation
- 将参考图像与文本标签共同作为多模态提示输入SAM
- 在PASCAL-5i和COCO-20i上分别提升6.3%和9.5% mIoU
- 支持未见物体分割,适合跨类别少样本场景
Segment Anything Model (SAM) 是一个大规模分割模型,可实现强大的零样本能力并支持灵活提示。尽管SAM能零样本分割任意物体,但需用户为每张目标图像提供提示,且生成的掩码不携带标签信息。少样本分割模型通过输入带标注的参考图像作为提示来解决此问题,使SAM可在无用户提示下分割特定物体。然而,以往基于SAM的少样本分割方法仅使用参考图像作为提示,因缺乏参考信息导致精度受限。本文提出一种新模型VLP-SAM,利用参考图像的视觉信息和文本标签的语义信息,同时输入图像与语言作为参考信息。VLP-SAM结构简单、可扩展,仅需少量可学习参数,通过多模态视觉语言模型将融合了视觉语言信息的提示嵌入输入SAM。在PASCAL-5i和COCO-20i数据集上的实验表明,VLP-SAM显著优于现有最先进模型(mIoU分别提升6.3%和9.5%)。此外,VLP-SAM在未见物体上也表现出良好泛化能力。代码已开源。
原文摘要 · Abstract (English)
Segment Anything Model (SAM) represents a large-scale segmentation model that enables powerful zero-shot capabilities with flexible prompts. While SAM can segment any object in zero-shot, it requires user-provided prompts for each target image and does not attach any label information to masks. Few-shot segmentation models addressed these issues by inputting annotated reference images as prompts to SAM and can segment specific objects in target images without user-provided prompts. Previous SAM-based few-shot segmentation models only use annotated reference images as prompts, resulting in limited accuracy due to a lack of reference information. In this paper, we propose a novel few-shot segmentation model, Vision and Language reference Prompt into SAM (VLP-SAM), that utilizes the visual information of the reference images and the semantic information of the text labels by inputting not only images but also language as reference information. In particular, VLP-SAM is a simple and scalable structure with minimal learnable parameters, which inputs prompt embeddings with vision-language information into SAM using a multimodal vision-language model. To demonstrate the effectiveness of VLP-SAM, we conducted experiments on the PASCAL-5i and COCO-20i datasets, and achieved high performance in the few-shot segmentation task, outperforming the previous state-of-the-art model by a large margin (6.3% and 9.5% in mIoU, respectively). Furthermore, VLP-SAM demonstrates its generality in unseen objects that are not included in the training data. Our code is available at https://github.com/kosukesakurai1/VLP-SAM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。