Neural Topic Modeling with Bidirectional Adversarial Training

04/26/2020 ∙ by Rui Wang, et al. ∙ Beijing Didi Infinity Technology and Development Co., Ltd. University of Warwick 0

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.



There are no comments yet.


page 1

page 2

page 3

page 4

This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.

1 Introduction

Topic models have been extensively explored in the Natural Language Processing (NLP) community for unsupervised knowledge discovery. Latent Dirichlet Allocation (LDA) 

Blei et al. (2003), the most popular topic model, has been extended Lin and He (2009); Zhou et al. (2014); Cheng et al. (2014) for various extraction tasks. Due to the difficulty of exact inference, most LDA variants require approximate inference methods, such as mean-field methods and collapsed Gibbs sampling. However, these approximate approaches have the drawback that small changes to the modeling assumptions result in a re-derivation of the inference algorithm, which can be mathematically arduous.

Figure 1:

Illustrated probability simplex with Logistic-Normal distribution and Dirichlet distribution.

One possible way in addressing this limitation is through neural topic models which employ black-box inference mechanism with neural networks. Inspired by variational autoencoder (VAE) 

Kingma and Welling (2013), Srivastava and Sutton Srivastava and Sutton (2017) used the Logistic-Normal prior to mimic the simplex in latent topic space and proposed the Neural Variational LDA (NVLDA). Moreover, they replaced the word-level mixture in NVLDA with a weighted product of experts and proposed the ProdLDA Srivastava and Sutton (2017) to further enhance the topic quality.

Although Srivastava and Sutton Srivastava and Sutton (2017) used the Logistic-Normal distribution to approximate the Dirichlet distribution, they are not exactly the same. An illustration of these two distributions is shown in Figure 1 in which the Logistic-Normal distribution does not exhibit multiple peaks at the vertices of the simplex as that in the Dirichlet distribution and as such, it is less capable to capture the multi-modality which is crucial in topic modeling Wallach et al. (2009). To deal with the limitation, Wang et al. Wang et al. (2019) proposed the Adversarial-neural Topic Model (ATM) based on adversarial training, it uses a generator network to capture the semantic patterns lying behind the documents. However, given a document, ATM is not able to infer the document-topic distribution which is useful for downstream applications, such as text clustering. Moreover, ATM take the bag-of-words assumption and do not utilize any word relatedness information captured in word embeddings which have been proved to be crucial for better performance in many NLP tasks Liu et al. (2018); Lei et al. (2018).

To address these limitations, we model topics with Dirichlet prior and propose a novel Bidirectional Adversarial Topic model (BAT) based on bidirectional adversarial training. The proposed BAT employs a generator network to learn the projection function from randomly-sampled document-topic distribution to document-word distribution. Moreover, an encoder network is used to learn the inverse projection, transforming a document-word distribution into a document-topic distribution. Different from traditional models that often resort to analytic approximations, BAT employs a discriminator which aims to discriminate between real distribution pair and fake distribution pair, thereby helps the networks (generator and encoder) to learn the two-way projections better. During the adversarial training phase, the supervision signal provided by the discriminator will guide the generator to construct a more realistic document and thus better capture the semantic patterns in text. Meanwhile, the encoder network is also guided to generate a more reasonable topic distribution conditioned on specific document-word distributions. Finally, to incorporate the word relatedness information captured by word embeddings, we extend the BAT by modeling each topic with a multivariate Gaussian in the generator and propose the Bidirectional Adversarial Topic model with Gaussian (Gaussian-BAT).

The main contributions of the paper are:

  • We propose a novel Bidirectional Adversarial Topic (BAT) model, which is, to our best knowledge, the first attempt of using bidirectional adversarial training in neural topic modeling;

  • We extend BAT to incorporate the word relatedness information into the modeling process and propose the Bidirectional Adversarial Topic model with Gaussian (Gaussian-BAT);

  • Experimental results on three public datasets show that BAT and Gaussian-BAT outperform the state-of-the-art approaches in terms of topic coherence measures. The effectiveness of BAT and Gaussian-BAT is further verified in text clustering.

Figure 2: The framework of the Bidirectional Adversarial Topic (BAT) model.

