Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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.

Xiaobao Wu, Thong Nguyen, Delvin Ce Zhang, William Yang Wang, Anh Tuan Luu• 2024

Related benchmarks

TaskDatasetResultRank
Text ClassificationNewsgroup Science (5-fold cross-validation)
Accuracy0.86
36
Text ClassificationYelp (5-fold cross-validation)
Accuracy71.7
36
Text ClassificationDrug Review Norgestimate (5-fold cross-validation)
Accuracy63.6
36
Text ClassificationNewsgroup Religion (5-fold cross-validation)
Accuracy53
36
Text ClassificationDrug Review Norethindrone (5-fold cross-validation)
Accuracy58.4
36
Text ClassificationSMS Spam Collection (5-fold cross-validation)
Accuracy87
36
Document ClusteringYelp
Purity69.9
18
Topic ModelingYelp
Cv0.454
18
Document ClusteringNewsgroup Science
Purity53.9
18
Document ClusteringDrug Review Norgestimate
Purity58.2
18
Showing 10 of 23 rows

Other info

Follow for update