論文

査読有り
2023年7月12日

Gaussian hierarchical latent Dirichlet allocation: Bringing polysemy back

PLOS ONE
  • Takahiro Yoshida
  • ,
  • Ryohei Hisano
  • ,
  • Takaaki Ohnishi

18
7
開始ページ
e0288274
終了ページ
e0288274
記述言語
英語
掲載種別
研究論文(学術雑誌)
DOI
10.1371/journal.pone.0288274
出版者・発行元
Public Library of Science (PLoS)

Topic models are widely used to discover the latent representation of a set of documents. The two canonical models are latent Dirichlet allocation, and Gaussian latent Dirichlet allocation, where the former uses multinomial distributions over words, and the latter uses multivariate Gaussian distributions over pre-trained word embedding vectors as the latent topic representations, respectively. Compared with latent Dirichlet allocation, Gaussian latent Dirichlet allocation is limited in the sense that it does not capture the polysemy of a word such as "bank." In this paper, we show that Gaussian latent Dirichlet allocation could recover the ability to capture polysemy by introducing a hierarchical structure in the set of topics that the model can use to represent a given document. Our Gaussian hierarchical latent Dirichlet allocation significantly improves polysemy detection compared with Gaussian-based models and provides more parsimonious topic representations compared with hierarchical latent Dirichlet allocation. Our extensive quantitative experiments show that our model also achieves better topic coherence and held-out document predictive accuracy over a wide range of corpus and word embedding vectors which significantly improves the capture of polysemy compared with GLDA and CGTM. Our model learns the underlying topic distribution and hierarchical structure among topics simultaneously, which can be further used to understand the correlation among topics. Moreover, the added flexibility of our model does not necessarily increase the time complexity compared with GLDA and CGTM, which makes our model a good competitor to GLDA.

リンク情報
DOI
https://doi.org/10.1371/journal.pone.0288274
URL
https://dx.plos.org/10.1371/journal.pone.0288274
ID情報
  • DOI : 10.1371/journal.pone.0288274
  • eISSN : 1932-6203

エクスポート
BibTeX RIS