2 Related work

Our work is related to two lines of research, which are adversarial training and neural topic modeling.

2.1 Adversarial Training

Adversarial training, first employed in Generative Adversarial Network (GAN) Goodfellow et al. (2014), has been extensively studied from both theoretical and practical perspectives.

Theoretically, Arjovsky Arjovsky et al. (2017) and Gulrajani Gulrajani et al. (2017) proposed the Wasserstein GAN which employed the Wasserstein distance between data distribution and generated distribution as the training objective. To address the limitation that most GANs Goodfellow et al. (2014); Radford et al. (2015) could not project data into a latent space, Bidirectional Generative Adversarial Nets (BiGAN)  Donahue et al. (2016) and Adversarially Learned Inference (ALI) Dumoulin et al. (2016) were proposed.

Adversarial training has also been extensively used for text generation. For example, SeqGAN 

Yu et al. (2017) incorporated a policy gradient strategy for text generation. RankGAN Lin et al. (2017) ranked a collection of human-written sentences to capture the language structure for improving the quality of text generation. To avoid mode collapse when dealing with discrete data, MaskGAN Fedus et al. (2018) used an actor-critic conditional GAN to fill in missing text conditioned on the context.

2.2 Neural Topic Modeling

To overcome the challenging exact inference of topic models based on directed graph, a replicated softmax model (RSM), based on the Restricted Boltzmann Machines was proposed in  

Hinton and Salakhutdinov (2009). Inspired by VAE, Miao et al. Miao et al. (2016) used the multivariate Gaussian as the prior distribution of latent space and proposed the Neural Variational Document Model (NVDM) for text modeling. To model topic properly, the Gaussian Softmax Model (GSM) Miao et al. (2017)

which constructs the topic distribution using a Gaussian distribution followed by a softmax transformation was proposed based on the NVDM.

Likewise, to deal with the inappropriate Gaussian prior of NVDM, Srivastava and Sutton Srivastava and Sutton (2017) proposed the NVLDA which approximates the Dirichlet prior using a Logistic-Normal distribution. Recently, the Adversarial-neural Topic Model (ATM) Wang et al. (2019) is proposed based on adversarial training, it models topics with Dirichlet prior which is able to capture the multi-modality compared with logistic-normal prior and obtains better topics. Besides, the Adversarial-neural Event (AEM) Wang et al. (2019) model is also proposed for open event extraction by representing each event as an entity distribution, a location distribution, a keyword distribution and a date distribution.

Despite the extensive exploration of this research field, scarce work has been done to incorporate Dirichlet prior, word embeddings and bidirectional adversarial training into neural topic modeling. In this paper, we propose two novel topic modeling approaches, called BAT and Gaussian-BAT, which are different from existing approaches in the following aspects: (1) Unlike NVDM, GSM, NVLDA and ProdLDA which model latent topic with Gaussian or logistic-normal prior, BAT and Gaussian-BAT explicitly employ Dirichlet prior to model topics; (2) Unlike ATM which could not infer topic distribution of a given document, BAT and Gaussian-BAT uses a encoder to generate the topic distribution corresponding to the document; (3) Unlike neural topic models that only utilize word co-occurrence information, Gaussian-BAT models topic with multivariate Gaussian and incorporates the word relatedness into modeling process.

3 Methodology

Our proposed neural topic models are based on bidirectional adversarial training Donahue et al. (2016) and aim to learn the two-way non-linear projection between two high-dimensional distributions. In this section, we first introduce the Bidirectional Adversarial Topic (BAT) model that only employs the word co-occurrence information. Then, built on BAT, we model topics with multivariate Gaussian in the generator of BAT and propose the Bidirectional Adversarial Topic model with Gaussian (Gaussian-BAT), which naturally incorporates word relatedness information captured in word embeddings into modeling process.

3.1 Bidirectional Adversarial Topic model

As depicted in Figure 2, the proposed BAT consists of three components: (1) The Encoder takes the -dimensional document representation sampled from text corpus as input and transforms it into the corresponding -dimensional topic distribution ; (2) The Generator takes a random topic distribution drawn from a Dirichlet prior as input and generates a -dimensional fake word distribution ; (3) The Discriminator takes the real distribution pair and fake distribution pair as input and discriminates the real distribution pairs from the fake ones. The outputs of the discriminator are used as supervision signals to learn , and during adversarial training. In what follows, we describe each component in more details.

