HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Neural Topic Modeling with Bidirectional Adversarial Training

Rui Wang Xuemeng Hu Deyu Zhou Yulan He Yuxuan Xiong Chenchen Ye Haiyang Xu

Neural Topic Modeling with Bidirectional Adversarial Training

Abstract

Recent years have witnessed a surge of interests of using neural topic models for automatic topic extraction from text, since they avoid the complicated mathematical derivations for model inference as in traditional topic models such as Latent Dirichlet Allocation (LDA). However, these models either typically assume improper prior (e.g. Gaussian or Logistic Normal) over latent topic space or could not infer topic distribution for a given document. To address these limitations, we propose a neural topic modeling approach, called Bidirectional Adversarial Topic (BAT) model, which represents the first attempt of applying bidirectional adversarial training for neural topic modeling. The proposed BAT builds a two-way projection between the document-topic distribution and the document-word distribution. It uses a generator to capture the semantic patterns from texts and an encoder for topic inference. Furthermore, to incorporate word relatedness information, the Bidirectional Adversarial Topic model with Gaussian (Gaussian-BAT) is extended from BAT. To verify the effectiveness of BAT and Gaussian-BAT, three benchmark corpora are used in our experiments. The experimental results show that BAT and Gaussian-BAT obtain more coherent topics, outperforming several competitive baselines. Moreover, when performing text clustering based on the extracted topics, our models outperform all the baselines, with more significant improvements achieved by Gaussian-BAT where an increase of near 6\% is observed in accuracy.

Code Repositories

zll17/Neural_Topic_Models
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
text-clustering-on-20-newsgroupsG-BAT
Accuracy: 41.25
text-clustering-on-20-newsgroupsBAT
Accuracy: 35.66

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Neural Topic Modeling with Bidirectional Adversarial Training | Papers | HyperAI