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 主会接收。
正文
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