Language-Driven Visual Consensus for Zero-Shot Semantic Segmentation

Zicheng Zhang, Wei Ke, Yi Zhu, Xiaodan Liang, Jianzhuang Liu, Qixiang Ye, Tong Zhang

Research output: Contribution to journalArticlepeer-review

Abstract

The pre-trained vision-language model, exemplified by CLIP [1], advances zero-shot semantic segmentation by aligning visual features with class embeddings through a transformer decoder to generate semantic masks. Despite its effectiveness, prevailing methods within this paradigm encounter challenges, including overfitting on seen classes and small fragmentation in segmentation masks. To mitigate these issues, we propose a Language-Driven Visual Consensus (LDVC) approach, fostering improved alignment of linguistic and visual information. Specifically, we leverage class embeddings as anchors due to their discrete and abstract nature, steering visual features toward class embeddings. Moreover, to achieve a more compact visual space, we introduce route attention into the transformer decoder to find visual consensus, thereby enhancing semantic consistency within the same object. Equipped with a vision-language prompting strategy, our approach significantly boosts the generalization capacity of segmentation models for unseen classes. Experimental results underscore the effectiveness of our approach, showcasing mIoU gains of 4.5% on the PASCAL VOC 2012 and 3.6% on the COCO-Stuff 164K for unseen classes compared with the state-of- the-art methods.

Original languageEnglish
JournalIEEE Transactions on Circuits and Systems for Video Technology
DOIs
Publication statusAccepted/In press - 2024
Externally publishedYes

Keywords

  • semantic segmentation
  • Vision-language model
  • vision-language prompt tuning
  • zero-shot

Fingerprint

Dive into the research topics of 'Language-Driven Visual Consensus for Zero-Shot Semantic Segmentation'. Together they form a unique fingerprint.

Cite this