Novel Visual Category Discovery with Dual Ranking Statistics and Mutual Knowledge Distillation
About
In this paper, we tackle the problem of novel visual category discovery, i.e., grouping unlabelled images from new classes into different semantic partitions by leveraging a labelled dataset that contains images from other different but relevant categories. This is a more realistic and challenging setting than conventional semi-supervised learning. We propose a two-branch learning framework for this problem, with one branch focusing on local part-level information and the other branch focusing on overall characteristics. To transfer knowledge from the labelled data to the unlabelled, we propose using dual ranking statistics on both branches to generate pseudo labels for training on the unlabelled data. We further introduce a mutual knowledge distillation method to allow information exchange and encourage agreement between the two branches for discovering new categories, allowing our model to enjoy the benefits of global and local features. We comprehensively evaluate our method on public benchmarks for generic object classification, as well as the more challenging datasets for fine-grained visual recognition, achieving state-of-the-art performance.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Generalized Category Discovery | CIFAR-100 | -- | 133 | |
| Open-world semi-supervised learning | CIFAR-100 (test) | Overall Accuracy23.1 | 40 | |
| Open-world semi-supervised learning | CIFAR-10 (test) | Overall Accuracy82.9 | 28 | |
| Novel Class Discovery | CIFAR-100 | ACC (Seen)0.4608 | 19 | |
| Clustering | ImageNet unlabelled (train) | Clustering Accuracy88.9 | 14 | |
| Clustering | CIFAR10 unlabelled (train) | Clustering Accuracy91.6 | 14 | |
| Clustering | CIFAR100-20 unlabelled (train) | Clustering Accuracy75.3 | 13 | |
| Continuous Novel Category Discovery | ImageNet-100 | Mf Score47.02 | 8 | |
| Continuous Novel Category Discovery | Stanford Cars | Mf50.7 | 8 | |
| Continuous Novel Category Discovery | FGVC Aircraft | Mf Score38.89 | 8 |