Language-Driven Visual Consensus for Zero-Shot Semantic Segmentation

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

研究成果: Article同行評審

摘要

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.

原文English
期刊IEEE Transactions on Circuits and Systems for Video Technology
DOIs
出版狀態Accepted/In press - 2024
對外發佈

指紋

深入研究「Language-Driven Visual Consensus for Zero-Shot Semantic Segmentation」主題。共同形成了獨特的指紋。

引用此