nvil_example
Reference implementation of the NVIL algorithm for a simple Gaussian Mixture Model
view repo
Highly expressive directed latent variable models, such as sigmoid belief networks, are difficult to train on large datasets because exact inference in them is intractable and none of the approximate inference methods that have been applied to them scale well. We propose a fast noniterative approximate inference method that uses a feedforward network to implement efficient exact sampling from the variational posterior. The model and this inference network are trained jointly by maximizing a variational lower bound on the loglikelihood. Although the naive estimator of the inference model gradient is too highvariance to be useful, we make it practical by applying several straightforward modelindependent variance reduction techniques. Applying our approach to training sigmoid belief networks and deep autoregressive networks, we show that it outperforms the wakesleep algorithm on MNIST and achieves stateoftheart results on the Reuters RCV1 document dataset.
READ FULL TEXT VIEW PDF
Learning latent variable models with stochastic variational inference is...
read it
How can we perform efficient inference and learning in directed probabil...
read it
Deep dynamic generative models are developed to learn sequential depende...
read it
Variational methods that rely on a recognition network to approximate th...
read it
We explore a new research direction in Bayesian variational inference wi...
read it
In this note, we study the relationship between the variational gap and ...
read it
Inference models are a key component in scaling variational inference to...
read it
Reference implementation of the NVIL algorithm for a simple Gaussian Mixture Model
Neural Variational Inference and Learning for Sigmoid Belief Network
Probabilistic programming for large datasets, via stochastic variational inference
None
Compared to powerful globallynormalized latent variable models, such as deep belief networks
(Hinton et al., 2006)and deep Boltzmann machines
(Salakhutdinov & Hinton, 2009a), which can now be trained on fairly large datasets, their purely directed counterparts have been left behind due to the lack of efficient learning algorithms. This is unfortunate, because their modularity and ability to generate observations efficiently make them better suited for integration into larger systems.Training highly expressive directed latent variable models on large datasets is a challenging problem due to the difficulties posed by inference. Although the generality of Markov Chain Monte Carlo (MCMC) methods makes them straightforward to apply to models of this type
(Neal, 1992), they tend to suffer from slow mixing and are usually too computationally expensive to be practical in all but the simplest models. Such methods are also difficult to scale to large datasets because they need to store the current state of the latent variables for all the training observations between parameter updates.Variational methods (Jordan et al., 1999) provide an optimizationbased alternative to the samplingbased Monte Carlo methods, and tend to be more efficient. They involve approximating the exact posterior using a distribution from a more tractable family, often a fully factored one, by maximizing a variational lower bound on the loglikelihood w.r.t. the parameters of the distribution. For a small class of models, using such variational posteriors allows the expectations that specify the parameter updates to be computed analytically. However, for highly expressive models such as the ones we are interested in, these expectations are intractable even with the simplest variational posteriors. This difficulty is usually dealt with by lower bounding the intractable expectations with tractable one by introducing more variational parameters, as was done for sigmoid belief nets by Saul et al. (1996). However, this technique increases the gap between the bound being optimized and the loglikelihood, potentially resulting in a poorer fit to the data. In general, variational methods tend to be more modeldependent than samplingbased methods, often requiring nontrivial modelspecific derivations.
We propose a new approach to training directed graphical models that combines the advantages of the samplingbased and variational methods. Its central idea is using a feedforward network to implement efficient exact sampling from the variational posterior for the given observation. We train this inference network jointly with the model by maximizing the variational lower bound on the loglikelihood, estimating all the required gradients using samples from the inference network. Although naive estimate of the gradient for the inference network parameters is unusable due to its high variance, we make the approach practical by applying several straightforward and general variance reduction techniques. The resulting training procedure for the inference network can be seen as an instance of the REINFORCE algorithm (Williams, 1992). Due to our use of stochastic feedforward networks for performing inference we call our approach Neural Variational Inference and Learning (NVIL).
Compared to MCMC methods, where many iterations over the latent variables are required to generate a sample from the exact posterior and successive samples tend to be highly correlated, NVIL does not suffer from mixing issues as each forward pass through the inference network generates an independent exact sample from the variational posterior. In addition to being much faster than MCMC, our approach has the additional advantage of not needing to store the latent variables for each observation and thus is not only more memory efficient but also applicable to the pure online learning setting, where each training case is seen once before being discarded.
In contrast to other work on scaling up variational inference, NVIL can handle both discrete and continuous latent variables (unlike Kingma & Welling (2013); Rezende et al. (2014)) as well variational posteriors with complex dependency structures (unlike Ranganath et al. (2013)). Moreover, the variance reduction methods we employ are simple and modelindependent, unlike the more sophisticated modelspecific control variates of Paisley et al. (2012).
Though the idea of training an inference model by following the gradient of the variational bound has been considered before, it was dismissed as infeasible (Dayan & Hinton, 1996). Our primary contribution is to show how to reduce the variance of the naive gradient estimator to make it practical without narrowing its range of applicability. We also show that the resulting method trains sigmoid belief networks better than the wakesleep algorithm (Hinton et al., 1995), which is the only algorithm we are aware of that is capable of training the same range of models efficiently. Finally, we demonstrate the effectiveness and scalability of NVIL by using it to achieve stateoftheart results on the Reuters RCV1 document dataset.
Suppose we are interested in training a latent variable model with parameters . We assume that exact inference in the model is intractable and thus maximum likelihood learning is not an option. For simplicity, we will also assume that all the latent variables in the model are discrete, though essentially the same approach applies if some or all of the variables are continuous.
We will train the model by maximizing a variational lower bound on the marginal loglikelihood. Following the standard variational inference approach (Jordan et al., 1999), given an observation , we introduce a distribution with parameters , which will serve as an approximation to its exact posterior . The variational posterior will have a simpler form than the exact posterior and thus will be easier to work with.
The contribution of to the loglikelihood can then be lowerbounded as follows (Jordan et al., 1999):
(1)  
By rewriting the bound as
(2) 
we see that its tightness is determined by the KullbackLeibler (KL) divergence between the variational distribution and the exact posterior. Maximizing the bound with respect to the parameters of the variational distribution makes the distribution a better approximation to the posterior (w.r.t. the KLdivergence) and tightens the bound.
In contrast to most applications of variational inference where the variational posterior for each observation is defined using its own set of variational parameters, our approach does not use any local variational parameters. Instead, we use a flexible feedforward model to compute the variational distribution from the observation. We call the model mapping to the inference network. The architecture of the inference network is constrained only by the requirement that it defines has to be efficient to evaluate and sample from. Using samples from the inference network we will be able to compute gradient estimates for the model and inference network parameters for a large class of highly expressive architectures, without having to deal with architecturespecific approximations.
Given a training set , consisting of observations , we train the model by (locally) maximizing using gradient ascent w.r.t. to the model and inference network parameters. To ensure scalability to large datasets, we will perform stochastic optimization by estimating gradients on small minibatches of randomly sampled training cases.
The gradient of the variational bound for a single observation w.r.t. to the model parameters is straightforward to derive and has the form
(3) 
where we left and off the list of the arguments of to simplify the notation. The corresponding gradient w.r.t. to the inference network parameters is somewhat more involved:
(4) 
We give its derivation in the supplementary material.
As both gradients involve expectations which are intractable in all but a handful of special cases, we will estimate them with Monte Carlo integration, using samples from the inference network. Having generated samples from , we compute
(5) 
and
(6) 
The above gradient estimators are unbiased and thus can be used to perform stochastic maximization of the variational objective using a suitable learning rate annealing schedule. The speed of convergence of this procedure, however, depends heavily on the variance of the estimators used, as we will see in Section 4.2.
The model gradient estimator (5) is wellbehaved and does not pose a problem. The variance of the inference network gradient estimator (2.2), however, can be very high due to the scaling of the gradient inside the expectation by a potentially large term. As a result, learning variational parameters with updates based on this estimator can be unacceptably slow. In fact, it is widely believed that learning variational parameters using gradient estimators of the form (2.2) is infeasible (Hinton & Zemel, 1994; Dayan & Hinton, 1996; Kingma & Welling, 2013). In the next section we will show how to make this approach practical by applying variance reduction techniques.
Though gradient estimates computed using Eq. 2.2 are usually too noisy to be useful in practice, it is easy to reduce their variance to a manageable level with the following modelindependent techniques.
Inspecting Eq. 2.2, we see that we are using
(7) 
as the learning signal for the inference network parameters, and thus are effectively fitting to . This might seem surprising, given that we want the inference network to approximate the posterior distribution
, as opposed to the joint distribution
. It turns out however that using the joint instead of the posterior distribution in Eq. 2.2 does not affect the value of the expectation. To see that we start by noting that(8) 
Therefore we can subtract any that does not depend on from the learning signal in Eq. 2.2 without affecting the value of the expectation:
(9) 
And as and does not depend on , using in Eq. 2.2 in place of does not affect the value of the expectation.
This equivalence allows us to compute the learning signal efficiently, without having to evaluate the intractable term. The price we pay for this tractability is the much higher variance of the estimates computed using Eq. 2.2. Fortunately, Eq. 2.3.1 suggests that we can reduce the variance by subtracting a carefully chosen from the learning signal. The simplest option is to make a parameter and adapt it as learning progresses. However, will not be able capture the systematic differences in the learning signal for different observations , which arise in part due to the presence of the term. Thus we can reduce the gradient variance further by subtracting an observationdependent term to minimize those differences. Doing this does not affect the expected value of the gradient estimator because
does not depend on the latent variables. Borrowing a name from the reinforcement learning literature we will refer to
and as baselines. We will elaborate on this connection in Section 3.4.We implement the inputdependent baseline
using a neural network and train it to minimize the expected square of the centered learning signal
. Though this approach to fitting the baseline does not result in the maximal variance reduction, it is simpler and in our experience works as well as the optimal approach of Weaver & Tao (2001) which requires taking into account the magnitude of the gradient of the inference network parameters. We also experimented with perparameter baselines but found that they did not improve on the global ones. Finally, we note that incorporating baselines into the learning signal can be seen as using simple control variates. In contrast to the more elaborate control variates (e.g. of Paisley et al. (2012)), baselines do not depend on the form of the model or of the variational distribution and thus are easier to use.Even after centering, using
as the learning signal is nontrivial as its average magnitude can change dramatically, and not necessarily monotonically, as training progresses. This variability makes training an inference network using a fixed learning rate difficult. We address this issue by dividing the centered learning signal by a running estimate of its standard deviation. This normalization ensures that the signal is approximately unit variance, and can be seen as a simple and efficient way of adapting the learning rate. To ensure that we stop learning when the magnitude of the signal approaches zero, we apply variance normalization only when the estimate of the standard deviation is greater than 1. The algorithm for computing NVIL parameter updates using the variance reduction techniques described so far is provided in the supplementary material.
So far we made no assumptions about the structure of the model or the inference network. However, by taking advantage of their conditional independence properties we can train the inference network using simpler and less noisy local learning signals instead of the monolithic global learning signal . Our approach to deriving a local signal for a set of parameters involves removing all the terms from the global signal that do not affect the value of the resulting gradient estimator.
We will derive the layerspecific learning signals for the common case of both the model and the inference network having layers of latent variables. The model and the variational posterior distributions then naturally factor as
(10)  
(11) 
where denotes the latent variables in the layer and the parameters of the variational distribution for that layer. We will also use to denote the latent variables in layers through .
To learn the parameters of the the variational distribution for layer , we need to compute the following gradient:
Using the law of iterated expectation we can rewrite the expectation w.r.t. as
where we also used the fact that under the variational posterior, is independent of and , given . As a consequence of Eq. 2.3.1, when computing the expectation w.r.t. , all the terms in the learning signal that do not depend on can be safely dropped without affecting the result. This gives us the following local learning signal for layer :
(12) 
To get the signal for the first hidden layer we simply use in place of , in which case we simply recover the global learning signal. For hidden layers , however, the local signal involves fewer terms than and thus can be expected to be less noisy. As we do not assume any withinlayer structure, Eq. 12 applies to models and inference networks whether or not and are factorial.
Since local signals can be significantly different from each other, we use separate baselines and variance estimates for each signal. For layers , the inputdependent baseline is replaced by .
In some cases, further simplification of the learning signal is possible, yielding a different signal per latent variable. We leave exploring this as future work.
The idea of training an approximate inference network by optimizing a variational lower bound is not new. It goes back at least to Hinton & Zemel (1994)
, who derived the variational objective from the Minimum Description Length (MDL) perspective and used it to train linear autoencoders. Their probabilistic encoder and decoder correspond to our inference network and model respectively. However, they computed the gradients analytically, which was possible due to the simplicity of their model, and dismissed the samplingbased approach as infeasible due to noise.
Salakhutdinov & Larochelle (2010)
proposed using a feedforward “recognition” model to perform efficient inputdependent initialization for the mean field inference algorithm in deep Boltzmann machines. As the recognition model is trained to match the marginal probabilities produced by mean field inference, it inherits the limitations of the inference procedure, such as the inability to model structured posteriors. In contrast, in NVIL the inference net is trained to match the true posterior directly, without involving an approximate inference algorithm, and thus the accuracy of the fit is limited only by the expressiveness of the inference network itself.
Recently a method for training nonlinear models with continuous latent variables, called Stochastic Gradient Variational Bayes (SGVB), has been proposed by Kingma & Welling (2013) and Rezende et al. (2014)
. Like NVIL, it involves using feedforward models to perform approximate inference and trains them by optimizing a samplingbased estimate of the variational bound on the loglikelihood. However, SGVB is considerably less general than NVIL, because it uses a gradient estimator obtained by taking advantage of special properties of realvalued random variables and thus is not applicable to models with discrete random variables. Moreover, unlike NVIL, SGVB method cannot handle inference networks with nonlinear dependencies between latent variables. The ideas of the two methods are complementary however, and NVIL is likely to benefit from the SGVBstyle treatment of continuousvalued variables, while SGVB might converge faster using the variance reduction techniques we proposed.
Gregor et al. (2013)
have recently proposed a related algorithm for training sigmoid belief network like models based on the MDL framework. They also use a feedforward model to perform approximate inference, but concentrate on the case of a deterministic inference network and can handle only binary latent variables. The inference network is trained by backpropagating through binary thresholding units, ignoring the thresholding nonlinearities, to approximately minimize the coding cost of the joint latentvisible configurations. This approach can be seen as approximately maximizing a looser variational lower bound than (
2.1) due to the absence of the entropy term.An inference network for efficient generation of samples from the approximate posterior can also be seen as a probabilistic generalization of the approximate feedforward inference methods developed for sparse coding models in the last few years (Kavukcuoglu et al., 2008; Bradley & Bagnell, 2008; Gregor & LeCun, 2010).
Like NVIL, Black Box Variational Inference (BBVI, Ranganath et al., 2013) learns the variational parameters of the posterior by optimizing the variational bound using samplingbased gradient estimates, which makes it applicable to a large range of models. However, unlike NVIL, BBVI follows the traditional approach of learning a separate set of variational parameters for each observation and does not use an inference network. Moreover, BBVI uses a fullyfactorized mean field approximation to the posterior, which limits its power.
NVIL shares many similarities with the wakesleep algorithm (Hinton et al., 1995), which enjoys the same scalability and applicability to a wide range of models. This algorithm was introduced for training Helmholtz machines (Dayan et al., 1995), which are multilayer belief networks augmented with recognition networks. These recognition networks are used for approximate inference and are directly analogous to NVIL inference networks. Wakesleep alternates between updating the model parameters in the wake phase and the recognition network parameters in the sleep phase. The model parameter update is based on the samples generated from the recognition network on the training data and is identical to the NVIL one (Eq. 5). However, in contrast to NVIL, the recognition network parameters are learned from samples generated by the model. In other words, the recognition network is trained to recover the hidden causes corresponding to the samples from the model distribution by following the gradient
(13) 
Unfortunately, this update does not optimize the same objective as the model parameter update, which means that the wakesleep algorithm does not optimize a welldefined objective function and is not guaranteed to converge. This is the algorithm’s main weakness, compared to NVIL, which optimizes a variational lower bound on the loglikelihood.
The wakesleep gradient for recognition network parameters does have the advantage of being much easier to estimate than the corresponding gradient of the variational bound. In fact, the idea of training the recognition networks using the gradient of the bound was mentioned in (Hinton & Zemel, 1994) and (Dayan & Hinton, 1996) but not seriously considered due concerns about the high variance of the estimates. In Section 4.2 we show that while the naive estimator of the gradient given in Eq. 2.2 does exhibit high variance, the variance reduction techniques from Section 2.3 improve it dramatically and make it practical.
Using the gradient (2.2) to train the inference network can be seen as an instance of the REINFORCE algorithm (Williams, 1992) from reinforcement learning (RL), which adapts the parameters of a stochastic model to maximize the external reward signal which depends on the model’s output. Given a model and a reward signal , REINFORCE updates the model parameters using the rule
(14) 
We can view NVIL as an application of REINFORCE on the pertrainingcase basis, with the inference network corresponding to the stochastic model, latent state to the output, and the learning signal to the reward. The term in Eq. 14, called a baseline
in the RL literature, is a hyperparameter that can be adapted to reduce the variance of the parameter update. Thus it serves the same function as
and that we subtract from the learning signal to center it in Section 2.3.1. The considerable body of work on baselines and other variance reduction methods done in the RL community (e.g. Greensmith et al., 2004) is likely to contain additional techniques relevant for training inference networks.We performed two sets of experiments, with the first set intended to evaluate the effectiveness of our variance reduction techniques and to compare NVIL’s performance to that of the wakesleep algorithm. In the second set of experiments, we demonstrate NVIL’s ability to handle larger realworld datasets by using it to train generative models of documents.
We trained all models using stochastic gradient ascent using minibatches of 20 observations sampled randomly from the training data. The gradient estimates were computed using a single sample from the inference network. For each dataset, we created a validation set by removing a random subset of 100 observations from the training set. The only form of regularization we used was early stopping based on the validation bound, implemented by keeping track of the parameter configuration with the best validation score seen so far. We implemented each inputdependent baseline using a neural network with a single hidden layer of 100 tanh units.
We used fixed learning rates because we found them to produce superior results to the annealing schedules we experimented with. The learning rates we report were selected based on the validation set performance in preliminary experiments with smaller models. We always make the learning rate for inference network five times smaller than for the model (which is the one we report), as we found this to improve performance. We used inference networks with layered structure given by Eq. 11, without dependencies within each layer except in the experiment with autoregressive inference networks. All multilayer inference networks were trained using layerspecific learning signals from Section 2.3.3.
As the models we train are intractable, we cannot compute the exact loglikelihoods for them. Instead we report the estimates of the variational bound (2.1) computed using 10 samples from the inference network, which we found to be sufficient to get the accurate bound estimates. We expect this approach to underestimate the loglikelihood considerably, but leave finding more direct and thus less pessimistic evaluation methods as future work.
Our first set of experiments was performed on the binarized version of the MNIST dataset, which has become the standard benchmark for evaluating generative models of binary data. The dataset consists of 70,000
binary images of handwritten digits, partitioned into a 60,000image training set and 10,000image test set. We used the binarization of Salakhutdinov & Murray (2008), which makes our scores directly comparable to those in the literature.We used
as the learning rate for training models with NVIL on this dataset. Centering the input vectors by subtracting the mean vector was essential for making the inference networks and inputdependent baselines work well.
To demonstrate the importance of variance reduction techniques, we trained two SBNs using a range of variance control settings. The first SBN had a single layer of 200 latent variables, while the second one had two layers of 200 variables each. Figure 1 shows the estimate of the variational objective on the validation set plotted against the number of parameter updates. For both models, it is clear that using all three techniques – the inputdependent and inputindependent baselines along with variance normalization – is essential for best performance. However, of the three techniques, the inputdependent baseline appears to be the least important. Comparing the plots for the two models suggests that variance reduction becomes more important for larger models, with the gap between the best combination and the others (excluding the very worst one) widening. For both models, learning with all three variance reduction techniques disabled makes barely any progress and is clearly infeasible.
We found that disabling layerspecific learning signals had little effect on the performance of the resulting model. The difference was about 0.4 nats for an SBN with two or three layers of latent variables.
Model  Dim  Test NLL  

