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

ZeroDL: Zero-shot Distribution Learning for Text Clustering via Large Language Models

About

The advancements in large language models (LLMs) have brought significant progress in NLP tasks. However, if a task cannot be fully described in prompts, the models could fail to carry out the task. In this paper, we propose a simple yet effective method to contextualize a task toward a LLM. The method utilizes (1) open-ended zero-shot inference from the entire dataset, (2) aggregate the inference results, and (3) finally incorporate the aggregated meta-information for the actual task. We show the effectiveness in text clustering tasks, empowering LLMs to perform text-to-text-based clustering and leading to improvements on several datasets. Furthermore, we explore the generated class labels for clustering, showing how the LLM understands the task through data.

Hwiyeol Jo, Hyunwoo Lee, Kang Min Yoo, Taiwoo Park• 2024

Related benchmarks

TaskDatasetResultRank
Text ClusteringDBp F
Accuracy66.7
39
Short Text ClusteringAGNews
ACC84.3
38
ClusteringIMDB
Accuracy94.3
34
Text ClusteringSST-5
Accuracy52.6
25
Text ClusteringSST-2
Accuracy88.3
25
Text ClusteringYRev
Accuracy53.8
25
Text ClusteringDBp B
Accuracy78.3
25
Text ClusteringYah B
Accuracy74.4
21
Text ClusteringYah(F)
Accuracy52.7
21
Text ClusteringAggregate
Macro Score70.3
21
Showing 10 of 19 rows

Other info

Follow for update