DPSeg: Dual-Prompt Cost Volume Learning for Open-Vocabulary Semantic Segmentation
About
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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic segmentation | ADE20K A-150 | mIoU36.4 | 188 | |
| Semantic segmentation | Pascal Context 59 | mIoU62 | 164 | |
| Semantic segmentation | PASCAL-Context 59 class (val) | mIoU62 | 125 | |
| Semantic segmentation | ADE20K 847 | mIoU1.49e+3 | 83 | |
| Semantic segmentation | PASCAL-Context 59 classes (test) | mIoU62.3 | 75 | |
| Semantic segmentation | PASCAL-Context PC-459 | mIoU24.1 | 69 | |
| Semantic segmentation | ADE20K A-150 (val) | mIoU36.4 | 65 | |
| Semantic segmentation | PASCAL Context P-459 (val) | mIoU23.5 | 60 | |
| Semantic segmentation | Pascal Context 459 | mIoU23.5 | 58 | |
| Semantic segmentation | ADE-20k A-847 (test) | mIoU15.7 | 37 |