3.1.1 Encoder Network

The encoder learns a mapping function to transform document-word distribution to document-topic distribution. As shown in the top-left panel of Figure 2, it contains a -dimensional document-word distribution layer, an -dimensional representation layer and a -dimensional document-topic distribution layer, where and denote vocabulary size and topic number respectively.

More concretely, for each document in text corpus, takes the document representation as input, where is the representation weighted by TF-IDF, and it is calculated by:

where denotes the number of -th word appeared in document , represents the number of documents in the corpus, and means the number of documents that contain -th word in the corpus. Thus, each document could be represented as a -dimensional multinomial distribution and the -th dimension denotes the semantic consistency between -th word and the document.

With as input, firstly projects it into an -dimensional semantic space through the representation layer as follows:


where and are weight matrix and bias term of the representation layer,

is the state vector normalized by batch normalization

, denotes the parameter of LeakyReLU activation and represents the output of representation layer.

Then, the encoder transforms into a -dimensional topic space based on the equation below:


where is the weight matrix of topic distribution layer, represents the bias term, denotes the corresponding topic distribution of the input and the -th () dimension represents the proportion of -th topic in document .

3.1.2 Generator network

The generator is shown in the bottom-left panel of Figure 2. Contrary to encoder, it provides an inverse projection from document-topic distribution to document-word distribution and contains a -dimensional document-topic layer, an -dimensional representation layer and a -dimensional document-word distribution layer.

As pointed out in Wallach et al. (2009), the choice of Dirichlet prior over topic distribution is important to obtain interpretable topics. Thus, BAT employs the Dirichlet prior parameterized with to mimic the multi-variate simplex over topic distribution . It can be drawn randomly based on the equation below:


where is the -dimensional hyper-parameter of Dirichlet prior, is the topic number that should be set in BAT, , follows the constrain that , represents the proportion of the -th topic in the document, and normalization term is defined as .

To learn the transformation from document-topic distribution to document-word distribution, firstly projects into an -dimensional representation space based on equations:


where is weight matrix of the representation layer, represents bias term, is the state vector normalized by batch normalization, Eq. 6 represents the LeakyReLU activation parameterized with , and is the output of the representation layer.

Then, to project into word distribution

, a subnet contains a linear layer and a softmax layer is used and the transformation follows:


where and are weight matrix and bias of word distribution layer, is the word distribution correspond to . For each , the -th dimension is the probability of the -th word in fake document .

3.1.3 Discriminator network

The discriminator is constituted by three layers (a

-dimensional joint distribution layer, an

-dimensional representation layer and an output layer) as shown in the right panel of Figure 2. It employs real distribution pair and fake distribution pair as input and then outputs to identify the input sources (fake or real). Concretely, a higher value of represents that is more prone to predict the input as real and vice versa.

3.2 BAT with Gaussian (Gaussian-BAT)

In BAT, the generator models topics based on the bag-of-words assumption as in most other neural topic models. To incorporate the word relatedness information captured in word embeddings Mikolov et al. (2013a, b); Pennington et al. (2014); Joulin et al. (2017); Athiwaratkun et al. (2018) into the inference process, we modify the generator of BAT and propose Gaussian-BAT, in which models each topic with a multivariate Gaussian as shown in Figure 3.

Figure 3: The generator of Gaussian-BAT.

Concretely, Gaussian-BAT employs the multivariate Gaussian to model the -th topic. Here, and are trainable parameters, they represent mean and covariance matrix respectively. Following its probability density, for each word , the probability in the -th topic is calculated by:


where means the word embedding of -th word, is the vocabulary size, is the determinant of covariance matrix , is the dimension of word embeddings, is the probability calculated by density, and is the normalized word distribution of -th topic. With randomly sampled topic distribution and the calculated topic-word distributions , the fake word distribution corresponding to can be obtained by:


