arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.03071cs.CLcs.IR

利用LLM驱动的上下文扩展与前缀微调VAE增强短文本主题建模

Enhancing Short-Text Topic Modeling with LLM-Driven Context Expansion and Prefix-Tuned VAEs

University of Illinois at Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校)

Pritom Saha Akash, Kevin Chen-Chuan Chang

AI总结:

本文针对短文本词语共现不足导致主题建模不连贯的问题,提出先用LLM扩展文本,再通过前缀微调小语言模型结合变分自编码器进行建模,在稀疏真实数据集上显著超越现有最优方法。

摘要:

主题建模是一种在文档集合中发现隐藏主题的强大技术。然而,传统主题模型的有效性通常依赖于充分的词语共现,而短文本中缺乏这种共现。因此,现有方法无论是概率模型还是神经模型,在从此类数据中提取有意义的模式时往往表现不佳,导致主题不连贯。为应对这一挑战,我们提出一种新方法,利用大型语言模型(LLMs)将短文本扩展为更详细的序列,然后再进行主题建模。为进一步提高效率并解决LLM生成文本带来的语义不一致问题,我们提出使用前缀微调(prefix tuning)训练一个较小的语言模型,并与变分自编码器(variational autoencoder)相结合用于短文本主题建模。在具有极端数据稀疏性的真实世界数据集上进行的大量实验表明,我们的方法显著提升了短文本主题建模性能,超越了当前最先进的主题模型。

英文摘要:

Topic modeling is a powerful technique for uncovering hidden themes within a collection of documents. However, the effectiveness of traditional topic models often relies on sufficient word co-occurrence, which is lacking in short texts. Therefore, existing approaches, whether probabilistic or neural, frequently struggle to extract meaningful patterns from such data, resulting in incoherent topics. To address this challenge, we propose a novel approach that leverages large language models (LLMs) to extend short texts into more detailed sequences before applying topic modeling. To further improve the efficiency and solve the problem of semantic inconsistency from LLM-generated texts, we propose to use prefix tuning to train a smaller language model coupled with a variational autoencoder for short-text topic modeling. Our method significantly improves short-text topic modeling performance, as demonstrated by extensive experiments on real-world datasets with extreme data sparsity, outperforming current state-of-the-art topic models.

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

↑