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

Talking to DINO: Bridging Self-Supervised Vision Backbones with Language for Open-Vocabulary Segmentation

About

Open-Vocabulary Segmentation (OVS) aims at segmenting images from free-form textual concepts without predefined training classes. While existing vision-language models such as CLIP can generate segmentation masks by leveraging coarse spatial information from Vision Transformers, they face challenges in spatial localization due to their global alignment of image and text features. Conversely, self-supervised visual models like DINO excel in fine-grained visual encoding but lack integration with language. To bridge this gap, we present Talk2DINO, a novel hybrid approach that combines the spatial accuracy of DINOv2 with the language understanding of CLIP. Our approach aligns the textual embeddings of CLIP to the patch-level features of DINOv2 through a learned mapping function without the need to fine-tune the underlying backbones. At training time, we exploit the attention maps of DINOv2 to selectively align local visual patches with textual embeddings. We show that the powerful semantic and localization abilities of Talk2DINO can enhance the segmentation process, resulting in more natural and less noisy segmentations, and that our approach can also effectively distinguish foreground objects from the background. Experimental results demonstrate that Talk2DINO achieves state-of-the-art performance across several unsupervised OVS benchmarks. Source code and models are publicly available at: https://lorebianchi98.github.io/Talk2DINO/.

Luca Barsellotti, Lorenzo Bianchi, Nicola Messina, Fabio Carrara, Marcella Cornia, Lorenzo Baraldi, Fabrizio Falchi, Rita Cucchiara• 2024

Related benchmarks

TaskDatasetResultRank
Semantic segmentationCOCO Stuff
mIoU27.89
379
Semantic segmentationADE20K
mIoU21.7
366
Semantic segmentationCityscapes
mIoU41.15
218
Semantic segmentationPascal Context
mIoU40.31
217
Semantic segmentationPascal VOC
mIoU85.68
129
Open Vocabulary Semantic SegmentationPascal VOC 20
mIoU87.1
104
Open Vocabulary Semantic SegmentationCOCOStuff (val)
mIoU30.2
60
3D Semantic SegmentationScanNet
mIoU29.13
51
Open Vocabulary Semantic SegmentationPASCAL Context 59 (val)
mIoU42.4
49
Open Vocabulary Semantic SegmentationCityscapes
mIoU36.6
43
Showing 10 of 27 rows

Other info

Follow for update