arXiv reaDer
SPICE: Semantic Pseudo-labeling for Image Clustering
サンプル間の類似性とクラスター間の不一致は、画像クラスタリングの2つの重要な側面です。ただし、現在のディープクラスタリング手法では、特徴の類似性または意味の不一致の推定が不正確になります。この論文では、セマンティック疑似ラベリングベースのImage ClustEring(SPICE)フレームワークを紹介します。これは、クラスタリングネットワークを、インスタンスレベルの類似性を測定するための機能モデルと、クラスターレベルの不一致を識別するためのクラスタリングヘッドに分割します。セマンティクスを意識した2つの疑似ラベリングアルゴリズム、プロトタイプ疑似ラベリング、および信頼性の高い疑似ラベリングを設計します。これにより、クラスタリングに対する正確で信頼性の高い自己監視が可能になります。グラウンドトゥルースラベルを使用せずに、3つの段階でクラスタリングネットワークを最適化します。1)インスタンスの類似性を測定するための対照的な学習を通じて特徴モデルをトレーニングします。2)クラスターセマンティクスを識別するためにプロトタイプの疑似ラベル付けアルゴリズムを使用してクラスタリングヘッドをトレーニングします。 3)クラスタリングのパフォーマンスを向上させるために、信頼性の高い疑似ラベリングアルゴリズムを使用して、フィーチャモデルとクラスタリングヘッドを共同でトレーニングします。広範な実験結果は、SPICEが既存の方法に比べて大幅な改善(約10%)を達成し、3つの一般的なメトリックに関して6つの画像ベンチマークデータセットで新しい最先端のクラスタリング結果を確立することを示しています。重要なことに、SPICEは、教師なし分類と完全教師あり分類の間のギャップを大幅に減らします。たとえば、CIFAR-10では2%(91.8%対93.8%)の精度の違いしかありません。私たちのコードはで公開されています。
The similarity among samples and the discrepancy between clusters are two crucial aspects of image clustering. However, current deep clustering methods suffer from the inaccurate estimation of either feature similarity or semantic discrepancy. In this paper, we present a Semantic Pseudo-labeling-based Image ClustEring (SPICE) framework, which divides the clustering network into a feature model for measuring the instance-level similarity and a clustering head for identifying the cluster-level discrepancy. We design two semantics-aware pseudo-labeling algorithms, prototype pseudo-labeling, and reliable pseudo-labeling, which enable accurate and reliable self-supervision over clustering. Without using any ground-truth label, we optimize the clustering network in three stages: 1) train the feature model through contrastive learning to measure the instance similarity, 2) train the clustering head with the prototype pseudo-labeling algorithm to identify cluster semantics, and 3) jointly train the feature model and clustering head with the reliable pseudo-labeling algorithm to improve the clustering performance. Extensive experimental results demonstrate that SPICE achieves significant improvements (~10%) over existing methods and establishes the new state-of-the-art clustering results on six image benchmark datasets in terms of three popular metrics. Importantly, SPICE significantly reduces the gap between unsupervised and fully-supervised classification; e.g., there is only a 2% (91.8% vs 93.8%) accuracy difference on CIFAR-10. Our code has been made publically available at
updated: Mon Oct 25 2021 14:11:41 GMT+0000 (UTC)
published: Wed Mar 17 2021 00:52:27 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト