通过双提示机制提升未见类别语义分割精度
DPSeg: Dual-Prompt Cost Volume Learning for Open-Vocabulary Semantic Segmentation
- 设计双提示成本体积生成框架,融合视觉与文本提示
- 在多个数据集上显著超越现有最先进方法
- 适合需要细粒度分割和开放词汇场景的研究者
开放词汇语义分割旨在对图像中已见和未见类别进行像素级区域划分。现有方法依赖于CLIP等预训练视觉语言模型的文本嵌入,但图像与文本嵌入间存在固有领域差距,即使经过大量对齐训练仍难以消除。此外,仅使用深层文本对齐特征会削弱浅层特征的指导作用,影响小物体和细节的检测,降低分割精度。为此,我们提出DPSeg双提示框架,结合双提示成本体积生成、成本体积引导解码器及语义引导提示优化策略,利用双提示机制缓解视觉提示生成中的对齐问题。通过引入视觉提示编码器的视觉嵌入,减少文本与图像嵌入间的领域差距,并通过浅层特征提供多层级指导。大量实验表明,该方法在多个公开数据集上显著优于现有最先进方法。
原文摘要 · Abstract (English)
Open-vocabulary semantic segmentation aims to segment images into distinct semantic regions for both seen and unseen categories at the pixel level. Current methods utilize text embeddings from pre-trained vision-language models like CLIP but struggle with the inherent domain gap between image and text embeddings, even after extensive alignment during training. Additionally, relying solely on deep text-aligned features limits shallow-level feature guidance, which is crucial for detecting small objects and fine details, ultimately reducing segmentation accuracy. To address these limitations, we propose a dual prompting framework, DPSeg, for this task. Our approach combines dual-prompt cost volume generation, a cost volume-guided decoder, and a semantic-guided prompt refinement strategy that leverages our dual prompting scheme to mitigate alignment issues in visual prompt generation. By incorporating visual embeddings from a visual prompt encoder, our approach reduces the domain gap between text and image embeddings while providing multi-level guidance through shallow features. Extensive experiments demonstrate that our method significantly outperforms existing state-of-the-art approaches on multiple public datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。