where is the topic proportion of the -th topic. Then, and are concatenated to form the fake distribution pair as shown in Figure 3. And encoder and discriminator of Gaussian-BAT are same as BAT, shown as Figure 2. In our experiments, the pre-trained 300-dimensional Glove Pennington et al. (2014) embedding is used.

3.3 Objective and Training Procedure

In Figure 2, the real distribution pair and the fake distribution pair can be viewed as random samples drawn from two -dimensional joint distributions and , each of them comprising of a -dimensional Dirichlet distribution and a -dimensional Dirichlet distribution. The training objective of BAT and Gaussian-BAT is to make the generated joint distribution close to the real joint distribution as much as possible. In this way, a two-way projection between document-topic distribution and document-word distribution could be built by the learned encoder and generator.

To measure the distance between and , we use the Wasserstein-distance as the optimization objective, since it was shown to be more effective compared to Jensen-Shannon divergence Arjovsky et al. (2017):


where represents the output signal of the discriminator. A higher value denotes that the discriminator is more prone to consider the input as a real distribution pair and vice versa. In addition, we use weight clipping which was proposed to ensure the Lipschitz continuity Arjovsky et al. (2017) of .

0:  , , , , , ,
0:  The trained encoder and generator .
1:  Initialize , and with , and
2:  while  and have not converged do
3:     for  do
4:        for  do
5:           Sample ,
6:           Sample a random
7:           ,
8:           ,
10:        end for
13:     end for
16:  end while
Algorithm 1 Training procedure for BAT and Gaussian-BAT

The training procedure of BAT and Gaussian-BAT is given in Algorithm. 1. Here, is the clipping parameter, represents the number of discriminator iterations per generator iteration, is the batch size, is the learning rate, and are hyper-parameters of Adam Kingma and Ba (2014), and represents . In our experiments, we set the , , , , and .

3.4 Topic Generation and Cluster Inference

After model training, learned and will build a two-way projection between document-topic distribution and document-word distribution. Thus, and could be used for topic generation and cluster inference.

To generate the word distribution of each topic, we use , a

-dimensional vector, as the one-hot encoding of the

-th topic. For example, in a six topic setting. And the word distribution of the -th topic is obtained by:


Likewise, given the document representation , topic distribution obtained by BAT/Gaussian-BAT could be used for cluster inference based on:


where denotes the inferred cluster of .

4 Experiments

In this section, we first present the experimental setup which includes the datasets used and the baselines, followed by the experimental results.

Figure 4: The comparison of average topic coherence vs. different topic proportion on three datasets.

4.1 Experimental Setup

We evaluate BAT and Gaussian-BAT on three datasets for topic extraction and text clustering, 20Newsgroups111 jason/20Newsgroups/ , Grolier222 and NYTimes333 . Details are summarized below:

20Newsgroups Lang (1995) is a collection of approximately 20,000 newsgroup articles, partitioned evenly across 20 different newsgroups.
Grolier is built from Grolier Multimedia Encycopedia, which covers almost all the fields in the world.
NYTimes is a collection of news articles published between 1987 and 2007, and contains a wide range of topics, such as sports, politics, education, etc.

We use the full datasets of 20Newsgroups and Grolier. For the NYTimes dataset, we randomly select 100,000 articles and remove the low frequency words. The final statistics are shown in Table 1:

Dataset #Doc (Train) #Doc (Test) #Words
20Newsgroups 11,259 7,488 1,995
Grolier 29,762 - 15,276
NYtimes 99,992 - 12,604
Table 1: The statistics of datasets.

We choose the following models as baselines:

LDA Blei et al. (2003) extracts topics based on word co-occurrence patterns from documents. We implement LDA following the parameter setting suggested in Griffiths and Steyvers (2004).
NVDM Miao et al. (2016) is an unsupervised text modeling approach based on VAE. We use the original implementation of the paper444
GSMMiao et al. (2017) is an enhanced topic model based on NVDM, we use the original implementation in our experiments555
 Srivastava and Sutton (2017), also built on VAE but with the logistic-normal prior. We use the implementation provided by the author666 vi for topic models.
ProdLDA Srivastava and Sutton (2017), is a variant of NVLDA, in which the distribution over individual words is a product of experts. The original implementation is used.
ATM Wang et al. (2019), is a neural topic modeling approach based on adversarial training, we implement the ATM following the parameter setting suggested in the original paper.

