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

Parametric Classification for Generalized Category Discovery: A Baseline Study

About

Generalized Category Discovery (GCD) aims to discover novel categories in unlabelled datasets using knowledge learned from labelled samples. Previous studies argued that parametric classifiers are prone to overfitting to seen categories, and endorsed using a non-parametric classifier formed with semi-supervised k-means. However, in this study, we investigate the failure of parametric classifiers, verify the effectiveness of previous design choices when high-quality supervision is available, and identify unreliable pseudo-labels as a key problem. We demonstrate that two prediction biases exist: the classifier tends to predict seen classes more often, and produces an imbalanced distribution across seen and novel categories. Based on these findings, we propose a simple yet effective parametric classification method that benefits from entropy regularisation, achieves state-of-the-art performance on multiple GCD benchmarks and shows strong robustness to unknown class numbers. We hope the investigation and proposed simple framework can serve as a strong baseline to facilitate future studies in this field. Our code is available at: https://github.com/CVMI-Lab/SimGCD.

Xin Wen, Bingchen Zhao, Xiaojuan Qi• 2022

Related benchmarks

TaskDatasetResultRank
Generalized Category DiscoveryImageNet-100
All Accuracy89.9
236
Generalized Category DiscoveryCIFAR-100
Accuracy (All)88.5
233
Generalized Category DiscoveryStanford Cars
Accuracy (All)75.9
208
Generalized Category DiscoveryCUB
Accuracy (All)75.9
186
Generalized Category DiscoveryCIFAR-10
All Accuracy98.7
152
Generalized Category DiscoveryFGVC Aircraft
Accuracy (All)63.9
115
Category DiscoveryCUB-200 2011
Overall Score39.89
87
Generalized Category DiscoveryCUB-200 (test)
Overall Accuracy71.5
81
Generalized Category DiscoveryHerbarium19
Score (All Categories)47.9
71
Generalized Category DiscoveryHerbarium19 (test)
Score (All Categories)44
52
Showing 10 of 80 rows
...

Other info

Code

Follow for update