arXiv:2503.10779cs.CV2025-03被引 4

一个视觉样本+微调,就能让大模型实现精准分割。

The Power of One: A Single Example is All it Takes for Segmentation in VLMs

  • 用单个图像样本微调文本-图像注意力层,提升分割性能。
  • 无需标签即可自动选最优注意力层,零样本表现显著提升。
  • 方法通用,适配多种视觉语言模型,适合研究开放词汇分割者。

大规模视觉语言模型(VLMs)通过海量图文对训练,隐式学习文本描述与图像区域的关联,具备强大的多模态理解能力,可实现零样本目标检测与分割。然而,现有方法依赖提示工程和人工选择注意力层或头部,性能受限。本文提出仅需为每个类别提供一个视觉示例,并微调文本到图像注意力层及嵌入,显著提升性能。同时,通过少样本微调在多个层和/或提示上构建集成,提出基于熵的注意力层排名选择机制,无需分割标签即可自动筛选最优层。该方法避免了超参数调优,更具灵活性与可扩展性。实验表明,该方法在零样本下表现优异,微调后进一步提升。结果具有普适性,可应用于多种视觉语言模型。

原文摘要 · Abstract (English)

Large-scale vision-language models (VLMs), trained on extensive datasets of image-text pairs, exhibit strong multimodal understanding capabilities by implicitly learning associations between textual descriptions and image regions. This emergent ability enables zero-shot object detection and segmentation, using techniques that rely on text-image attention maps, without necessarily training on abundant labeled segmentation datasets. However, performance of such methods depends heavily on prompt engineering and manually selected layers or head choices for the attention layers. In this work, we demonstrate that, rather than relying solely on textual prompts, providing a single visual example for each category and fine-tuning the text-to-image attention layers and embeddings significantly improves the performance. Additionally, we propose learning an ensemble through few-shot fine-tuning across multiple layers and/or prompts. An entropy-based ranking and selection mechanism for text-to-image attention layers is proposed to identify the top-performing layers without the need for segmentation labels. This eliminates the need for hyper-parameter selection of text-to-image attention layers, providing a more flexible and scalable solution for open-vocabulary segmentation. We show that this approach yields strong zero-shot performance, further enhanced through fine-tuning with a single visual example. Moreover, we demonstrate that our method and findings are general and can be applied across various vision-language models (VLMs).

视觉语言模型零样本分割少样本学习注意力机制

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