Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Parameter-efficient Fine-tuning in Hyperspherical Space for Open-vocabulary Semantic Segmentation

About

Open-vocabulary semantic segmentation seeks to label each pixel in an image with arbitrary text descriptions. Vision-language foundation models, especially CLIP, have recently emerged as powerful tools for acquiring open-vocabulary capabilities. However, fine-tuning CLIP to equip it with pixel-level prediction ability often suffers three issues: 1) high computational cost, 2) misalignment between the two inherent modalities of CLIP, and 3) degraded generalization ability on unseen categories. To address these issues, we propose H-CLIP a symmetrical parameter-efficient fine-tuning (PEFT) strategy conducted in hyperspherical space for both of the two CLIP modalities. Specifically, the PEFT strategy is achieved by a series of efficient block-diagonal learnable transformation matrices and a dual cross-relation communication module among all learnable matrices. Since the PEFT strategy is conducted symmetrically to the two CLIP modalities, the misalignment between them is mitigated. Furthermore, we apply an additional constraint to PEFT on the CLIP text encoder according to the hyperspherical energy principle, i.e., minimizing hyperspherical energy during fine-tuning preserves the intrinsic structure of the original parameter space, to prevent the destruction of the generalization ability offered by the CLIP text encoder. Extensive evaluations across various benchmarks show that H-CLIP achieves new SOTA open-vocabulary semantic segmentation results while only requiring updating approximately 4% of the total parameters of CLIP.

Zelin Peng, Zhengqin Xu, Zhilin Zeng, Yaoming Wang, Wei Shen• 2024

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K A-150
mIoU38.4
217
Semantic segmentationPascal Context 59
mIoU64.1
204
Semantic segmentationPascal VOC 20
mIoU97.7
130
Semantic segmentationADE20K 847
mIoU16.5
105
Semantic segmentationPascal Context 459
mIoU24.2
82
Open Vocabulary Semantic SegmentationADE20K A-150
mIoU38.4
71
Open Vocabulary Semantic SegmentationPASCAL Context 59 (val)
mIoU64.1
49
Open Vocabulary Semantic SegmentationADE20K 847 (val)
mIoU16.5
17
Open Vocabulary Semantic SegmentationPASCAL Context 459 (val)
mIoU24.2
17
Open Vocabulary Semantic SegmentationPASCAL VOC-20 (val)
mIoU97.7
15
Showing 10 of 12 rows

Other info

Follow for update