Multi-Modal Representation Learning via Semi-Supervised Rate Reduction for Generalized Category Discovery
Organizations: School of Artificial Intelligence, Beijing University of Posts and Telecommunications · Intellifusion Inc., Shenzhen, P.R. China
Abstract
Generalized Category Discovery (GCD) aims to identify both known and unknown categories, with only partial labels given for the known categories, posing a challenging open-set recognition problem. State-of-the-art approaches for GCD task are usually built on multi-modality representation learning, which is heavily dependent upon inter-modality alignment. However, few of them cast a proper intra-modality alignment to generate a desired underlying structure of representation distributions. In this paper, we propose a novel and effective multi-modal representation learning framework for GCD via Semi-Supervised Rate Reduction, called SSR-GCD, to learn cross-modality representations with desired structural properties based on emphasizing to properly align intra-modality relationships. Moreover, to boost knowledge transfer, we integrate prompt candidates by leveraging the inter-modal alignment offered by Vision Language Models. We conduct extensive experiments on generic and fine-grained benchmark datasets demonstrating superior performance of our approach.
Figures & tables
| ImageNet-100 | ImageNet-1k | ||||||
|---|---|---|---|---|---|---|---|
| Method | All | Old | New | All | Old | New | |
| DINOv1 | GCD | 74.1 | 89.8 | 66.3 | 52.5 | 72.5 | 42.2 |
| SimGCD | 83.0 | 93.1 | 77.9 | 57.1 | 77.3 | 46.9 | |
| SPTNet | 85.4 | 93.2 | 81.4 | - | - | - | |
| SelEx | 83.1 | 93.6 | 77.8 | - | - | - | |
| Hyp-SelEx | 86.8 | 94.6 | 82.8 | - | - | - | |
| CIFAR-10 | CIFAR-100 | CUB | Stanford Cars | Oxford Pets | Flowers102 | ||||||||||||||
| Method | All | Old | New | All | Old | New | All | Old | New | All | Old | New | All | Old | New | All | Old | New | |
| DINOv1 | GCD | 91.5 | 97.9 | 88.2 | 73.0 | 76.2 | 66.5 | 51.3 | 56.6 | 48.7 | 39.0 | 57.6 | 29.9 | 80.2 | 85.1 | 77.6 | 74.4 | 74.9 | 74.1 |
| GPC | 92.2 | 98.2 | 89.1 | 77.9 | 85.0 | 63.0 | 55.4 | 58.2 | 53.1 | 42.8 | 59.2 | 32.8 | - | - | - | - | - | - | |
| SimGCD | 97.1 | 95.1 | 98.1 | 80.1 | 81.2 | 77.8 | 60.3 | 65.6 | 57.7 | 53.8 | 71.9 | 45.0 | 87.7 | 85.9 | 88.6 | 71.3 | 80.9 | 66.5 | |
| PromptCAL | 97.9 | 96.6 | 98.5 | 81.2 | 84.2 | 75.3 | 62.9 | 64.4 | 62.1 | 50.2 | 70.1 | 40.6 | - | - | - | - | - | - | |
| SPTNet | 97.3 | 95.0 | 98.6 | 81.3 | 84.3 | 75.6 | 65.8 | 68.8 | 65.1 | 59.0 | 79.2 | 49.3 | - | - | - | - | - | - | |
| Rep. Losses | Inter | Intra | CIFAR-10 | CIFAR-100 | CUB | Stanford Cars | Oxford Pets | Flowers102 |
|---|---|---|---|---|---|---|---|---|
| N/A | 97.9 | 84.1 | 74.5 | 86.0 | 91.9 | 87.4 | ||
| 98.3 | 86.0 | 76.7 | 87.0 | 94.1 | 89.7 | |||
| 98.4 | 86.7 | 77.5 | 87.9 | 94.9 | 91.8 | |||
| \mathcal{L}_{\text{SSR{}^{2}}} | 98.5 | 86.4 | 78.3 | 89.2 | 95.7 | 93.5 | ||
| + | 98.2 | 86.3 | 78.0 | 86.7 | 95.0 | 90.9 | ||
| + \mathcal{L}_{\text{SSR{}^{2}}} | 98.3 | 86.1 | 77.2 | 88.1 | 95.0 | 92.9 |
| Dual | RTA | \mathcal{L}_{\text{SSR{}^{2}}} | Stanford Cars | Flowers102 | ||||
|---|---|---|---|---|---|---|---|---|
| All | Old | New | All | Old | New | |||
| 75.2 | 85.4 | 71.8 | 78.3 | 88.1 | 72.5 | |||
| 81.7 | 90.3 | 77.1 | 83.9 | 88.3 | 81.3 | |||
| 86.0 | 91.0 | 83.4 | 87.4 | 90.8 | 85.5 | |||
| 85.5 | 91.7 | 82.2 | 89.1 | 91.8 | 88.0 | |||
| 89.2 | 93.1 | 87.3 | 93.5 | 93.3 | 93.9 | |||
Appendix figures & tables19 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Labelled | Unlabelled | ||
|---|---|---|---|---|
| #Image | #Class | #Image | #Class | |
| CIFAR10 | 12.5K | 5 | 37.5K | 10 |
| CIFAR100 | 20.0K | 80 | 30.0K | 100 |
| ImageNet-100 | 31.9K | 50 | 95.3K | 100 |
| ImageNet-1K | 321K | 500 | 960K | 1000 |
| CUB | 1.5K | 100 | 4.5K | 200 |
| CLIP | Last residual attention block: |
|---|---|
| Image/text projector: | |
| Cls. | Linear projection: |
| Softmax function |
| from torchvision.transforms import * |
|---|
| Compose([ |
| RandomResizedCrop(32, BILINEAR) |
| RandomHorizontalFlip(p=0.5), |
| ColorJitter() |
| ToTensor(), |
| Normalize([0.485, 0.456, 0.406], |
| Input: text |
|---|
| For each word in text : |
| If len(word) 3 : |
| index random(1, len(word) 2) |
| action random({replace, delete, add, none}) |
| Case action : |
| replace : word replace random char at index |
| CIFAR-10 | CIFAR-100 | ImageNet-100 | ImageNet-1k | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Method | All | Old | New | All | Old | New | All | Old | New | All | Old | New | |
| TextGCD | 98.2 | 98.0 | 98.6 | 85.7 | 86.3 | 84.6 | 88.0 | 92.4 | 85.2 | 64.8 | 77.8 | 58.3 | |
| GET | 97.2 | 94.6 | 98.5 | 82.1 | 85.5 | 75.5 | 91.7 | 95.7 | 89.7 | 62.4 | 74.0 | 56.6 | |
| Ours | 98.5 | 98.3 | 98.6 | 86.4 | 86.2 | 86.9 | 92.1 | 96.0 | 90.2 | 66.7 | 77.3 | 61.1 | |
| CUB | Stanford Cars | Oxford Pets | Flowers102 | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Method | All | Old | New | All | Old | New | All | Old | New | All | Old | New | |
| TextGCD | 76.6 | 80.6 | 74.7 | 86.1 | 91.8 | 83.9 | 93.7 | 93.2 | 94.0 | 87.2 | 90.7 | 85.4 | |
| GET | 77.0 | 78.1 | 76.4 | 78.5 | 86.8 | 74.5 | 91.1 | 89.7 | 92.4 | 85.5 | 90.8 | 81.3 | |
| Ours | 78.3 | 78.5 | 78.2 | 89.2 | 93.1 | 87.3 | 95.7 | 95.1 | 96.0 | 93.5 | 93.3 | 93.9 | |
| Dataset | # Unknown categories | # Removed tags |
|---|---|---|
| CUB-200 | 100 | 90 |
| Stanford Cars | 98 | 83 |
| Flowers102 | 51 | 43 |
| Oxford Pets | 19 | 17 |
| Datasets | CUB | Cars | Pets | Flowers |
|---|---|---|---|---|
| TexGCD [ 32 ] | ||||
| TextGCD (w/o) | ||||
| SSR 2 -GCD (w/o) |
| Rep. Losses | Inter | Intra | C-10 | C-100 | CUB | Cars | Pets | Flowers |
|---|---|---|---|---|---|---|---|---|
| N/A | 97.9 | 84.1 | 74.5 | 86.0 | 91.9 | 87.4 | ||
| 98.3 | 86.0 | 76.6 | 86.7 | 93.9 | 89.7 | |||
| 98.0 | 85.0 | 76.4 | 86.1 | 94.9 | 87.2 | |||
| 98.4 | 86.7 | 77.5 | 87.9 | 94.9 | 91.8 | |||
| 98.5 | 86.4 | 78.3 | 89.2 | 95.7 | 93.5 | |||
| + | 98.2 | 86.3 | 78.0 | 86.7 | 95.0 | 90.9 |
| Text retrieval models | Prompt qualities | Stanford Cars | Flowers102 | ||||
|---|---|---|---|---|---|---|---|
| CLIP-H/14 | High | ||||||
| CLIP-L/14 | Medium | ||||||
| CLIP-B/16 | Low | ||||||
| CIFAR-10 | CIFAR-100 | CUB | Stanford Cars | Oxford Pets | Flowers102 | |||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Method | All | Old | New | All | Old | New | All | Old | New | All | Old | New | All | Old | New | All | Old | New |
| GCD | 91.5 | 97.9 | 88.2 | 73.0 | 76.2 | 66.5 | 51.3 | 56.6 | 48.7 | 39.0 | 57.6 | 29.9 | 80.2 | 85.1 | 77.6 | 74.4 | 74.9 | 74.1 |
| GCD+SSR 2 | 92.5 | 96.4 | 91.6 | 73.9 | 79.0 | 63.2 | 51.9 | 55.0 | 47.1 | 47.9 | 56.1 | 47.3 | 83.6 | 87.7 | 79.8 | 80.0 | 83.3 | 78.5 |
| SimGCD | 97.1 | 95.1 | 98.1 | 80.1 | 81.2 | 77.8 | 60.3 | 65.6 | 57.7 | 53.8 | 71.9 | 45.0 | 87.7 | 85.9 | 88.6 | 71.3 | 80.9 | 66.5 |
| SimGCD+SSR 2 | 97.6 | 97.5 | 97.7 | 81.1 | 82.5 | 78.9 | 60.8 | 64.7 | 59.0 | 57.1 | 66.8 | 53.9 | 90.0 | 89.8 | 91.2 | 81.6 | 83.5 | 80.1 |
| Methods | Running Time (sec/iter) | Memory (MB) | |||
|---|---|---|---|---|---|
| Forward | Backward | Overall | w/o RTA † | w/ RTA | |
| TextGCD [ 32 ] | 7,622 | - | |||
| SSR 2 -GCD | 8,035 | 11,080 | |||
Generalized Category Discovery
Generalized category discovery learns known classes from labeled examples while grouping unlabeled data into known and previously unseen categories.
See all 25 papers