arXiv:2410.11473cs.CV2024-10AAAI被引 5

通过测试时反演文本提示,让分割模型更精准理解类别语义。

InvSeg: Test-Time Prompt Inversion for Semantic Segmentation

  • 测试时将图像上下文反演为丰富文本提示,增强语义对齐。
  • 在PASCAL VOC等3个数据集上达到当前最优性能。
  • 适合需要开放词汇分割的视觉理解任务研究者使用。

基于文本到图像扩散模型注意力图中的视觉-文本关联,有助于密集视觉预测任务(如语义分割)。然而,由于图像生成使用的上下文丰富句子与语义分割中常用的孤立类别名之间存在输入分布差异,导致扩散模型难以捕捉准确的视觉-文本关联。为此,我们提出InvSeg,一种测试时提示反演方法,通过将图像特定的视觉上下文反演至文本提示嵌入空间,利用扩散模型重建过程中的结构信息来丰富文本提示,使每个类别关联到结构一致的掩码。具体地,引入对比软聚类(CSC)对生成掩码与图像结构信息进行对齐,软选择每类锚点并计算加权距离,促使同类像素更接近、异类像素更分离,从而保证掩码的区分度与内部一致性。通过引入样本特定上下文,InvSeg在嵌入空间中学习到丰富的文本提示,实现跨模态精准对齐。实验表明,InvSeg在PASCAL VOC、PASCAL Context和COCO Object数据集上均达到当前最优表现。

原文摘要 · Abstract (English)

Visual-textual correlations in the attention maps derived from text-to-image diffusion models are proven beneficial to dense visual prediction tasks, e.g., semantic segmentation. However, a significant challenge arises due to the input distributional discrepancy between the context-rich sentences used for image generation and the isolated class names typically used in semantic segmentation. This discrepancy hinders diffusion models from capturing accurate visual-textual correlations. To solve this, we propose InvSeg, a test-time prompt inversion method that tackles open-vocabulary semantic segmentation by inverting image-specific visual context into text prompt embedding space, leveraging structure information derived from the diffusion model's reconstruction process to enrich text prompts so as to associate each class with a structure-consistent mask. Specifically, we introduce Contrastive Soft Clustering (CSC) to align derived masks with the image's structure information, softly selecting anchors for each class and calculating weighted distances to push inner-class pixels closer while separating inter-class pixels, thereby ensuring mask distinction and internal consistency. By incorporating sample-specific context, InvSeg learns context-rich text prompts in embedding space and achieves accurate semantic alignment across modalities. Experiments show that InvSeg achieves state-of-the-art performance on the PASCAL VOC, PASCAL Context and COCO Object datasets.

语义分割扩散模型开放词汇

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