NVIL  WS  
SBN  200  113.1  120.8 
SBN  500  112.8  121.4 
SBN  200200  99.8  107.7 
SBN  200200200  96.7  102.2 
SBN  200200500  97.0  102.3 
fDARN  200  92.5  95.9 
fDARN  500  90.7  97.2 
fDARN  400  96.3  
DARN  400  93.0  
NADE  500  88.9  
RBM (CD3)  500  105.5  
RBM (CD25)  500  86.3  
MoB  500  137.6 
We next compared NVIL to the wakesleep algorithm, which is its closest competitor in terms of scalability and breadth of applicability, by training a range of models using both algorithms. Wakesleep training used a learning rate of , as we found this algorithm to be more sensitive to the choice of the learning rate than NVIL, performing considerably better with lower learning rates. The results, along with some baselines from the literature, are shown in Table 1. We report only the means of the bound estimates as their standard deviations were all very small, none exceeding 0.1 nat. We can see that models trained with NVIL have considerably better bounds on the loglikelihood, compared to their wakesleep counterparts, with the difference ranging from 3.4 to 8.6 nats. Additional layers make SBNs perform better, independently of the training method. Interestingly, singlelayer fDARN (Gregor et al., 2013)
models, which have autoregressive connections between the latent variables, perform better than any of the SBN models trained using the same algorithm. Comparing to results from the literature, we see that all the SBN and fDARN models we trained perform much better than a mixture of 500 factorial Bernoulli distributions (MoB) but not as well as the deterministic Neural Autoregressive Distribution Estimator (NADE)
(Larochelle & Murray, 2011). The NVILtrained fDARN models with 200 and 500 latent variables also outperform the fDARN (as well as the more expressive DARN) model with 400 latent variables from (Gregor et al., 2013), which were trained using an MDLbased algorithm. The fDARN and multilayer SBN models trained using NVIL also outperform a 500hiddenunit RBM trained with 3step contrastive divergence (CD), but not the one trained with 25step CD
(Salakhutdinov & Murray, 2008). However, both sampling and CD25 training in an RBM is considerably more expensive than sampling or NVIL training for any of our models.The samplingbased approach to computing gradients allows NVIL to handle variational posteriors with complex dependencies. To demonstrate this ability, we retrained several of the SBN models using inference networks with autoregressive connections within each layer. These networks can capture the dependencies between variables within layers and thus are considerably more expressive than the ones with factorial layers. Results in Table 2 indicate that using inference networks with autoregressive connections produces better models, with the singlelayer models exhibiting large gains.
Model  Dim  Test NLL  

