论文

广义类别发现中的语义错配问题:评估与缓解方法

Semantic Mismapping in Generalized Category Discovery: Evaluation and Mitigation

精选理由

训练用的监督信号会让评估指标悄悄虚高,这篇论文发现新类准确率能被高估 25 个点,还给了修复的评估和训练方法,做半监督聚类的值得看看。

论文研究广义类别发现(GCD)任务中评估指标的缺陷:新类别的索引可交换而旧类别索引被监督信号锚定后,全局匈牙利匹配(GH)会把锚定的旧索引重新分配,导致新类别准确率被高估,最高达 25.63 个百分点。作者将这一现象命名为语义错配(semantic mismapping),并推导出置换环条件来刻画何时会抬高匹配分数。为修复评估,论文提出语义保持评估(SPE),将匈牙利匹配限制在新类别区域。同时提出错配缓解优化(MMO),由区域平衡损失(RBL)和新类多样性损失(NDL)组成,理论上可收紧虚假准确率增益的上界。

原文 · arXiv cs.LG

Semantic Mismapping in Generalized Category Discovery: Evaluation and Mitigation

Generalized Category Discovery (GCD) aims to perform old-category recognition and novel-category discovery in the same unlabeled set. Early methods address both objectives through clustering and evaluate the exchangeable predicted indices with global Hungarian (GH) matching. Recent methods adopt trainable parametric heads, using supervision for old-category recognition and discovery-oriented objectives for new categories. In this way, the output space becomes partially exchangeable: old indices are semantically anchored, while new indices remain exchangeable. However, these methods still apply global matching, which can reassign anchored old indices to other categories and hide some prediction errors. Empirically, we are surprised to observe that this reassignment can induce an overestimation of new-category accuracy by up to 25.63 points. We term this phenomenon semantic mismapping and further derive a permutation-cycle condition that characterizes when semantic mismapping can increase the global matching score. To fix this issue, we propose Semantics-Preserving Evaluation (SPE), which preserves the supervised semantics of old indices and restricts Hungarian matching to the new region. Furthermore, we propose Mismapping-Mitigating Optimization (MMO) to alleviate the prediction errors that can give rise to semantic mismapping, combining a region-balance loss (RBL) for old/new routing with a novel-diversity loss (NDL) that encourages orthogonality among new prototypes. Theoretically, we show that minimizing the two losses can tighten an upper bound on spurious accuracy gains from semantic mismapping, with controllable residual terms. Empirical results under both GH and SPE demonstrate the effectiveness of our proposed method.