CAT-Seg: Cost Aggregation for Open-Vocabulary Semantic Segmentation
About
Open-vocabulary semantic segmentation presents the challenge of labeling each pixel within an image based on a wide range of text descriptions. In this work, we introduce a novel cost-based approach to adapt vision-language foundation models, notably CLIP, for the intricate task of semantic segmentation. Through aggregating the cosine similarity score, i.e., the cost volume between image and text embeddings, our method potently adapts CLIP for segmenting seen and unseen classes by fine-tuning its encoders, addressing the challenges faced by existing methods in handling unseen classes. Building upon this, we explore methods to effectively aggregate the cost volume considering its multi-modal nature of being established between image and text embeddings. Furthermore, we examine various methods for efficiently fine-tuning CLIP.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic segmentation | ADE20K | mIoU31.8 | 936 | |
| Semantic segmentation | PASCAL Context (val) | mIoU62 | 323 | |
| Semantic segmentation | ADE20K A-150 | mIoU37.9 | 188 | |
| Semantic segmentation | Pascal Context 59 | mIoU63.3 | 164 | |
| Semantic segmentation | PASCAL-Context 59 class (val) | mIoU63.3 | 125 | |
| Semantic segmentation | Pascal VOC 20 | mIoU97 | 105 | |
| Semantic segmentation | Pascal VOC 21 classes (val) | mIoU77.3 | 103 | |
| Semantic segmentation | GTA5 to {Cityscapes, Mapillary, BDD} (test) | mIoU (Cityscapes)57.3 | 94 | |
| Semantic segmentation | Stanford2D3DS (3-fold cross-validation) | mIoU39.6 | 90 | |
| Semantic segmentation | ADE20K 847 | mIoU1.60e+3 | 83 |