Autoreg  Factorial  
SBN  200  103.8  113.1 
SBN  500  104.4  112.8 
SBN  200200200  94.5  96.7 
SBN  200200500  96.0  97.0 
We also applied NVIL to the more practical task of document modelling. The goal is to train a generative model of documents which are represented as vectors of word counts, also known as bags of words. We trained two simple models on the 20 Newsgroups and Reuters Corpus Volume I (RCV1v2) datasets, which have been used to evaluate similar models in (Salakhutdinov & Hinton, 2009b; Larochelle & Lauly, 2012). 20 Newsgroups is a fairly small dataset of Usenet newsgroup posts, consisting of about 11K training and 7.5K test documents. RCV1 is a much larger dataset of Reuters newswire articles, with about 794.4K training and 10K test documents. We use the standard preprocessed versions of the datasets from Salakhutdinov & Hinton (2009b), which have vocabularies of 2K and 10K words respectively.
We experimented with two simple document models, based on the SBN and DARN architectures. Both models had a single layer of latent variables and a multinomial visible layer and can be seen as directed counterparts of the Replicated Softmax model (Salakhutdinov & Hinton, 2009b). We used the same training procedure as on MNIST with the exception of the learning rates which were on 20 Newsgroups and on RCV1.
The established evaluation metric for such models is the perplexity per word, computed as
, where is the number of documents, is the length of document , and the probability of the document under the model. As we cannot compute , we use the variational lower bound in its place and thus report an upper bound on perplexity.The results for our models, along with ones for the Replicated Softmax and DocNADE models from (Salakhutdinov & Hinton, 2009b) and (Larochelle & Lauly, 2012) respectively, are shown in Table 3. We can see that the SBN and fDARN models with 50 latent variables perform well, producing better scores than LDA and Replicated Softmax on both datasets. Their performance is also competitive with that of DocNADE on 20 Newsgroups. The score of 724 for fDARN with 50 latent variables on RCV1 is already better than DocNADE’s 742, the best published result on that dataset. fDARN with 200 hidden units, however, performs even better, setting a new record with 598.
Model  Dim  20 News  Reuters 

