この論文では、画像のディープクラスタリングと教師なし表現学習に焦点を当てます。ディープクラスタリングと教師なし表現学習の最近の進歩は、入力画像のさまざまなビュー(データ拡張技術によって生成されたもの)を表現空間でより近くに配置する必要がある(例の一貫性)、および/または同様の画像に同様のクラスター割り当てがあるという考えに基づいています(人口の一貫性)。一貫性、コンセンサス一貫性の追加の概念を定義します。これにより、表現が学習され、表現空間の変化、異なるクラスタリングアルゴリズム、またはクラスタリングアルゴリズムの異なる初期化に対して同様のパーティションが誘導されます。表現空間でバリエーションを実行することによってクラスタリング損失を定義し、3つの一貫性(コンセンサス、エグザンプラ、および母集団)すべてをエンドツーエンドの学習フレームワークにシームレスに統合します。提案されたアルゴリズムである教師なし表現学習(ConCURL)を使用したコンセンサスクラスタリングは、5つの画像データセットのうち4つで最先端の方法よりもクラスタリングのパフォーマンスを向上させます。さらに、クラスタリングの評価手順を拡張して、分散シフトの場合のクラスタリングパフォーマンスなど、実際のクラスタリングタスクの課題を反映します。また、アルゴリズムをより深く理解するために、詳細なアブレーション調査を実行します。
In this paper, we focus on deep clustering and unsupervised representation learning for images. Recent advances in deep clustering and unsupervised representation learning are based on the idea that different views of an input image (generated through data augmentation techniques) must be closer in the representation space (exemplar consistency), and/or similar images have a similar cluster assignment (population consistency). We define an additional notion of consistency, consensus consistency, which ensures that representations are learnt to induce similar partitions for variations in the representation space, different clustering algorithms or different initializations of a clustering algorithm. We define a clustering loss by performing variations in the representation space and seamlessly integrate all three consistencies (consensus, exemplar and population) into an end-to-end learning framework. The proposed algorithm, Consensus Clustering using Unsupervised Representation Learning (ConCURL) improves the clustering performance over state-of-the art methods on four out of five image datasets. Further, we extend the evaluation procedure for clustering to reflect the challenges in real world clustering tasks, such as clustering performance in the case of distribution shift. We also perform a detailed ablation study for a deeper understanding of the algorithm.