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

Universal Guideline-Driven Image Clustering via a Hybrid LLM Agent

About

Unifying image clustering across different clustering scenarios remains challenging due to fundamental gaps among tasks. We introduce a Guideline-Driven Image Clustering Agent, the first universal framework that bridges these gaps through textual guidelines. To incorporate complex guidelines without task-specific training, we propose Generative Concept Proxy Modeling, which generates guideline-aware embeddings via concept proxy extraction. For scenarios requiring automatic cluster discovery, we introduce LLM Traversal based on Minimum Spanning Tree that selectively applies LLM reasoning for complex semantic judgments. Our method generalizes across diverse clustering scenarios spanning from general to fine-grained categorization, from global to local criteria, and from balanced to long-tail distributions. Our framework consistently outperforms specialized methods across diverse clustering tasks.

Wenliang Zhong, Rob Barton, Lucas Goncalves, Kushal Kumar, Feng Jiang, Hehuan Ma, Yuzhi Guo, Vidit Bansal, Karim Bouyarmane, Junzhou Huang• 2026

Related benchmarks

TaskDatasetResultRank
ClusteringCIFAR-10 (test)
ARI0.845
224
ClusteringSTL-10 (test)
Accuracy98.8
177
ClusteringImageNet-10 (test)
NMI0.967
101
Fine-grained ClusteringCUB-Birds
NMI89.9
22
Fine-grained ClusteringStanford Dogs
Normalized Mutual Information (NMI)85.9
22
Fine-grained ClusteringStanford Cars
NMI90
22
Fine-grained ClusteringOxford Flowers
NMI94.9
19
Multiple ClusteringCard
NMI (Number)91.1
18
Multiple ClusteringFruit
Color NMI99.8
17
Multiple ClusteringCIFAR10-MC
Type NMI55.2
17
Showing 10 of 11 rows

Other info

Follow for update