无需训练,用注意力图实现开放世界目标检测与分割
Training-Free Open-Ended Object Detection and Segmentation via Attention as Prompts
- 通过视觉语言模型的注意力图作为提示,连接分割模型
- 在LVIS和CODA数据集上均超越现有开放末端方法
- 支持多种模型组合,适用于真实场景中的未知物体
现有感知模型依赖大量标注数据,但在开放世界场景下仍表现不佳。为解决此问题,研究者提出开集感知任务,但需预先定义类别,不适用于真实场景。本文提出新型开放末端目标检测与分割问题,即无需类别输入即可发现未知物体。我们提出VL-SAM框架,无需额外训练,将视觉语言模型(VLM)与Segment-Anything Model(SAM)结合,利用注意力图作为提示。通过头聚合与正则化注意力流,生成高质量注意力图;再迭代采样正负点并送入SAM进行分割。在长尾实例分割数据集LVIS上,该方法优于现有开放末端方法,并提供实例分割掩码;在角落案例检测数据集CODA上也表现良好,验证其在真实场景中的有效性。此外,该框架具备良好泛化能力,可兼容多种VLM与SAM。
原文摘要 · Abstract (English)
Existing perception models achieve great success by learning from large amounts of labeled data, but they still struggle with open-world scenarios. To alleviate this issue, researchers introduce open-set perception tasks to detect or segment unseen objects in the training set. However, these models require predefined object categories as inputs during inference, which are not available in real-world scenarios. Recently, researchers pose a new and more practical problem, \textit{i.e.}, open-ended object detection, which discovers unseen objects without any object categories as inputs. In this paper, we present VL-SAM, a training-free framework that combines the generalized object recognition model (\textit{i.e.,} Vision-Language Model) with the generalized object localization model (\textit{i.e.,} Segment-Anything Model), to address the open-ended object detection and segmentation task. Without additional training, we connect these two generalized models with attention maps as the prompts. Specifically, we design an attention map generation module by employing head aggregation and a regularized attention flow to aggregate and propagate attention maps across all heads and layers in VLM, yielding high-quality attention maps. Then, we iteratively sample positive and negative points from the attention maps with a prompt generation module and send the sampled points to SAM to segment corresponding objects. Experimental results on the long-tail instance segmentation dataset (LVIS) show that our method surpasses the previous open-ended method on the object detection task and can provide additional instance segmentation masks. Besides, VL-SAM achieves favorable performance on the corner case object detection dataset (CODA), demonstrating the effectiveness of VL-SAM in real-world applications. Moreover, VL-SAM exhibits good model generalization that can incorporate various VLMs and SAMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。