少样本下自动发现新类别,让模型像人一样快速学习新东西。
Few-shot Novel Category Discovery
- 基于少量样本支持,动态切换已知类识别与未知类聚类任务。
- 在五个数据集上表现领先,适应不同场景的开放世界设定。
- 适合研究少样本学习、开放集识别与自监督发现的学者。
近期提出的新型类别发现(NCD)采用归纳式学习范式,限制了其在真实场景中的应用。实际上,仅需对部分新类别提供少量标注数据,即可显著缓解这一问题,这与人类能快速标注少量新类别数据的现象一致。因此,本文提出一种新设置:训练好的模型能够随着查询样本数量增加,灵活切换已知类别识别与完全无标签的新类别聚类任务,利用仅少数(少量)支持样本中学到的知识。受基于先验聚类算法发现新类别的启发,我们引入一个新框架,进一步将假设放宽至真实世界的开放集级别,通过统一少样本学习中的模型可适应性概念。我们将该设置称为少样本新型类别发现(FSNCD),并提出半监督分层聚类(SHC)与不确定性感知K均值聚类(UKC)以检验模型推理能力。在五个常用数据集上的大量实验和详细分析表明,所提方法在不同任务设置和场景下均能达到领先性能。
原文摘要 · Abstract (English)
The recently proposed Novel Category Discovery (NCD) adapt paradigm of transductive learning hinders its application in more real-world scenarios. In fact, few labeled data in part of new categories can well alleviate this burden, which coincides with the ease that people can label few of new category data. Therefore, this paper presents a new setting in which a trained agent is able to flexibly switch between the tasks of identifying examples of known (labelled) classes and clustering novel (completely unlabeled) classes as the number of query examples increases by leveraging knowledge learned from only a few (handful) support examples. Drawing inspiration from the discovery of novel categories using prior-based clustering algorithms, we introduce a novel framework that further relaxes its assumptions to the real-world open set level by unifying the concept of model adaptability in few-shot learning. We refer to this setting as Few-Shot Novel Category Discovery (FSNCD) and propose Semi-supervised Hierarchical Clustering (SHC) and Uncertainty-aware K-means Clustering (UKC) to examine the model's reasoning capabilities. Extensive experiments and detailed analysis on five commonly used datasets demonstrate that our methods can achieve leading performance levels across different task settings and scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。