Document Neural Autoregressive Distribution Estimation
Abstract
We present an approach based on feed-forward neural networks for learning the distribution of textual documents. This approach is inspired by the Neural Autoregressive Distribution Estimator (NADE) model, which has been shown to be a good estimator of the distribution of discrete-valued high-dimensional vectors. In this paper, we present how NADE can successfully be adapted to the case of textual data, retaining from NADE the property that sampling or computing the probability of observations can be done exactly and efficiently. The approach can also be used to learn deep representations of documents that are competitive to those learned by the alternative topic modeling approaches. Finally, we describe how the approach can be combined with a regular neural network N-gram model and substantially improve its performance, by making its learned representation sensitive to the larger, document-specific context.
120161-194/0010/00Stanislas Lauly, Yin Zheng, Alexandre Allauzen and Hugo Larochelle \ShortHeadingsDocNADELauly, Zheng, Allauzen & Larochelle \firstpageno1
Leslie Pack Kaelbling
Neural networks, Deep learning, Topic models, Language models, Autoregressive models
1 Introduction
One of the most common problem addressed by machine learning is estimating the distribution of multidimensional data from a set of examples . Indeed, good estimates for implicitly requires modeling the dependencies between the variables in , which is required to extract meaningful representations of this data or make predictions about this data.
The biggest challenge one faces in distribution estimation is the well-known curse of dimensionality. In fact, this issue is particularly important in distribution estimation, even more so than in other machine learning problems. This is because a good distribution estimator effectively requires providing an accurate value for for any value of (i.e. not only for likely values of ), with the number of possible values taken by growing exponentially as the number of the dimensions of the input vector increases.
One example of a model that has been successful at tackling the curse of dimensionality is the restricted Boltzmann machine (RBM) \citepHinton2002. The RBM and other models derived from it (e.g. the Replicated Softmax of [Salakhutdinov and Hinton(2009)]) are frequently trained as models of the probability distribution of high-dimensional observations and then used as feature extractors. Unfortunately, one problem with these models is that for moderately large models, calculating their estimate of is intractable. Indeed, this calculation requires computing the so-called partition function, which normalizes the model distribution. The consequences of this property of the RBM are that approximations must be taken to train it by maximum likelihood and its estimation of cannot be entirely trusted.
In an attempt to tackle these issues of the RBM, the Neural Autoregressive Distribution Estimator (NADE) was introduced by [Larochelle and Murray(2011)]. NADE’s parametrization is inspired by the RBM, but uses feed-forward neural networks and the framework of autoregression for modeling the probability distribution of binary variables in high-dimensional vectors. Importantly, computing the probability of an observation under NADE can be done exactly and efficiently.
In this paper, we describe a variety of ways to extend NADE to model data from text documents. We start by describing Document NADE (DocNADE), a single hidden layer feed-forward neural network model for bag-of-words observations, i.e. orderless sets of words. This requires adapting NADE to vector observations , where each of element represents a word and where the order of the dimensions is random. Each word is represented with a lower-dimensional, real-valued embedding vector, where similar words should have similar embeddings. This is in line with much of the recent work on using feed-forward neural network models to learn word vector embeddings \citepbengio2003neural,mnih2007three,Mnih+Hinton-2009,MikolovT2013 to counteract the curse of dimensionality. However, in DocNADE, the word representations are trained to reflect the topics (i.e. semantics) of documents only, as opposed to their syntactical properties, due to the orderless nature of bags-of-words.
Then, we describe how to train deep versions of DocNADE. First described by [Zheng et al.(2015)Zheng, Zhang, and Larochelle] in the context of image modeling, here we empirically evaluate them for text documents and show that they are competitive to alternative topic models, both in terms of perplexity and document retrieval performances.
Finally, we present how the topic-level modeling ability of DocNADE can be used to obtain a useful representation of context for language modeling. We empirically demonstrate that by learning a topical representation of previous sentences, we can improve the perplexity performance of an N-gram neural language model.
2 Document NADE (DocNADE)
DocNADE is derived from the Neural Autoregressive Distribution Estimation (NADE) that will be first described in this section. Implemented as a feed-forward architecture, it extends NADE to provide an efficient and meaningful generative model of document bags-of-words.
2.1 Neural Autoregressive Distribution Estimation (NADE)
NADE, introduced in [Larochelle and Murray(2011)], is a tractable distribution estimator for modeling the distribution of high-dimensional vectors of binary variables. Let us consider a binary vector of observations, . The NADE model estimates the probability of this vector by applying the probability chain rule as follows:
(1) |
where denotes the -th component of and contains the first components of : is the sub-vector . The peculiarity of NADE lies in the neural architecture designed to estimate the conditional probabilities involved in Equation 1. To predict the component , the model first computes its hidden layer of dimension
(2) |
leading to the following probability model:
(3) |
In these two equations, denotes the sigmoid activation function while function could be any activation function, though [Larochelle and Murray(2011)] also used the sigmoid function. and are the parameter matrices along with the associated bias terms and , with being a matrix made of the first columns of .
Instead of a single projection of the input vector, the NADE model relies on a set of separate hidden layers that each represent the previous inputs in a latent space. The connections between input dimension and each hidden layer are tied as shown in figure 1, allowing the model to compute all the hidden layers for one input in . The parameters are learned by minimizing the average negative log-likelihood using stochastic gradient descent.
2.2 From NADE to DocNADE
The Document NADE model (DocNADE) aims at learning meaningful representations of texts from an unlabeled collection of documents. This model embeds, like NADE, a set of hidden layers. Their role is to capture salient statistical patterns in the co-occurrence of words within documents and can be considered as modeling hidden topics.
To represent a document, the vector is now a sequence of arbitrary size . Each element of corresponds to a multinomial observation over a fixed vocabulary of size . Therefore represents the index in the vocabulary of the -th word of the document. For now, we’ll assume that an ordering for the words is given, but we will discuss the general case of orderless bags-of-words in Section 2.3.
The main approach taken by DocNADE is similar to NADE, but differs significantly in the design of parameter tying. The probability of a document is estimated using the probability chain rule, but the architecture is modified to cope with large vocabularies. Each word observation of the document leads to a hidden layer , which represents the past observations . This hidden layer is computed as follows:
(4) |
where each column of the matrix acts as a vector of size that represents a word. The embedding of the word in the document is thus the column of index in the matrix .
Notice that by sharing the word representation matrix across positions in the document, each hidden layer is in fact independent of the order of the words within . The implications of this choice is that the learned hidden representation will not model the syntactic structure of the documents and focus on its document-level semantics, i.e. its topics.
It is also worth noticing that we can compute recursively by keeping track of the pre-activation of the previous hidden layer as follows:
(5) |
The weight sharing between hidden layers enables us to compute all hidden layers for a document in .
Then, to compute the probability of a full document , we need to estimate all conditional probabilities . A straightforward solution would be to compute each using softmax layers with a shared weight matrix and bias, each fed with the corresponding hidden layer . However, the computational cost of this approach is prohibitive, since it scales linearly with the vocabulary size
More formally, let’s denote by the sequence of nodes composing the path, from the root of the tree to the leaf corresponding to word . Then, is the sequence of left/right decisions of the nodes in . For example, the root of the tree is always the first element and the value will be 0 if the word is in the left sub-tree and 1 if it is in the right sub-tree. The matrix stores by row the weights associated to each logistic classifier. There is one logistic classifier per node in the tree. Let and be the weights and bias for the logistic unit associated to the node . The probability given the tree and the hidden layer is computed with the following formulas:
(6) |
(7) |
This hierarchical architecture allows us to efficiently compute the probability for each word in a document and therefore the probability of every documents with the probability chain rules (see Equation 1). As in the NADE model, the parameters of the model are learnt by minimizing the negative log-likelihood using stochastic gradient descent.
Since there is logistic regression units for a word (one per node), each of them has a time complexity of , the complexity of computing the probability for a document of words is in .
As for using DocNADE to extract features from a complete document, we propose to use
(8) |
which would be the hidden layer computed to obtain the conditional probability of a word appearing in the document.
2.3 Training from bag-of-word counts
So far, we have assumed that the ordering of the words in the document is known. However, document datasets often take the form of word-count vectors in which the original word order, required to specify the sequence of conditionals , has been lost.
Thankfully, it is still possible to successfully train DocNADE despite the absence of this information. The idea is to assume that each observed document was generated by initially sampling a seed document from DocNADE, whose words were then shuffled using a randomly generated ordering to produce . With this approach, we can express the probability distribution of by computing the marginal over all possible seed document:
(9) |
where is modeled by DocNADE, is the same as the observed document but with a different (random) word sequence order and is the set of all the documents that have the same word count . With the assumption of orderings being uniformly sampled, we can replace with giving us:
(10) |
In practice, one approach to training the DocNADE model over is to artificially generate ordered documents by uniformly sampling words, without replacement, from the bags-of-words in the dataset. This would be equivalent to taking each original document and shuffling the order of its words. This approach can be shown to optimize a stochastic upper bound on the actual negative log-likelihood of documents. As we’ll see, experimental results show that convincing performance can still be reached.
With this training procedure, DocNADE shows its ability to learn and predict a new word in a document at a random position while preserving the overall semantic properties of the document. The model is therefore learning not to insert âintruderâ words, i.e. words that do not belong with the others. After training, a document’s learned representation should contain valuable information to identify intruder words for this document. It’s interesting to note that the detection of such intruder words has been used previously as a task in user studies to evaluate the quality of the topics learned by LDA, though at the level of single topics and not whole documents \citepChangJ2009.
3 Deep Document NADE
The single hidden layer version of DocNADE already achieves very competitive performance for topic modeling \citeplarochelle2012neural. Extending it to a deep, multiple hidden layer architecture could however allow for even better performance, as suggested by the recent and impressive success of deep neural networks. Unfortunately, deriving a deep version of DocNADE that is practical cannot be achieved solely by adding hidden layers to the definition of the conditionals . Indeed, computing requires computing each conditional (one for each word), and it is no longer possible to exploit Equation 5 to compute the sequence of all hidden layers in when multiple deep hidden layers are used.
In this section, we describe an alternative training procedure that enables us the introduction of multiple stacked hidden layers. This procedure was first introduced in [Zheng et al.(2015)Zheng, Zhang, and Larochelle] to model images, which was itself borrowing from the training scheme introduced by [Uria et al.(2014)Uria, Murray, and Larochelle].
As mentioned in section 2.3, DocNADE can be trained on random permutations of the words from training documents. As noticed by [Uria et al.(2014)Uria, Murray, and Larochelle], the use of many orderings during training can be seen as the instantiation of many different DocNADE models that share a single set of parameters. Thus, training DocNADE with random permutations also amounts to minimizing the negative log-likelihood averaged across all possible orderings, for each training example .
In the context of deep NADE models, a key observation is that training on all possible orderings implies that for a given context , we wish the model to be equally good at predicting any of the remaining words appearing next, since for each there is an ordering such that they appear at position . Thus, we can redesign the training algorithm such that, instead of sampling a complete ordering of all words for each update, we instead sample a single context and perform an update of the conditionals using that context. This is done as follows. For a given document, after generating vector by shuffling the words from the document, a split point is randomly drawn. From this split point results two parts of the document: and . The former is considered as the input and the latter contains the targets to be predicted by the model. Since in this setting a training update relies on the computation of a single latent representation, that of for the drawn value of , deeper hidden layers can be added at a reasonable increase in computation.
Thus, in DeepDocNADE the conditionals are modeled as follows. The first hidden layer represents the conditioning context as in the single hidden layer DocNADE:
(11) |
where is the histogram vector representation of the word sequence , and the exponent is used as an index over the hidden layers and its parameters, with referring to the first layer. We can now easily add new hidden layers as in a regular deep feed-forward neural network:
(12) |
for , where is the total number of hidden layers. From the last hidden layer , we can finally compute the conditional , for any word .
Finally, the loss function used to update the model for the given context is:
(13) |
where is the number of words in and the sum iterates over all words present in . Thus, as described earlier, the model predicts each remaining word after the splitting position as if it was actually at position . The factors in front of the sum comes from the fact that the complete log-likelihood would contain log-conditionals and that we are averaging over possible choices for the word ordered at position . For a more detailed presentation, see [Zheng et al.(2015)Zheng, Zhang, and Larochelle]. The average loss function of Equation 13 is optimized with stochastic gradient descent
Note that to compute the probabilities , a probabilistic tree could again be used. However, since all probabilities needed for an update are based on a single context , a single softmax layer is sufficient to compute all necessary quantities. Therefore the computational burden of a conventional softmax is not as prohibitive as for DocNADE, especially with an efficient implementation on the GPU. For this reason, in our experiments with DeepDocNADE we opted for a regular softmax.
4 DocNADE Language Model
While topic models such as DocNADE can be useful to learn topical representations of documents, they are actually very poor models of language. In DocNADE, this is due to the fact that, when assigning a probability to the next word in a sentence, DocNADE actually ignores in which order the previously observed words appeared. Yet, this ordering of words conveys a lot of information regarding the most likely syntactic role of the next word or the finer semantics within the sentence. In fact, most of that information is predictable from the last few words, which is why N-gram language models remain the dominating approach to language modeling.
In this section, we propose a new model that extends DocNADE to mitigate the influence of both short and long term dependencies in a single model, which we refer to as the DocNADE language model or DocNADE-LM. The solution we propose enhances the bag-of-word representation of a word’s history with the explicit inclusion of -gram dependencies for each word to predict.
The figure 3 depicts the overall architecture. This model can be seen as an extension of the seminal work on neural language models of [Bengio et al.(2003)Bengio, Ducharme, Vincent, and Janvin] that includes a representation of a document’s larger context. It can also be seen as a neural extension of the cache-based language model introduced in \citepkuhn90, where the -gram probability is interpolated with the word distribution observed in a dynamic cache. This cache of a fixed size keeps track of previously observed words to include long term dependencies in the prediction and preserve semantic consistency beyond the scope of the -gram. In our case, the DocNADE language model maintains an unbounded cache and define a proper, jointly trained solution to mitigate these two kinds of dependencies.
As in DocNADE, a document is modeled as a sequence of multinomial observations. The sequence size is arbitrary and each element consists in the index of the -th word in a vocabulary of size . The conditional probability of a word given its history is now expressed as a smooth function of a hidden layer used to predict word . The peculiarity of the DocNADE language model lies in the definition of this hidden layer, which now includes two terms:
(14) |
The first term borrows from the DocNADE model by aggregating embeddings for all the previous words in the history:
(15) |
The second contribution derives from neural -gram language models as follows:
(16) |
In this formula, the history for the word is restricted to the preceding words, following the common -gram assumption. The term hence represents the continuous representation of these words, in which word embeddings are linearly transformed by the ordered matrices . Moreover, gathers the bias terms for the hidden layer. In this model, two sets of word embeddings are defined, and , which are respectively associated to the DocNADE and neural language model parts. For simplicity, we assume both are of the same size .
Given hidden layer , conditional probabilities can be estimated, and thus . For the aforementioned reason, the output layer is structured for efficient computations. Specifically, we decided to use a variation of the probabilistic binary tree, known as a hierarchical softmax layer. In this case, instead of having binary nodes with multiple levels in the tree, we have only two levels where all words have their leaf at level two and each node is a multiclass (i.e. softmax) logistic regression with roughly classes (one for each children). Computing probabilities in such a structured layer can be done using only two matrix multiplications, which can be efficiently computed on the GPU.
With a hidden layer of size , the complexity of computing the softmax at one node is . If we have words in a given document, the complexity of computing all necessary probabilities from the hidden layers is thus . It also requires computations to compute the hidden representations for the DocNADE part and for the language model part. The full complexity for computing and the updates for the parameters are thus computed in .
Once again, the loss function of the model is the negative log-likelihood and we minimize it by using stochastic gradient descent over documents, to learn the values of the parameters .
5 Related Work
Much like NADE was inspired by the RBM, DocNADE can be seen as related to the Replicated Softmax model \citepSalakhutdinov-NIPS2010-softmax, an extension of the RBM to document modeling. Here, we describe in more detail the Replicated Softmax, along with its relationship with DocNADE.
Much like the RBM, the Replicated Softmax models observations using a latent, stochastic binary layer . Here, the observations are the documents , which interact with the hidden layer through an energy function similar to RBM’s:
(17) |
where is a bag-of-word vector of size (the size of the vocabulary) containing the word count of each word in the vocabulary for document . is the stochastic, binary hidden layer vector and is the column vector of matrix . and are the bias vectors for the visible and the hidden layers. We see here that the larger is, the bigger the number of terms in the sum over is, resulting in a high energy value. For this reason, the hidden bias term is multiplied by , to be commensurate with the contribution of the visible layer.. We can see also that connection parameters are shared across different positions in , as illustrated by figure 4).
The conditional probabilities of the hidden and the visible layer factorize much like in the RBM, in the following way:
(18) |
where the factors and are such that
(19) | ||||
(20) |
The normalized exponential part in is simply the softmax function. To train this model, we’d like to minimize the negative log-likelihood (NLL). Its gradients for a document with respect to the parameters are calculated as follows:
(21) |
As with conventional RBMs, the second expectation in Equation 21 is computationally too expensive. The gradient of the negative log-likelihood is therefore approximated by replacing the second expectation with an estimated value obtained by contrastive divergence \citepHinton2002. This approach consists of performing steps of blocked Gibbs sampling, starting at and using Equations 19 and 20, to obtain point estimates of the expectation over . Large values of must be used to reduce the bias of gradient estimates and obtain good estimates of the distribution. This approximation is used to perform stochastic gradient descent.
During Gibbs sampling, the Replicated Softmax model must compute and sample from , which requires the computation of a large softmax. Most importantly, the computation of the softmax most be repeated times for each update, which can be prohibitive, especially for large vocabularies. Unlike DocNADE, this softmax cannot simply be replaced by a structured softmax.
It is interesting to see that DocNADE is actually related to how the Replicated Softmax approximates, through mean-field inference, the conditionals . Computing the conditional with the Replicated Softmax is intractable. However, we could use mean-field inference to approximate the full conditional as the factorized
(22) |
where and . We would find the parameters and that minimize the KL divergence between and by applying the following message passing equations until convergence:
(23) | |||
(24) |
with , and . The conditional could then be estimated with for all . We note that one iteration of mean-field (with initialized to 0) in the Replicated Softmax corresponds to the conditional computed by DocNADE with a single hidden layer and a flat softmax output layer.
In our experiment, we’ll see that DocNADE compares favorably to Replicated Softmax.
6 Topic modeling experiments
To compare the topic models, two kinds of quantitative comparisons are used. The first one evaluates the generative ability of the different models, by computing the perplexity of held-out texts. The second one compares the quality of document representations for an information retrieval task.
Two different datasets are used for the experiments of this section, 20 Newsgroups and RCV1-V2 (Reuters Corpus Volume I). The 20 Newsgroups corpus has 18,786 documents (postings) partitioned into 20 different classes (newsgroups). RCV1-V2 is a much bigger dataset composed of 804,414 documents (newswire stories) manually categorized into 103 classes (topics). The two datasets were preprocessed by stemming the text and removing common stop-words. The 2,000 most frequent words of the 20 Newsgroups training set and the 10,000 most frequent words of the RCV1-V2 training set were used to create the dictionary for each dataset. Also, every word counts , used to represent the number of times a word appears in a document was replaced by rounded to the nearest integer, following [Salakhutdinov and Hinton(2009)].
6.1 Generative Model Evaluation
For the generative model evaluation, we follow the experimental setup proposed by \citetSalakhutdinov-NIPS2010-softmax for 20 Newsgroups and RCV1-V2 datasets. We use the exact same split for the sake of comparison. The setup consists in respectively 11,284 and 402,207 training examples for 20 Newsgroups and RCV1-V2. We randomly extracted 1,000 and 10,000 documents from the training sets of 20 Newsgroups and RCV1-V2, respectively, to build a validation set. The average perplexity per word is used for comparison. This perplexity is estimated using the 50 first test documents, as follows:
(25) |
where is the total number of examples and is the test document
Dataset | LDA | Replicated | fDARN | DocNADE | DeepDN | DeepDN | DeepDN |
---|---|---|---|---|---|---|---|
Softmax | (1layer) | (2layer) | (3layer) | ||||
20 News | 1091 | 953 | 917 | 896 | 835 | 877 | 923 |
RCV1-v2 | 1437 | 988 | 724 | 742 | 579 | 552 | 539 |
Table 1 gathers the perplexity per word results for 20 Newsgroups and RCV1-V2. Theere we compare 5 different models: the Latent Dirichlet Allocation (LDA) \citepBleiD2003, the Replicated Softmax, the recent fast Deep AutoRegressive Networks (fDARN) \citepmnih2014neural, DocNADE and DeepDocNADE (DeepDN in the table). Each model uses 50 latent topics. For the experiments with DeepDocNADE, we provide the performance when using 1, 2, and 3 hidden layers. As shown in Table 1, DeepDocNADE provides the best generative performances. Our best DeepDocNADE models were trained with the Adam optimizer \citepkingma2014adam and with the tanh activation function. The hyper-parameters of Adam were selected on the validation set.
Crucially, following [Uria et al.(2014)Uria, Murray, and Larochelle], an ensemble approach is used to compute the probability of documents, where each component of the ensembles are the same DeepDocNADE model evaluated on a different word ordering. Specifically, the perplexity with ensembles becomes as follows:
(26) |
where is the total number of examples, is the number of ensembles (word orderings) and denotes the word ordering for the documnet. We try , with the results in Table 1 using . For the 20 Newsgroups dataset, adding more hidden layers to DocDocNADE fails to provide further improvements. We hypothesize that the relatively small size of this dataset makes it hard to successfully train a deep model. However, the opposite is observed on the RCV1-V2 dataset, which is more than an order of magnitude larger than 20 Newsgroups. In this case, DeepDocNADE outperforms fDARN and DocNADE, with a relative perplexity reduction of 20%, 24% and 26% with respectively 1,2 and 3 hidden layers.
To illustrate the impact of on the performance of DeepDocNADE, Figure 5 shows the perplexity on both datasets using the different values for that we tried. We can observe that beyond , this hyper-parameter has only a minimal impact on the perplexity.
6.2 Document Retrieval Evaluation
A document retrieval evaluation task was also used to evaluate the quality of the document representation learned by each model. As in the previous section, the datasets under consideration are 20 Newsgroups and RCV1-V2. The experimental setup is the same for the 20 Newsgroups dataset, while for the RCV1-V2 dataset, we reproduce the same setup as the one used in \citetsrivastava2013modeling, where the training set contained 794,414 examples and 10,000 examples constituted the test set.
For DocNADE and DeepDocNADE, the representation of a document is obtained simply by computing the top-most hidden layer when feeding all words as input.
The retrieval task follows the setup of [Srivastava et al.(2013)Srivastava, Salakhutdinov, and Hinton]. The documents in the training and validation sets are used as the database for retrieval, while the test set is used as the query set. The similarity between a query and all examples in the database is computed using the cosine similarity between their vector representations. For each query, documents in the database are then ranked according to this similarity, and precision/recall (PR) curves are computed, by comparing the label of the query documents with those of the database documents. Since documents sometimes have multiple labels (specifically those in RCV1-V2), for each query the PR curves for each of its labels are computed individually and then averaged. Finally, we report the global average of these (query-averaged) curves to compare models against each other. Training and model selection is otherwise performed as in the generative modeling evaluation.
As shown in Figure 6, DeepDocNADE always yields very competitive results, on both datasets, and outperforming the other models in most cases. Specifically, for the 20 Newsgroups dataset, DeepDocNADE with 2 and 3 hidden layers always perform better than the other methods. DeepDocNADE with 1 hidden layer also performs better than the other baselines when retrieving the top few documents ( e.g. when recall is smaller than 0.2).
6.3 Qualitative Inspection of Learned Representations
In this section, we want to assess if the DocNADE approach for topic modeling can capture meaningful semantic properties of texts.
First, one way to explore the semantic properties of trained models is through their learned word embeddings. Each of the columns of the matrix represents a word where is the vector representation of . Table 3 shows for some chosen words the five nearest words according to their embeddings, foro a DocNADE model. We can observe for each example the semantic consistency of the word representations. Similar results can be observed for DeepDocNADE models.
weapons | medical | companies | define | israel | book | windows |
---|---|---|---|---|---|---|
weapon | treatment | demand | defined | israeli | reading | dos |
shooting | medecine | commercial | definition | israelis | read | microsoft |
firearms | patients | agency | refer | arab | books | version |
assault | process | company | make | palestinian | relevent | ms |
armed | studies | credit | examples | arabs | collection | pc |
We’ve also attempted to measure whether the hidden units of the first hidden layer of DocNADE and DeepDocNADE models modeled distinct topics. Understanding the function represented by hidden units in neural networks is not a trivial affair, but we considered the following, simple approach. For a given hidden unit, its connections to words were interpreted as the importance of the word for the associated topic. Therefore, for a hidden unit, we selected the words having the strongest positive connections, i.e. for the hidden unit we chose the words that have the highest connection values .
With this approach, four topics were obtained from a DocNADE model using the sigmoid activation function and trained on 20 Newsgroups, as shown in Table 3 and can be readily interpreted as topics representing religion, space, sports and security. Note that those four topics are actually (sub)categories in 20 Newsgroups.
That said, we’ve had less success understanding the topics extracted when using the tanh activation function or when using DeepDocNADE. It thus seems that these models are then choosing to learn a latent representation that isn’t aligning its dimensions with concepts that are easily interpretable, even though it is clearly capturing well the statistics of documents (since our qualitative results with DeepDocNADE are excellent).
Hidden unit topics | |||
---|---|---|---|
jesus | shuttle | season | encryption |
atheism | orbit | players | escrow |
christianity | lunar | nhl | pgp |
christ | spacecraft | league | crypto |
athos | nasa | braves | nsa |
atheists | space | playoffs | rutgers |
bible | launch | rangers | clipper |
christians | saturn | hockey | secure |
sin | billion | pitching | encrypted |
atheist | satellite | team | keys |
7 Language Modeling Experiments
In this section, we test whether our proposed approach to incorporating a DocNADE component to a neural language model can improve the performance of a neural language model. Specifically, we considered treating a text corpus as a sequence of documents. We used the APNews dataset, as provided by \citetMnih+Hinton-2009. Unfortunately, information about the original segmentation into documents of the corpus wasn’t available in the data as provided by \citetMnih+Hinton-2009, thus we simulated the presence of documents by grouping one or more adjacent sentences, for training and evaluating DocNADE-LM, making sure the generated documents were non-overlapping. Thankfully, this approach still allows us to test whether DocNADE-LM is able to effectively leverage the larger context of words in making its predictions.
Since language models are generative models, the perplexity measured on some held-out texts provides an intrinsic and widely used evaluation criterion. Following \citetmnih2007three and \citetMnih+Hinton-2009, we used the APNews dataset containing Associated Press news stories from 1995 and 1996. The dataset is again split into training, validation and test sets, with respectively 633,143, 43,702 and 44,601 sentences. The vocabulary is composed of 17,964 words. A 100 dimensional feature vectors are used for these experiments. The validation set is used for model selection and the perplexity scores of Table 4 are computed on the test set.
Models | Number of grouped sentences | Perplexity |
---|---|---|
KN5 | - | 123.2 |
KN6 | - | 123.5 |
LBL | - | 117.0 |
HLBL | - | 112.1 |
FFN | - | 119.78 |
DocNADE-LM | 1 | 111.93 |
DocNADE-LM | 2 | 110.9 |
DocNADE-LM | 3 | 109.8 |
DocNADE-LM | 4 | 109.78 |
DocNADE-LM | 5 | 109.22 |
The FFN model in Table 4 corresponds to a regular neural (feed-forward) network language model. It is equivalent to setting to zero in Equation 14. These results are meant to measure whether the DocNADE part of DocNADE-LM can indeed help to improve performances.
We also compare to the log-bilinear language (LBL) model of [Mnih and Hinton(2007)]). While for the FFN model we used a hierarchical softmax to compute the conditional word probabilities (see Section 4), the LBL model uses a full softmax output layer that uses the same word representation matrix at the input and output. This latter model is therefore slower to train. Later, \citetMnih+Hinton-2009 also proposed adaptive approaches to learning a structured softmax layer, thus we also compare with their best approach. All aforementioned baselines are 6-gram models, taking in consideration the last 5 previous words to predict the next one. We also compare with more traditional 5-gram and 6-gram models using Kneser-Ney smoothing, taken from \citetmnih2007three.
From Table 4, we see that adding context to DocNADE-LM, by increasing the size of the multi-sentence segments, significantly improves the performance of the model (compared to FFN) and also surpasses the performance of the most competitive alternative, the HLBL model.
7.1 Qualitative Inspection of Learned Representations
In this section we explore the semantic properties of texts learned by the DocNADE-LM model. Interestingly, we can examine the two different components (DN and LM) separately. Because the DocNADE part and the language modeling part of the model each have their own word matrix, and respectively, we can compare their contribution through these learned embeddings. As explained in the previous section, each of the columns of the matrices represents a word where and are two different vector representations of the same word .
We can see by observing Tables 5 and 6 that the two parts of the DocNADE-LM model have learned different semantic properties of words. An interesting example is seen in the nearest neighbors of the word israel, where the DocNADE focuses on the politico-cultural relation between these words, whereas the language model part seems to have learned the concept of countries in general.
weapons | medical | companies | define | israel | book | windows |
---|---|---|---|---|---|---|
security | doctor | industry | spoken | israeli | story | yards |
humanitarian | health | corp | to | palestinian | novel | rubber |
terrorists | medicine | firms | think_of | jerusalem | author | piled |
deployment | physicians | products | of | lebanon | joke | fire_department |
diplomats | treatment | company | bottom_line | palestinians | writers | shell |
weapons | medical | companies | define | israel | book | windows |
---|---|---|---|---|---|---|
systems | special | countries | talk_about | china | film | houses |
aircraft | japanese | nations | place | russia | service | room |
drugs | bank | states | destroy | cuba | program | vehicle |
equipment | media | americans | show | north_korea | movie | restaurant |
services | political | parties | over | lebanon | information | car |
8 Conclusion
We have presented models inspired by NADE that can achieve state-of-the-art performances for modeling documents.
Indeed, for topic modeling, DocNADE had competitive results while its deep version, DeepDocNADE, outperformed the current state-of-the-art in generative document modeling, based on test set perplexity. The similarly good performances where observed when we used these models as feature extractors to represent documents for the task of information retrieval.
As for language modeling, the competitive performances of the DocNADE language models showed that combining contextual information by leveraging the DocNADE neural network architecture can significantly improve the performance of a neural probabilistic N-gram language model.
Footnotes
- For most natural language processing task, the vocabulary size exceeds .
- A document is usually represented as bag-of-words. Generating a word vector from its bag-of-words, shuffling the word count vector , splitting it, and then regenerating the histogram and is unfortunately fairly inefficient for processing samples in a mini-batch fashion. Hence, in practice, we split the original histogram directly by uniformly sampling, for each word individually, how many are put on the left of the split (the others are put on the right of the split). This procedure, used also by [Zheng et al.(2015)Zheng, Zhang, and Larochelle], is only an approximation of the correct procedure mentioned in the main text, but produces a substantial speedup while also yielding good performance. Thus, we used it also in this paper.
- Note that there is a difference between the sizes, for the training sets and test sets of 20 Newsgroups and RCV1-V2 reported in this paper and the one reported in the original data paper of \citetSalakhutdinov-NIPS2010-softmax. The correct values are the ones given in this section, which was confirmed after personal communication with \citetSalakhutdinov-NIPS2010-softmax.
References
- Yoshua Bengio, Réjean Ducharme, Pascal Vincent, and Christian Janvin. A neural probabilistic language model. The Journal of Machine Learning Research, 3:1137–1155, 2003.
- David M. Blei, Andrew Y. Ng, and Michael I. Jordan. Latent Dirichlet Allocation. Journal of Machine Learning Research, 3(4-5):993–1022, 2003.
- Jonathan Chang, Jordan Boyd-Graber, Sean Gerrish, Chong Wang, and David Blei. Reading Tea Leaves: How Humans Interpret Topic Models. In Advances in Neural Information Processing Systems 22 (NIPS 2009), pages 288–296, 2009.
- Geoffrey E. Hinton. Training products of experts by minimizing contrastive divergence. Neural Computation, 14:1771–1800, 2002.
- Diederik Kingma and Jimmy Ba. Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980, 2014.
- R. Kuhn and R. De Mori. A cache-based natural language model for speech recognition. IEEE Transactions on Pattern Analysis and Machine Intelligence, 12(6):570–583, june 1990.
- Hugo Larochelle and Stanislas Lauly. A neural autoregressive topic model. In Advances in Neural Information Processing Systems, pages 2708–2716, 2012.
- Hugo Larochelle and Ian Murray. The Neural Autoregressive Distribution Estimator. In Proceedings of the 14th International Conference on Artificial Intelligence and Statistics (AISTATS 2011), volume 15, pages 29–37, Ft. Lauderdale, USA, 2011. JMLR W&CP.
- Andriy Mnih and Karol Gregor. Neural variational inference and learning in belief networks. arXiv preprint arXiv:1402.0030, 2014.
- Andriy Mnih and Geoffrey Hinton. Three new graphical models for statistical language modelling. In Proceedings of the 24th international conference on Machine learning, pages 641–648. ACM, 2007.
- Andriy Mnih and Geoffrey E Hinton. A Scalable Hierarchical Distributed Language Model. In Advances in Neural Information Processing Systems 21 (NIPS 2008), pages 1081–1088, 2009.
- Frederic Morin and Yoshua Bengio. Hierarchical Probabilistic Neural Network Language Model. In Proceedings of the 10th International Workshop on Artificial Intelligence and Statistics (AISTATS 2005), pages 246–252. Society for Artificial Intelligence and Statistics, 2005.
- Ruslan Salakhutdinov and Geoffrey Hinton. Replicated Softmax: an Undirected Topic Model. In Advances in Neural Information Processing Systems 22 (NIPS 2009), pages 1607–1614, 2009.
- Nitish Srivastava, Ruslan R Salakhutdinov, and Geoffrey E Hinton. Modeling documents with deep boltzmann machines. arXiv preprint arXiv:1309.6865, 2013.
- Greg Corrado Jeffrey Dean Tomas Mikolov, Kai Chen. Efficient Estimation of Word Representations in Vector Space. In Workshop Track of the 1st International Conference on Learning Representations (ICLR 2013), 2013.
- Benigno Uria, Iain Murray, and Hugo Larochelle. A deep and tractable density estimator. JMLR: W&CP, 32(1):467–475, 2014.
- Yin Zheng, Yu-Jin Zhang, and Hugo Larochelle. A deep and autoregressive approach for topic modeling of multimodal data. TPAMI, 2015.