Share your thoughts, 1 month free Claude Pro on usSee more
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
318
Image ClusteringSTL-10
ACC98.2
282
Image ClusteringImageNet-10
NMI0.985
201
ClusteringCIFAR-10 (test)
Accuracy92.1
190
ClusteringSTL-10 (test)
Accuracy98.3
152
ClusteringCIFAR-100 (test)
ACC55.4
123
Image ClusteringCIFAR-100
ACC56.1
111
ClusteringImagenet Dogs
NMI8.06e+3
85
ClusteringImageNet-10 (test)
ACC99.2
74
Image ClusteringDTD
NMI62.1
49
Showing 10 of 32 rows

Other info

Code

Follow for update