arXiv reaDer
グラフ対照クラスタリング
Graph Contrastive Clustering
最近、表現とクラスタリングの割り当てを同時に学習するためのいくつかの対照的な学習方法が提案され、大幅な改善が達成されました。ただし、これらの方法では、カテゴリ情報とクラスタリングの目的が考慮されていないため、学習した表現はクラスタリングに最適ではなく、パフォーマンスが制限される可能性があります。この問題に向けて、最初に新しいグラフ対照学習フレームワークを提案し、それをクラスタリングタスクに適用して、グラフ対照クラスタリング〜(GCC)メソッドを考案します。画像とその拡張のみが同様の表現とクラスタリング割り当てを共有することを前提とする基本的な対照クラスタリングとは異なり、1つのクラスター内のサンプルとその拡張がすべて類似していることを前提として、インスタンスレベルの整合性をクラスターレベルの整合性に引き上げます。具体的には、一方では、グラフラプラシアンベースの対照損失が、より識別的でクラスタリングに適した機能を学習するために提案されています。一方、よりコンパクトなクラスタリング割り当てを学習するために、新しいグラフベースの対照学習戦略が提案されています。どちらも潜在的なカテゴリ情報を組み込んで、クラスター内の分散を減らし、クラスター間の分散を増やします。 6つの一般的に使用されるデータセットでの実験は、最先端の方法に対する提案されたアプローチの優位性を示しています。
Recently, some contrastive learning methods have been proposed to simultaneously learn representations and clustering assignments, achieving significant improvements. However, these methods do not take the category information and clustering objective into consideration, thus the learned representations are not optimal for clustering and the performance might be limited. Towards this issue, we first propose a novel graph contrastive learning framework, which is then applied to the clustering task and we come up with the Graph Constrastive Clustering~(GCC) method. Different from basic contrastive clustering that only assumes an image and its augmentation should share similar representation and clustering assignments, we lift the instance-level consistency to the cluster-level consistency with the assumption that samples in one cluster and their augmentations should all be similar. Specifically, on the one hand, the graph Laplacian based contrastive loss is proposed to learn more discriminative and clustering-friendly features. On the other hand, a novel graph-based contrastive learning strategy is proposed to learn more compact clustering assignments. Both of them incorporate the latent category information to reduce the intra-cluster variance while increasing the inter-cluster variance. Experiments on six commonly used datasets demonstrate the superiority of our proposed approach over the state-of-the-art methods.
updated: Sat Apr 03 2021 15:32:49 GMT+0000 (UTC)
published: Sat Apr 03 2021 15:32:49 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アソシエイト