Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Image Clustering Conditioned on Text Criteria

About

Classical clustering methods do not provide users with direct control of the clustering results, and the clustering results may not be consistent with the relevant criterion that a user has in mind. In this work, we present a new methodology for performing image clustering based on user-specified text criteria by leveraging modern vision-language models and large language models. We call our method Image Clustering Conditioned on Text Criteria (IC|TC), and it represents a different paradigm of image clustering. IC|TC requires a minimal and practical degree of human intervention and grants the user significant control over the clustering results in return. Our experiments show that IC|TC can effectively cluster images with various criteria, such as human action, physical location, or the person's mood, while significantly outperforming baselines.

Sehyun Kwon, Jaeseung Park, Minkyu Kim, Jaewoong Cho, Ernest K. Ryu, Kangwook Lee• 2023

Related benchmarks

TaskDatasetResultRank
ClusteringCIFAR-10 (test)
ARI0.759
224
ClusteringSTL-10 (test)
Accuracy97.4
177
Criteria-conditioned ClusteringCOCO 4c OpenSMC
CAcc62.8
30
Criteria-conditioned ClusteringFood-4c OpenSMC
Clustering Accuracy (Criteria)70.5
30
Criteria-conditioned ClusteringClevr-4c OpenSMC
Clustering Accuracy (CAcc)70
30
Criteria-conditioned ClusteringAction-3c OpenSMC
CAcc86.4
24
Criteria-conditioned ClusteringCard-2c OpenSMC
CAcc94.6
18
Criteria-conditioned ClusteringFruit-2c OpenSMC
CAcc69.3
18
ClusteringABO-LC (test)
NMI35.3
10
Dataset GroupingCOCO 4c--
8
Showing 10 of 16 rows

Other info

Follow for update