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

Effective Prompt Pool Learning for Continual Category Discovery

About

This paper studies effective prompt pool learning for Continual Category Discovery (CCD), a challenging open-world setting where a model must discover novel categories from a continuous stream of unlabelled data containing both known and novel classes, while mitigating catastrophic forgetting of previously learned concepts. We introduce a series of novel prompt-pool-based frameworks for CCD, each exploring a different design of prompt pools. First, we propose PromptCCD, which focuses on global class prototypes via a Gaussian Mixture Prompt (GMP) module. GMP fits a generative Gaussian mixture model over feature embeddings, where each mixture component serves as both a class prototype and a dynamic prompt that conditions the backbone's representations. This design enables label-free prompt selection and on-the-fly estimation of the number of emerging categories. Through a systematic spectrum study, we then show that category count, rather than sample size, is the primary bottleneck for discovery performance, motivating the need for finer-grained representations. Building on this finding, we propose PromptCCD++, which focuses on object-part prototypes via Part-level Prompting (PLP) modules. PLP decomposes prompt pool into multiple, specialized part-level prompt pools. During discovery phase, these pools dynamically assign part-specific prompts to local object regions without the need for manual part annotations, enabling the model to learn object-part representations that boost category discovery. Extensive evaluations on both generic and fine-grained benchmarks, supported by comprehensive ablation studies, demonstrate the effectiveness of our framework for CCD.

Fernando Julio Cendra, Xinghui Li, Kai Han• 2024

Related benchmarks

TaskDatasetResultRank
Generalized Category DiscoveryCUB
Accuracy (All)55.2
186
Category DiscoveryCUB-200 2011
Overall Score76.02
87
Category DiscoveryCIFAR-100
Accuracy (All Categories)77.68
39
Continual Category DiscoveryAverage fine-grained
cACC (All)81.49
32
Continual Category DiscoveryFGVC Aircraft
cACC (All)71.4
16
Continual Category DiscoveryStanford Cars
cACC (All)70.35
16
Continual Category DiscoveryImageNet-100
cACC (All)83.39
16
Continual Category DiscoveryTinyImageNet
cACC (All)73.02
16
Continual Category DiscoveryCaltech-101
cACC (All)91.85
16
Novel Class DiscoveryTiny-ImageNet
Accuracy (Seen)64.57
12
Showing 10 of 11 rows

Other info

Follow for update