English

Top2Vec: Distributed Representations of Topics

Computation and Language 2020-08-24 v1 Machine Learning Machine Learning

Abstract

Topic modeling is used for discovering latent semantic structure, usually referred to as topics, in a large collection of documents. The most widely used methods are Latent Dirichlet Allocation and Probabilistic Latent Semantic Analysis. Despite their popularity they have several weaknesses. In order to achieve optimal results they often require the number of topics to be known, custom stop-word lists, stemming, and lemmatization. Additionally these methods rely on bag-of-words representation of documents which ignore the ordering and semantics of words. Distributed representations of documents and words have gained popularity due to their ability to capture semantics of words and documents. We present top2vec\texttt{top2vec}, which leverages joint document and word semantic embedding to find topic vectors\textit{topic vectors}. This model does not require stop-word lists, stemming or lemmatization, and it automatically finds the number of topics. The resulting topic vectors are jointly embedded with the document and word vectors with distance between them representing semantic similarity. Our experiments demonstrate that top2vec\texttt{top2vec} finds topics which are significantly more informative and representative of the corpus trained on than probabilistic generative models.

Keywords

Cite

@article{arxiv.2008.09470,
  title  = {Top2Vec: Distributed Representations of Topics},
  author = {Dimo Angelov},
  journal= {arXiv preprint arXiv:2008.09470},
  year   = {2020}
}

Comments

Implementation available at https://github.com/ddangelov/Top2Vec

R2 v1 2026-06-23T18:01:06.253Z