4.2 Topic Coherence Evaluation

Topic models are typically evaluated with the likelihood of held-out documents and topic coherence. However, Chang et al. Chang et al. (2009) showed that a higher likelihood of held-out documents does not correspond to human judgment of topic coherence. Thus, we follow Röder et al. (2015) and employ four topic coherence metrics (C_P, C_A, NPMI and UCI) to evaluate the topics generated by various models. In all experiments, each topic is represented by the top 10 words according to the topic-word probabilities, and all the topic coherence values are calculated using the Palmetto library777

Dataset Model C_P C_A NPMI UCI
20Newsgroups NVDM -0.2558 0.1286 -0.0984 -2.9496
GSM -0.2318 0.1067 -0.0400 -1.6083
NVLDA 0.1205 0.1763 -0.0207 -1.3466
ProdLDA 0.1858 0.2155 -0.0083 -1.5044
LDA 0.2361 0.1769 0.0523 0.3399
ATM 0.1914 0.1720 0.0207 -0.3871
BAT 0.2597 0.1976 0.0472 0.0969
Gaussian-BAT 0.3758 0.2251 0.0819 0.5925
Grolier NVDM -0.1877 0.1456 -0.0619 -2.1149
GSM 0.1974 0.1966 0.0491 -0.0410
NVLDA -0.2205 0.1504 -0.0653 -2.4797
ProdLDA -0.0374 0.1733 -0.0193 -1.6398
LDA 0.1908 0.2009 0.0497 -0.0503
ATM 0.2105 0.2188 0.0582 0.1051
BAT 0.2312 0.2108 0.0608 0.1709
Gaussian-BAT 0.2606 0.2142 0.0724 0.2836
NYtimes NVDM -0.4130 0.1341 -0.1437 -4.3072
GSM 0.3426 0.2232 0.0848 0.6224
NVLDA -0.1575 0.1482 -0.0614 -2.4208
ProdLDA -0.0034 0.1963 -0.0282 -1.9173
LDA 0.3083 0.2127 0.0772 0.5165
ATM 0.3568 0.2375 0.0899 0.6582
BAT 0.3749 0.2355 0.0951 0.7073
Gaussian-BAT 0.4163 0.2479 0.1079 0.9215
Table 2: Average topic coherence on three datasets with five topic settings [20, 30, 50, 75, 100].
Figure 5: The comparison of average topic coherence vs. different topic number on 20Newsgroups, Grolier and NYTimes.
Model Topics
Gaussian-BAT voter campaign poll candidates democratic election republican vote presidential democrat
song album music band rock pop sound singer jazz guitar
film movie actor character movies director series actress young scenes
flight airline passenger airlines aircraft shuttle airport pilot carrier planes
BAT vote president voter campaign election democratic governor republican black candidates
album band music rock song jazz guitar pop musician record
film actor play acting role playing character father movie actress
flight airline delay airlines plane pilot airport passenger carrier attendant
LDA voter vote poll election campaign primary candidates republican race party
music song band sound record artist album show musical rock
film movie character play actor director movies minutes theater cast
flight plane ship crew air pilot hour boat passenger airport
ATM voter vote poll republican race primary percent election campaign democratic
music song musical album jazz band record recording mp3 composer
film movie actor director award movies character theater production play
jet flight airline hour plane passenger trip plan travel pilot
Table 3: Topic examples extracted by models, italics means out-of-topic words. These topics correspond to ‘election’, ‘music’, ‘film’ and ‘airline’ respectively, and topic examples of other models are omitted due to poor quality.

We firstly make a comparison of topic coherence vs. different topic proportions. Experiments are conducted on the datasets with five topic number settings [20, 30, 50, 75, 100]. We calculate the average topic coherence values among topics whose coherence values are ranked at the top 50, 70, 90, 100 positions. For example, to calculate the average C_P value of BAT , we first compute the average C_P coherence with the selected topics whose C_P values are ranked at the top 90% for each topic number setting, and then average the five coherence values with each corresponding to a particular topic number setting.

