FASTopic: Pretrained Transformer is a Fast, Adaptive, Stable, and Transferable Topic Model
About
Topic models have been evolving rapidly over the years, from conventional to recent neural models. However, existing topic models generally struggle with either effectiveness, efficiency, or stability, highly impeding their practical applications. In this paper, we propose FASTopic, a fast, adaptive, stable, and transferable topic model. FASTopic follows a new paradigm: Dual Semantic-relation Reconstruction (DSR). Instead of previous conventional, VAE-based, or clustering-based methods, DSR directly models the semantic relations among document embeddings from a pretrained Transformer and learnable topic and word embeddings. By reconstructing through these semantic relations, DSR discovers latent topics. This brings about a neat and efficient topic modeling framework. We further propose a novel Embedding Transport Plan (ETP) method. Rather than early straightforward approaches, ETP explicitly regularizes the semantic relations as optimal transport plans. This addresses the relation bias issue and thus leads to effective topic modeling. Extensive experiments on benchmark datasets demonstrate that our FASTopic shows superior effectiveness, efficiency, adaptivity, stability, and transferability, compared to state-of-the-art baselines across various scenarios.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Text Classification | Newsgroup Science (5-fold cross-validation) | Accuracy0.86 | 36 | |
| Text Classification | Yelp (5-fold cross-validation) | Accuracy71.7 | 36 | |
| Text Classification | Drug Review Norgestimate (5-fold cross-validation) | Accuracy63.6 | 36 | |
| Text Classification | Newsgroup Religion (5-fold cross-validation) | Accuracy53 | 36 | |
| Text Classification | Drug Review Norethindrone (5-fold cross-validation) | Accuracy58.4 | 36 | |
| Text Classification | SMS Spam Collection (5-fold cross-validation) | Accuracy87 | 36 | |
| Document Clustering | Yelp | Purity69.9 | 18 | |
| Topic Modeling | Yelp | Cv0.454 | 18 | |
| Document Clustering | Newsgroup Science | Purity53.9 | 18 | |
| Document Clustering | Drug Review Norgestimate | Purity58.2 | 18 |