arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2406.13342cs.CLcs.AI
收藏

ZeroDL:基于大语言模型的文本聚类零样本分布学习

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

NAVER Cloud(NAVER云);NAVER;NAVER Search US(NAVER美国搜索)

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

AI总结:

针对提示难以完整描述任务导致 LLM 失效的问题,本文提出 ZeroDL,通过全数据集零样本推理、结果聚合并注入元信息,提升 LLM 的文本聚类效果并解释其任务理解。

摘要:

大语言模型(LLM)的进展为 NLP 任务带来了显著进步。然而,如果某项任务无法在提示中被完整描述,模型可能无法执行该任务。本文提出一种简单而有效的方法,用于使任务情境化以适配 LLM。该方法包括:(1)基于整个数据集进行开放式零样本推理;(2)聚合推理结果;(3)最终将聚合得到的元信息纳入实际任务。我们在文本聚类任务中证明了其有效性,使 LLM 能够执行基于文本到文本的聚类,并在多个数据集上取得改进。此外,我们还探究了为聚类生成的类别标签,展示 LLM 如何通过数据理解任务。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