不用存储旧数据,用聚类和预训练模型发现新类别
Freeze and Cluster: A Simple Baseline for Rehearsal-Free Continual Category Discovery
- 用预训练模型+k-means聚类生成伪标签,仅更新分类器层
- 在多个数据集上达到当前最佳性能,验证了方法有效性
- 适合研究持续学习与新类别发现的学者参考
本文针对无需重放的持续类别发现(RF-CCD)问题,提出一种简单但高效的基线方法。现有方法通常从头训练,忽视预训练模型潜力,且依赖数据存储防止遗忘。由于RF-CCD同时涉及持续学习与新类别发现,以往方法难以有效融合两领域先进技术。我们首次整合两领域进展,并在Stanford Cars、CUB、iNat、Tiny-ImageNet等多个基准上进行广泛实验。结果表明,在预训练模型基础上引入无标签数据可能反而导致表征退化。为此,我们提出先利用已知类别估计新类别数,再用基础类模型提取特征,通过k-means生成高质量伪标签,仅训练分类器层。实验充分验证了该方法的有效性与结论的可靠性。
原文摘要 · Abstract (English)
This paper addresses the problem of Rehearsal-Free Continual Category Discovery (RF-CCD), which focuses on continuously identifying novel class by leveraging knowledge from labeled data. Existing methods typically train from scratch, overlooking the potential of base models, and often resort to data storage to prevent forgetting. Moreover, because RF-CCD encompasses both continual learning and novel class discovery, previous approaches have struggled to effectively integrate advanced techniques from these fields, resulting in less convincing comparisons and failing to reveal the unique challenges posed by RF-CCD. To address these challenges, we lead the way in integrating advancements from both domains and conducting extensive experiments and analyses. Our findings demonstrate that this integration can achieve state-of-the-art results, leading to the conclusion that in the presence of pre-trained models, the representation does not improve and may even degrade with the introduction of unlabeled data. To mitigate representation degradation, we propose a straightforward yet highly effective baseline method. This method first utilizes prior knowledge of known categories to estimate the number of novel classes. It then acquires representations using a model specifically trained on the base classes, generates high-quality pseudo-labels through k-means clustering, and trains only the classifier layer. We validate our conclusions and methods by conducting extensive experiments across multiple benchmarks, including the Stanford Cars, CUB, iNat, and Tiny-ImageNet datasets. The results clearly illustrate our findings, demonstrate the effectiveness of our baseline, and pave the way for future advancements in RF-CCD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。