Neural topic model training with the REBAR gradient estimator

A Kumar, N Esmaili, M Piccardi - ACM Transactions on Asian and Low …, 2022 - dl.acm.org
ACM Transactions on Asian and Low-Resource Language Information Processing, 2022dl.acm.org
Topic modelling is an important approach of unsupervised machine learning that allows
automatically extracting the main “topics” from large collections of documents. In addition,
topic modelling is able to identify the topic proportions of each individual document, which
can be helpful for organizing the collections. Many topic modelling algorithms have been
proposed to date, including several that leverage advanced techniques such as variational
inference and deep autoencoders. However, to date topic modelling has made limited use of …
Topic modelling is an important approach of unsupervised machine learning that allows automatically extracting the main “topics” from large collections of documents. In addition, topic modelling is able to identify the topic proportions of each individual document, which can be helpful for organizing the collections. Many topic modelling algorithms have been proposed to date, including several that leverage advanced techniques such as variational inference and deep autoencoders. However, to date topic modelling has made limited use of reinforcement learning, a framework that has obtained vast success in many other unsupervised learning tasks. For this reason, in this article we propose training a neural topic model using a reinforcement learning objective and minimizing the objective with the recently-proposed REBAR gradient estimator. Experiments performed over two probing datasets have shown that the proposed model has achieved improvements over all the compared models in terms of both model perplexity and topic coherence, and produced topics that appear qualitatively informative and consistent.
ACM Digital Library
以上显示的是最相近的搜索结果。 查看全部搜索结果