Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Image Clustering with External Guidance

About

The core of clustering is incorporating prior knowledge to construct supervision signals. From classic k-means based on data compactness to recent contrastive clustering guided by self-supervision, the evolution of clustering methods intrinsically corresponds to the progression of supervision signals. At present, substantial efforts have been devoted to mining internal supervision signals from data. Nevertheless, the abundant external knowledge such as semantic descriptions, which naturally conduces to clustering, is regrettably overlooked. In this work, we propose leveraging external knowledge as a new supervision signal to guide clustering, even though it seems irrelevant to the given data. To implement and validate our idea, we design an externally guided clustering method (Text-Aided Clustering, TAC), which leverages the textual semantics of WordNet to facilitate image clustering. Specifically, TAC first selects and retrieves WordNet nouns that best distinguish images to enhance the feature discriminability. Then, to improve image clustering performance, TAC collaborates text and image modalities by mutually distilling cross-modal neighborhood information. Experiments demonstrate that TAC achieves state-of-the-art performance on five widely used and three more challenging image clustering benchmarks, including the full ImageNet-1K dataset.

Yunfan Li, Peng Hu, Dezhong Peng, Jiancheng Lv, Jianping Fan, Xi Peng• 2023

Related benchmarks

TaskDatasetResultRank
Image ClusteringCIFAR-10
NMI0.833
243
Image ClusteringSTL-10
ACC98.2
229
Image ClusteringImageNet-10
NMI0.985
166
ClusteringImagenet Dogs
NMI8.06e+3
46
Image ClusteringCIFAR-20
NMI61.1
19
Image ClusteringDTD
NMI62.1
6
Image ClusteringUCF-101
NMI82.3
6
Image ClusteringImageNet-1K
NMI79.9
6
ClusteringAircraft
NMI47.7
4
ClusteringFood
NMI69.4
4
Showing 10 of 16 rows

Other info

Code

Follow for update