Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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.

Seokju Cho, Heeseong Shin, Sunghwan Hong, Anurag Arnab, Paul Hongsuck Seo, Seungryong Kim• 2023

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K
mIoU31.8
936
Semantic segmentationPASCAL Context (val)
mIoU62
323
Semantic segmentationADE20K A-150
mIoU37.9
188
Semantic segmentationPascal Context 59
mIoU63.3
164
Semantic segmentationPASCAL-Context 59 class (val)
mIoU63.3
125
Semantic segmentationPascal VOC 20
mIoU97
105
Semantic segmentationPascal VOC 21 classes (val)
mIoU77.3
103
Semantic segmentationGTA5 to {Cityscapes, Mapillary, BDD} (test)
mIoU (Cityscapes)57.3
94
Semantic segmentationStanford2D3DS (3-fold cross-validation)
mIoU39.6
90
Semantic segmentationADE20K 847
mIoU1.60e+3
83
Showing 10 of 139 rows
...

Other info

Code

Follow for update