arXiv reaDer
新規クラス発見のための近隣対照学習
Neighborhood Contrastive Learning for Novel Class Discovery
このホワイトペーパーでは、既知のクラスを持つラベル付きデータセットを指定して、ラベルなしサンプルのセットで新しいクラスを公開するタスクである、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).
updated: Sun Jun 20 2021 17:34:55 GMT+0000 (UTC)
published: Sun Jun 20 2021 17:34:55 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト