不微调模型也能让主题模型在专业领域表现更好
Domain-Agnostic Neural Topic Modeling with Contextual Token-Level Semantic Graph Representation

- 用可学习的图结构捕捉文档级语境,弥补预训练模型缺陷
- 在三个领域上主题连贯性超越基线,且无需微调编码器
- 适合需要快速适配新领域的研究者使用
基于预训练语言模型(PLM)的神经主题模型虽在通用领域表现优异,但在专业语料上主题可解释性下降,主因是嵌入空间中未见术语会坍缩为不可区分区域。现有方法如领域微调、词级图增强或参数高效微调均受限于编码器容量上限。本文提出核心洞察:在冻结的PLM嵌入上构建可学习的词元级语义图,能捕获编码器缺失的语料特定语义结构,因其保留了文档局部上下文,且与主题目标联合优化可直接重塑目标域的嵌入几何。我们构建DARTopic框架,通过冻结的PLM生成词元级语义图,并联合训练图神经网络编码器与主题推断。在涵盖通用、生物医学和法律领域的三个基准上,DARTopic在主题连贯性和文档聚类任务中持续优于强基线,且无需编码器微调,对PLM选择具有鲁棒性,运行效率优于依赖微调的方法。
原文摘要 · Abstract (English)
Recent advances in neural topic models with pre-trained language models (PLMs) have achieved strong performance by leveraging general-domain pre-training, yet their topic interpretability often degrades on specialized corpora. This limitation primarily stems from the geometry of the embedding space, where domain-specific terms unseen during pre-training collapse into an indistinguishable region, and neither domain-specific re-training, word-level graph enrichment, nor parameter-efficient fine-tuning can restructure this space without inheriting the capacity ceiling of the underlying encoder. Our key insight is that a learnable graph layer operating on token-level PLM embeddings can acquire corpus-specific semantic structure that the frozen encoder lacks, because token-level graphs preserve document-local context that word-level representations discard and joint optimization with the topic objective reshapes embedding geometry directly from target-domain evidence. We instantiate this insight as DARTopic, a domain-agnostic framework that constructs token-level semantic graphs from frozen PLM embeddings and jointly trains a GNN encoder with topic inference. Across three benchmarks spanning general, biomedical, and legal domains, DARTopic consistently outperforms strong baselines in topic coherence and document clus- tering without any encoder fine-tuning, while demonstrating robustness to PLM choice and favorable runtime efficiency over fine-tuning based alternatives.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。