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

ClusterFusion: Hybrid Clustering with Embedding Guidance and LLM Adaptation

About

Text clustering is a fundamental task in natural language processing, yet traditional clustering algorithms with pre-trained embeddings often struggle in domain-specific contexts without costly fine-tuning. Large language models (LLMs) provide strong contextual reasoning, yet prior work mainly uses them as auxiliary modules to refine embeddings or adjust cluster boundaries. We propose ClusterFusion, a hybrid framework that instead treats the LLM as the clustering core, guided by lightweight embedding methods. The framework proceeds in three stages: embedding-guided subset partition, LLM-driven topic summarization, and LLM-based topic assignment. This design enables direct incorporation of domain knowledge and user preferences, fully leveraging the contextual adaptability of LLMs. Experiments on three public benchmarks and two new domain-specific datasets demonstrate that ClusterFusion not only achieves state-of-the-art performance on standard tasks but also delivers substantial gains in specialized domains. To support future work, we release our newly constructed dataset and results on all benchmarks.

Yiming Xu, Yuan Yuan, Vijay Viswanathan, Graham Neubig• 2025

Related benchmarks

TaskDatasetResultRank
Short Text ClusteringTweet
Accuracy92.1
28
ClusteringBank77
NMI92.6
19
ClusteringCLINC
Accuracy91
15
ClusteringOpenAI Codex (test)
Accuracy80
9
ClusteringAdobe Lightroom (test)
Accuracy92.3
9
Showing 5 of 5 rows

Other info

Follow for update