The detailed comparison is shown in Figure 4. It can be observed that BAT outperforms the baselines on all the coherence metrics for NYTimes datasets. For Grolier dataset, BAT outperforms all the baselines on C_P, NPMI and UCI metrics, but gives slightly worse results compared to ATM on C_A. For 20Newsgroups dataset, BAT performs the best on C_P and NPMI, but gives slightly worse results compared to ProdLDA on C_A, and LDA on UCI. By incorporating word embeddings through trainable Gaussian distribution, Gaussian-BAT outperforms all the baselines and BAT on four coherence metrics, often by a large margin, across all the three datasets except for Grolier dataset on C_A when considering 100% topics. This may be attribute to the following factors: (1) The Dirichlet prior employed in BAT and Gaussian-BAT could exhibit a multi-modal distribution in latent space and is more suitable for discovering semantic patterns from text; (2) ATM does not consider the relationship between topic distribution and word distribution since it only carry out adversarial training in word distribution space; (3) The incorporation of word embeddings in Gaussian-BAT helps generating more coherent topics.

We also compare the average topic coherence values (all topics taken into account) numerically to show the effectiveness of proposed BAT and Gaussian-BAT. The results of numerical topic coherence comparison are listed in Table 2 and each value is calculated by averaging the average topic coherences over five topic number settings. The best coherence value on each metric is highlighted in bold. It can be observed that Gaussian-BAT gives the best overall results across all metrics and on all the datasets except for Grolier dataset on C_A. To make the comparison of topics more intuitive, we provide four topic examples extracted by models in Table 3. It can be observed that the proposed BAT and Gaussian-BAT can generate more coherent topics.

Moreover, to explore how topic coherence varies with different topic numbers, we also provide the comparison of average topic coherence vs. different topic number on 20newsgroups, Grolier and NYTimes (all topics taken into account). The detailed comparison is shown in Figure 5. It could be observed that Gaussian-BAT outperforms the baselines with 20, 30, 50 and 75 topics except for Grolier dataset on C_A metric. However, when the topic number is set to 100, Gaussian-BAT performs slightly worse than LDA (e.g., UCI for 20Newsgroups and C_A for NYTimes). This may be caused by the increased model complexity due to the larger topic number settings. Likewise, BAT can achieve at least the second-best results among all the approaches in most cases for NYTimes dataset. For Grolier, BAT also performs the second-best except on C_A metric. However, for 20newsgroups, the results obtained by BAT are worse than ProdLDA (C_A) and LDA (UCI) due to the limited training documents in the dataset, though it still largely outperforms other baselines.

4.3 Text Clustering

We further compare our proposed models with baselines on text clustering. Due to the lack of document label information in Grolier and NYTimes, we only use 20Newsgroups dataset in our experiments. The topic number is set to 20 (ground-truth categories) and the performance is evaluated by accuracy :


where is the number of documents in the test set, is the indicator function, is the ground-truth label of -th document, is the category assignment, and ranges over all possible one-to-one mappings between labels and clusters. The optimal map function can be obtained by the Kuhn-Munkres algorithm Kuhn (1955). A larger accuracy value indicates a better text clustering results.

20NG 33.31% 33.82% 35.36% 35.66% 41.25%
Table 4: Text clustering accuracy on 20Newsgroups (20NG). ‘G-BAT’ refers to ‘Gaussian-BAT’. The best result is highlighted in bold.

The comparison of text clustering results on 20Newsgroups is shown in Table 4. Due to the poor performance of NVDM in topic coherence evaluation, its result is excluded here. Not surprisingly, NVLDA and ProdLDA perform worse than BAT and Gaussian-BAT that model topics with the Dirichlet prior. This might be caused by the fact that Logistic-Normal prior does not exhibit multiple peaks at the vertices of the simplex, as depicted in Figure 1. Compared with LDA, BAT achieves a comparable result in accuracy since both models have the same Dirichlet prior assumption over topics and only employ the word co-occurrence information. Gaussian-BAT outperforms the second best model, BAT, by nearly 6% in accuracy. This shows that the incorporation of word embeddings is important to improve the semantic coherence of topics and thus results in better consistency between cluster assignments and ground-truth labels.

5 Conclusion

