Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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.

Bingchen Zhao, Kai Han• 2021

Related benchmarks

TaskDatasetResultRank
Generalized Category DiscoveryCIFAR-100--
133
Open-world semi-supervised learningCIFAR-100 (test)
Overall Accuracy23.1
40
Open-world semi-supervised learningCIFAR-10 (test)
Overall Accuracy82.9
28
Novel Class DiscoveryCIFAR-100
ACC (Seen)0.4608
19
ClusteringImageNet unlabelled (train)
Clustering Accuracy88.9
14
ClusteringCIFAR10 unlabelled (train)
Clustering Accuracy91.6
14
ClusteringCIFAR100-20 unlabelled (train)
Clustering Accuracy75.3
13
Continuous Novel Category DiscoveryImageNet-100
Mf Score47.02
8
Continuous Novel Category DiscoveryStanford Cars
Mf50.7
8
Continuous Novel Category DiscoveryFGVC Aircraft
Mf Score38.89
8
Showing 10 of 24 rows

Other info

Code

Follow for update