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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Generalized Category Discovery | CUB | Accuracy (All)55.2 | 186 | |
| Category Discovery | CUB-200 2011 | Overall Score76.02 | 87 | |
| Category Discovery | CIFAR-100 | Accuracy (All Categories)77.68 | 39 | |
| Continual Category Discovery | Average fine-grained | cACC (All)81.49 | 32 | |
| Continual Category Discovery | FGVC Aircraft | cACC (All)71.4 | 16 | |
| Continual Category Discovery | Stanford Cars | cACC (All)70.35 | 16 | |
| Continual Category Discovery | ImageNet-100 | cACC (All)83.39 | 16 | |
| Continual Category Discovery | TinyImageNet | cACC (All)73.02 | 16 | |
| Continual Category Discovery | Caltech-101 | cACC (All)91.85 | 16 | |
| Novel Class Discovery | Tiny-ImageNet | Accuracy (Seen)64.57 | 12 |