Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Top2Vec: Distributed Representations of Topics

About

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 $\texttt{top2vec}$, which leverages joint document and word semantic embedding to find $\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 $\texttt{top2vec}$ finds topics which are significantly more informative and representative of the corpus trained on than probabilistic generative models.

Dimo Angelov• 2020

Related benchmarks

TaskDatasetResultRank
Topic ModelingJamuna News
CV0.83
29
Topic ModelingNCTBText
CV0.74
29
Topic ModelingBanFakeNews
CV0.8
25
Topic ModelingLIAR labeled holdout (test)
AUPC63.6
7
Topic ModelingIMDB labeled holdout IM (test)
AUPC91.3
7
Topic ModelingHumAID HU (test)
AUPC71.7
7
Codebook Developmentheart failure interview transcripts (test)
Cosine Similarity0.377
5
Topic Quality EvaluationAggregate Topic Modeling Datasets (including StackExchangeClusteringP2P, MedrxivClusteringP2P) (test)
Cin0.21
3
Showing 8 of 8 rows

Other info

Follow for update