跳到正文
arXiv:cs.CL· Thiago C\'esar Castilho Almeida, Daniel Carlos Guimar\~aes Pedronette·· 4 小时前AI 评分34

MARETopic:基于排序原型的流形感知主题建模方法

A Manifold-Aware Topic Modeling Approach via Rank-Based Prototypes

AI 导读

MARETopic 是一个免训练的主题发现框架,将主题建模转化为基于排序的原型选择:把嵌入投影到低维流形后构建排序列表,用贪心算法选出恰好 K 篇真实语料文档作为示例。其 Corr 变体在两个类别最多的基准上取得领先的 Purity 和 NMI,Diff 变体无需额外度量且运行速度快 1.7 至 1.9 倍,并在三个数据集中的两个上取得最佳主题连贯性。该工作已被 EMNLP 2026 主会接收。

正文

View PDF HTML (experimental)

Abstract:Recent topic models leverage pretrained embeddings, but neural architectures produce latent representations without grounding in specific texts, and clustering-based pipelines assign representative documents only post hoc, relying on absolute distances distorted by hubness and anisotropy in high-dimensional spaces. We introduce MARETopic, a training-free framework that casts topic discovery as rank-based prototype selection. After projecting embeddings onto a low-dimensional manifold, MARETopic builds ranked lists encoding ordinal neighborhood structure. A greedy algorithm selects exactly K exemplar documents, real corpus texts, whose neighborhoods cover the corpus. Two variants share this criterion. MARETopic$_\text{Corr}$ scores candidates with a query performance predictor and a rank correlation measure, leading Purity and NMI on the two benchmarks with the most categories, ahead of both neural and clustering-based topic models. MARETopic$_\text{Diff}$ scores them with a rank-based diffusion matrix, needs neither measure, and runs 1.7 to 1.9 times faster. Without a single gradient update, MARETopic leads topic coherence on two of three datasets. A novel inter-topic Maximal Marginal Relevance step raises vocabulary diversity at little cost in coherence. Our code is available at this https URL.
Comments: Accepted at the Main Conference of 2026 Conference on Empirical Methods in Natural Language Processing (EMNLP 2026)
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Computation and Language (cs.CL); Information Retrieval (cs.IR)
Cite as: arXiv:2609.29630 [cs.LG]
  (or arXiv:2609.29630v1 [cs.LG] for this version)
  https://doi.org/10.48550/arXiv.2609.29630

arXiv-issued DOI via DataCite

Submission history

From: Thiago César Castilho Almeida [view email]
[v1] Sat, 29 Aug 2026 02:17:08 UTC (259 KB)

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