arXiv:2409.03525cs.CV2024-09被引 8

融合空间与语义知识,实现零样本分割新突破

FrozenSeg: Harmonizing Frozen Foundation Models for Open-Vocabulary Segmentation

  • 用SAM提供空间信息,CLIP提取语义特征,协同生成精准分割图
  • 仅训练轻量解码器,冻结主干模型,提升效率与性能
  • 无需微调,在COCO上零样本测试即达顶尖水平

开放词汇分割需在未受约束环境中识别并分割任意类别物体,挑战巨大。尽管基于视觉-语言(ViL)基础模型(如CLIP)的零样本能力已有进展,但对未见类别和场景仍难以生成精确掩码,导致分割性能受限。为此,我们提出FrozenSeg,一种协同整合定位模型(如SAM)的空间知识与ViL模型(如CLIP)语义知识的新框架。以CLIP的视觉编码器为特征骨干,将空间感知特征注入可学习查询与CLIP特征中的Transformer解码器。同时设计掩码提议集成策略,进一步提升召回率与掩码质量。为充分复用预训练知识并最小化训练开销,冻结两个基础模型,仅优化轻量级解码器用于掩码生成——这是性能瓶颈所在。大量实验表明,FrozenSeg在多个分割基准上达到当前最佳水平,仅在COCO全景数据集上训练,并以零样本方式测试。代码已开源。

原文摘要 · Abstract (English)

Open-vocabulary segmentation poses significant challenges, as it requires segmenting and recognizing objects across an open set of categories in unconstrained environments. Building on the success of powerful vision-language (ViL) foundation models, such as CLIP, recent efforts sought to harness their zero-short capabilities to recognize unseen categories. Despite notable performance improvements, these models still encounter the critical issue of generating precise mask proposals for unseen categories and scenarios, resulting in inferior segmentation performance eventually. To address this challenge, we introduce a novel approach, FrozenSeg, designed to integrate spatial knowledge from a localization foundation model (e.g., SAM) and semantic knowledge extracted from a ViL model (e.g., CLIP), in a synergistic framework. Taking the ViL model's visual encoder as the feature backbone, we inject the space-aware feature into the learnable queries and CLIP features within the transformer decoder. In addition, we devise a mask proposal ensemble strategy for further improving the recall rate and mask quality. To fully exploit pre-trained knowledge while minimizing training overhead, we freeze both foundation models, focusing optimization efforts solely on a lightweight transformer decoder for mask proposal generation-the performance bottleneck. Extensive experiments demonstrate that FrozenSeg advances state-of-the-art results across various segmentation benchmarks, trained exclusively on COCO panoptic data, and tested in a zero-shot manner. Code is available at https://github.com/chenxi52/FrozenSeg.

分割零样本CLIPSAM

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