In this paper, we have explored the use of bidirectional adversarial training in neural topic models and proposed two novel approaches: the Bidirectional Adversarial Topic (BAT) model and the Bidirectional Adversarial Topic model with Gaussian (Gaussian-BAT). BAT models topics with the Dirichlet prior and builds a two-way transformation between document-topic distribution and document-word distribution via bidirectional adversarial training. Gaussian-BAT extends from BAT by incorporating word embeddings into the modeling process, thereby naturally considers the word relatedness information captured in word embeddings. The experimental comparison on three widely used benchmark text corpus with the existing neural topic models shows that BAT and Gaussian-BAT achieve improved topic coherence results. In the future, we would like to devise a nonparametric neural topic model based on adversarial training. Besides, developing correlated topic modelsis another promising direction.


We would like to thank anonymous reviewers for their valuable comments and helpful suggestions. This work was funded by the National Key Research and Development Program of China(2017YFB1002801) and the National Natural Science Foundation of China (61772132). And YH is partially supported by EPSRC (grant no. EP/T017112/1).


  • M. Arjovsky, S. Chintala, and L. Bottou (2017) Wasserstein generative adversarial networks. In

    International conference on machine learning

    pp. 214–223. Cited by: §2.1, §3.3.
  • B. Athiwaratkun, A. Wilson, and A. Anandkumar (2018) Probabilistic fasttext for multi-sense word embeddings. In Proceedings of the 56th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pp. 1–11. Cited by: §3.2.
  • D. M. Blei, A. Y. Ng, and M. I. Jordan (2003) Latent dirichlet allocation. Journal of machine Learning research 3 (Jan), pp. 993–1022. Cited by: §1, §4.1.
  • J. Chang, S. Gerrish, C. Wang, J. L. Boyd-Graber, and D. M. Blei (2009) Reading tea leaves: how humans interpret topic models. In Advances in neural information processing systems, pp. 288–296. Cited by: §4.2.
  • X. Cheng, X. Yan, Y. Lan, and J. Guo (2014) Btm: topic modeling over short texts. IEEE Transactions on Knowledge and Data Engineering 26 (12), pp. 2928–2941. Cited by: §1.
  • J. Donahue, P. Krähenbühl, and T. Darrell (2016) Adversarial feature learning. arXiv preprint arXiv:1605.09782. Cited by: §2.1, §3.
  • V. Dumoulin, I. Belghazi, B. Poole, O. Mastropietro, A. Lamb, M. Arjovsky, and A. Courville (2016) Adversarially learned inference. arXiv preprint arXiv:1606.00704. Cited by: §2.1.
  • W. Fedus, I. Goodfellow, and A. M. Dai (2018) MaskGAN: better text generation via filling in the_. arXiv preprint arXiv:1801.07736. Cited by: §2.1.
  • I. Goodfellow, J. Pouget-Abadie, M. Mirza, B. Xu, D. Warde-Farley, S. Ozair, A. Courville, and Y. Bengio (2014) Generative adversarial nets. In Advances in neural information processing systems, pp. 2672–2680. Cited by: §2.1, §2.1.
  • T. L. Griffiths and M. Steyvers (2004) Finding scientific topics. Proceedings of the National academy of Sciences 101 (suppl 1), pp. 5228–5235. Cited by: §4.1.
  • I. Gulrajani, F. Ahmed, M. Arjovsky, V. Dumoulin, and A. C. Courville (2017) Improved training of wasserstein gans. In Advances in neural information processing systems, pp. 5767–5777. Cited by: §2.1.
  • G. E. Hinton and R. R. Salakhutdinov (2009) Replicated softmax: an undirected topic model. In Advances in neural information processing systems, pp. 1607–1614. Cited by: §2.2.
  • A. Joulin, E. Grave, and P. B. T. Mikolov (2017) Bag of tricks for efficient text classification. EACL 2017, pp. 427. Cited by: §3.2.
  • D. P. Kingma and J. Ba (2014) Adam: a method for stochastic optimization. arXiv preprint arXiv:1412.6980. Cited by: §3.3.
  • D. P. Kingma and M. Welling (2013) Auto-encoding variational bayes. arXiv preprint arXiv:1312.6114. Cited by: §1.
  • H. W. Kuhn (1955) The hungarian method for the assignment problem. Naval research logistics quarterly 2, pp. 83–97. Cited by: §4.3.
  • K. Lang (1995) Newsweeder: learning to filter netnews. In Proceedings of the Twelfth International Conference on Machine Learning, pp. 331–339. Cited by: §4.1.
  • Z. Lei, Y. Yang, and M. Yang (2018)

    SAAN: a sentiment-aware attention network for sentiment analysis

    In The 41st International ACM SIGIR Conference on Research & Development in Information Retrieval, pp. 1197–1200. Cited by: §1.
  • C. Lin and Y. He (2009) Joint sentiment/topic model for sentiment analysis. In Proceedings of the 18th ACM conference on Information and knowledge management, pp. 375–384. Cited by: §1.
  • K. Lin, D. Li, X. He, Z. Zhang, and M. Sun (2017) Adversarial ranking for language generation. In Advances in Neural Information Processing Systems, pp. 3155–3165. Cited by: §2.1.
  • Q. Liu, H. Zhang, Y. Zeng, Z. Huang, and Z. Wu (2018)

    Content attention model for aspect based sentiment analysis

    In Proceedings of the 2018 World Wide Web Conference, pp. 1023–1032. Cited by: §1.
  • Y. Miao, E. Grefenstette, and P. Blunsom (2017) Discovering discrete latent topics with neural variational inference. In Proceedings of the 34th International Conference on Machine Learning-Volume 70, pp. 2410–2419. Cited by: §2.2, §4.1.
  • Y. Miao, L. Yu, and P. Blunsom (2016) Neural variational inference for text processing. In International conference on machine learning, pp. 1727–1736. Cited by: §2.2, §4.1.
  • T. Mikolov, K. Chen, G. Corrado, and J. Dean (2013a)

    Efficient estimation of word representations in vector space

    arXiv preprint arXiv:1301.3781. Cited by: §3.2.
  • T. Mikolov, I. Sutskever, K. Chen, G. S. Corrado, and J. Dean (2013b) Distributed representations of words and phrases and their compositionality. In Advances in neural information processing systems, pp. 3111–3119. Cited by: §3.2.
  • J. Pennington, R. Socher, and C. Manning (2014) Glove: global vectors for word representation. In Proceedings of the 2014 conference on empirical methods in natural language processing (EMNLP), pp. 1532–1543. Cited by: §3.2, §3.2.
  • A. Radford, L. Metz, and S. Chintala (2015) Unsupervised representation learning with deep convolutional generative adversarial networks. arXiv preprint arXiv:1511.06434. Cited by: §2.1.
  • M. Röder, A. Both, and A. Hinneburg (2015) Exploring the space of topic coherence measures. In Proceedings of the eighth ACM international conference on Web search and data mining, pp. 399–408. Cited by: §4.2.
  • A. Srivastava and C. Sutton (2017) Autoencoding variational inference for topic models. arXiv preprint arXiv:1703.01488. Cited by: §1, §1, §2.2, §4.1.
  • H. M. Wallach, D. M. Mimno, and A. McCallum (2009) Rethinking lda: why priors matter. In Advances in neural information processing systems, pp. 1973–1981. Cited by: §1, §3.1.2.
  • R. Wang, D. Zhou, and Y. He (2019) Open event extraction from online text using a generative adversarial network. In Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP), Hong Kong, China, pp. 282–291. Cited by: §2.2.
  • R. Wang, D. Zhou, and Y. He (2019) Atm: adversarial-neural topic model. Information Processing & Management 56 (6), pp. 102098. Cited by: §1, §2.2, §4.1.
  • L. Yu, W. Zhang, J. Wang, and Y. Yu (2017) Seqgan: sequence generative adversarial nets with policy gradient. In

    Thirty-First AAAI Conference on Artificial Intelligence

    Cited by: §2.1.
  • D. Zhou, L. Chen, and Y. He (2014) A simple bayesian modelling approach to event extraction from twitter. In Proceedings of the 52nd Annual Meeting of the Association for Computational Linguistics (Volume 2: Short Papers), pp. 700–705. Cited by: §1.