中文

神经嵌入分配:主题模型的分布式表示

计算与语言 2019-09-12 v1 信息检索 机器学习

摘要

诸如skip-gram之类的词嵌入模型学习词语义关系的向量表示,文档嵌入模型学习文档的类似表示。另一方面,主题模型提供文档主题主题的潜在表示。为同时获得这些表示的益处,我们提出一种统一算法,称为神经嵌入分配(NEA),它通过学习神经嵌入来模仿主题模型,将主题模型解构为词、主题、文档、作者等可解释的向量空间嵌入。我们在LDA、作者主题模型以及最近提出的混合隶属skip gram主题模型上展示了NEA的有效性和通用性,并且相较于若干SOTA模型,使用嵌入取得了更好的性能。此外,我们证明当主题数量较大时,使用NEA平滑主题可相比原始主题模型提高一致性分数。

关键词

引用

@article{arxiv.1909.04702,
  title  = {Neural Embedding Allocation: Distributed Representations of Topic Models},
  author = {Kamrun Naher Keya and Yannis Papanikolaou and James R. Foulds},
  journal= {arXiv preprint arXiv:1909.04702},
  year   = {2019}
}