このホワイトペーパーでは、既知のクラスを持つラベル付きデータセットを指定して、ラベルなしサンプルのセットで新しいクラスを公開するタスクである、Novel Class Discovery(NCD)について説明します。 NCDの特性を利用して、Neighborhood Contrastive Learning(NCL)という名前の新しいフレームワークを構築し、クラスタリングのパフォーマンスに重要な識別表現を学習します。私たちの貢献は2つあります。まず、ラベル付きセットでトレーニングされた特徴抽出器が、ジェネリッククエリサンプルとその隣接サンプルが同じクラスを共有する可能性が高い表現を生成することがわかります。この観察結果を利用して、対照的な学習で疑似陽性ペアを取得および集約し、モデルがより識別力のある表現を学習するように促します。次に、ほとんどのインスタンスがネットワークによって簡単に識別され、対照的な損失への寄与が少ないことがわかります。この問題を克服するために、特徴空間でラベル付きサンプルとラベルなしサンプルを混合してハードネガを生成することを提案します。これらの2つの要素がクラスタリングのパフォーマンスに大きく貢献し、モデルが最先端の方法を大幅に上回っていることを実験的に示しています(たとえば、クラスタリングの精度はCIFAR-100で+ 13%、ImageNetで+ 8%)。
In this paper, we address Novel Class Discovery (NCD), the task of unveiling new classes in a set of unlabeled samples given a labeled dataset with known classes. We exploit the peculiarities of NCD to build a new framework, named Neighborhood Contrastive Learning (NCL), to learn discriminative representations that are important to clustering performance. Our contribution is twofold. First, we find that a feature extractor trained on the labeled set generates representations in which a generic query sample and its neighbors are likely to share the same class. We exploit this observation to retrieve and aggregate pseudo-positive pairs with contrastive learning, thus encouraging the model to learn more discriminative representations. Second, we notice that most of the instances are easily discriminated by the network, contributing less to the contrastive loss. To overcome this issue, we propose to generate hard negatives by mixing labeled and unlabeled samples in the feature space. We experimentally demonstrate that these two ingredients significantly contribute to clustering performance and lead our model to outperform state-of-the-art methods by a large margin (e.g., clustering accuracy +13% on CIFAR-100 and +8% on ImageNet).