arXiv:2504.03755cs.LGcs.AI2025-04TPAMI被引 31

统一建模新旧类别,解决通用类别发现中的偏差问题。

ProtoGCD: Unified and Unbiased Prototype Learning for Generalized Category Discovery

  • 用联合原型统一建模新旧类别,避免分类偏差。
  • 双层自适应伪标签机制提升聚类准确性,优于现有方法。
  • 可自动估计新类别数量,适合实际场景应用。

通用类别发现(GCD)是一个实用但研究不足的问题,要求模型利用旧类别的有标签样本自动聚类并发现新类别。挑战在于无标签数据同时包含旧类和新类。早期方法使用参数化分类器进行伪标签,分别处理新旧类别,导致准确率不平衡;近期方法采用对比学习,却忽视潜在正样本且与聚类目标解耦,造成表征偏差和次优结果。为此,本文提出统一且无偏的原型学习框架ProtoGCD,将新旧类别统一建模于共享原型下,通过统一的学习目标实现联合建模。具体地,设计双层自适应伪标签机制缓解确认偏差,并引入两项正则化项共同优化表示学习。此外,针对实际需求,提出新类别数量估计准则;进一步扩展至检测未见异常值,实现任务级统一。大量实验表明,ProtoGCD在通用与细粒度数据集上均达到当前最优性能,代码已开源。

原文摘要 · Abstract (English)

Generalized category discovery (GCD) is a pragmatic but underexplored problem, which requires models to automatically cluster and discover novel categories by leveraging the labeled samples from old classes. The challenge is that unlabeled data contain both old and new classes. Early works leveraging pseudo-labeling with parametric classifiers handle old and new classes separately, which brings about imbalanced accuracy between them. Recent methods employing contrastive learning neglect potential positives and are decoupled from the clustering objective, leading to biased representations and sub-optimal results. To address these issues, we introduce a unified and unbiased prototype learning framework, namely ProtoGCD, wherein old and new classes are modeled with joint prototypes and unified learning objectives, {enabling unified modeling between old and new classes}. Specifically, we propose a dual-level adaptive pseudo-labeling mechanism to mitigate confirmation bias, together with two regularization terms to collectively help learn more suitable representations for GCD. Moreover, for practical considerations, we devise a criterion to estimate the number of new classes. Furthermore, we extend ProtoGCD to detect unseen outliers, achieving task-level unification. Comprehensive experiments show that ProtoGCD achieves state-of-the-art performance on both generic and fine-grained datasets. The code is available at https://github.com/mashijie1028/ProtoGCD.

类别发现原型学习聚类无监督

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。