SBN  50  909  784 
fDARN  50  917  724 
fDARN  200  598  
LDA  50  1091  1437 
LDA  200  1058  1142 
RepSoftMax  50  953  988 
DocNade  50  896  742 

We developed, NVIL, a new training method for intractable directed latent variable models which is general and easy to apply to new models. We showed that NVIL consistently outperforms the wakesleep algorithm at training sigmoidbeliefnetworklike models. Finally, we demonstrated the potential of our approach by achieving stateoftheart results on a sizable dataset of documents (Reuters RCV1).
As the emphasis of this paper is on the training method, we applied it to some of the simplest possible model and inference network architectures, which was sufficient to obtain promising results. We believe that considerable performance gains can be made by using more expressive architectures, such as those with nonlinearities between layers of stochastic variables. Applying NVIL to models with continuous latent variables is another promising direction since binary latent variables are not always appropriate.
We expect NVIL to be also applicable to training conditional latent variable models for modelling the distribution of observations given some context, which would require making the inference network take both the context and the observation as input. This would make it an alternative to the importancesampling training method of Tang & Salakhutdinov (2013) for conditional models with structured highdimensional outputs.
We hope that the generality and flexibility of our approach will make it easier to apply powerful directed latent variable models to realworld problems.
We thank Koray Kavukcuoglu, Volodymyr Mnih, and Nicolas Heess for their helpful comments. We thank Ruslan Salakhutdinov for providing us with the preprocessed document datasets.
Journal of Machine Learning Research
, 5:1471–1530, 2004.Variational bayesian inference with stochastic search.
In ICML, 2012.Algorithm 1 provides an outline of our implementation of NVIL gradient computation for a minibatch of randomly chosen training cases. The exponential smoothing factor used for updating the estimates of the mean and variance of the inference network learning signal was set to 0.8 in our experiments.
Differentiating the variational lower bound w.r.t. to the inference network parameters gives
where we used the fact that . Using the identity , then gives
Comments
There are no comments yet.