跳到正文
arXiv:cs.LG· Matthew Raffel, Lizhong Chen, Sai Munikoti, Sameera Horawalavithana·· 16 小时前

TopicMerge:分治式主题建模新方法,LLM 调用量降低超 92%

TopicMerge: A Divide and Conquer Approach to Topic Modeling

AI 导读

TopicMerge 是一种多阶段主题建模流水线,先对文档语料过分割成细粒度微簇并赋予初始语义标签,再由 LLM 依据局部邻域上下文迭代仲裁候选簇的合并与重标注,从而逐层构建更连贯的主题。在 Bills 和 Wiki 数据集上,该方法取得 SOTA 聚类性能,同时相比文档级方法减少超过 92% 的 LLM 调用。

正文

View PDF HTML (experimental)

Abstract:Recent advancements in topic modeling have split into two main paths: efficient embedding-based methods that offer scalability but are often outperformed on standard clustering metrics, and Large Language Model (LLM) approaches that achieve state-of-the-art clustering performance but at the cost of prohibitive computational expense and poor scalability. This paper introduces TopicMerge, a novel multi-stage pipeline that bridges these approaches to achieve both state-of-the-art clustering performance and computational efficiency. TopicMerge first employs an efficient clustering pipeline to deliberately over-segment a document corpus into a large set of fine-grained, semantically pure micro-clusters, which are then given initial semantic labels. It then introduces a novel, iterative merging algorithm where an LLM, guided by local neighborhood context, arbitrates on consolidating candidate pairs. These approved merges are then selectively executed and semantically relabeled by the LLM, hierarchically building broader, more coherent topics. Extensive experiments on the Bills and Wiki datasets show that TopicMerge achieves state-of-the-art clustering performance while reducing required LLM calls by over 92% compared to document-level methods, offering a practical and scalable solution.
Comments: Findings of EMNLP 2026
Subjects: Computation and Language (cs.CL); Machine Learning (cs.LG)
Cite as: arXiv:2610.19323 [cs.CL]
  (or arXiv:2610.19323v1 [cs.CL] for this version)
  https://doi.org/10.48550/arXiv.2610.19323

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Lizhong Chen [view email]
[v1] Wed, 14 Oct 2026 00:01:35 UTC (5,227 KB)

来源:arXiv:cs.LG · arxiv.org