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

GET: Unlocking the Multi-modal Potential of CLIP for Generalized Category Discovery

About

Given unlabelled datasets containing both old and new categories, generalized category discovery (GCD) aims to accurately discover new classes while correctly classifying old classes. Current GCD methods only use a single visual modality of information, resulting in a poor classification of visually similar classes. As a different modality, text information can provide complementary discriminative information, which motivates us to introduce it into the GCD task. However, the lack of class names for unlabelled data makes it impractical to utilize text information. To tackle this challenging problem, in this paper, we propose a Text Embedding Synthesizer (TES) to generate pseudo text embeddings for unlabelled samples. Specifically, our TES leverages the property that CLIP can generate aligned vision-language features, converting visual embeddings into tokens of the CLIP's text encoder to generate pseudo text embeddings. Besides, we employ a dual-branch framework, through the joint learning and instance consistency of different modality branches, visual and semantic information mutually enhance each other, promoting the interaction and fusion of visual and text knowledge. Our method unlocks the multi-modal potentials of CLIP and outperforms the baseline methods by a large margin on all GCD benchmarks, achieving new state-of-the-art. Our code is available at: https://github.com/enguangW/GET.

Enguang Wang, Zhimao Peng, Zhengyuan Xie, Fei Yang, Xialei Liu, Ming-Ming Cheng• 2024

Related benchmarks

TaskDatasetResultRank
Generalized Category DiscoveryImageNet-100
All Accuracy91.7
208
Generalized Category DiscoveryCIFAR-100
Accuracy (All)82.1
185
Generalized Category DiscoveryStanford Cars
Accuracy (All)78.5
160
Generalized Category DiscoveryCUB
Accuracy (All)77
133
Generalized Category DiscoveryFGVC Aircraft
Accuracy (All)58.9
105
Generalized Category DiscoveryCIFAR-10
All Accuracy97.2
105
Generalized Category DiscoveryHerbarium19
Score (All Categories)49.7
71
Generalized Category DiscoveryOxford Pets
Accuracy (All)91.1
50
Generalized Category DiscoveryImageNet-1K
Accuracy (All)62.4
19
Generalized Category DiscoveryFine-grained Avg
Overall Accuracy71.5
12
Showing 10 of 13 rows

Other info

Follow for update