Perturbative Black Box Variational Inference

09/21/2017 ∙ by Robert Bamler, et al. ∙ 0

Black box variational inference (BBVI) with reparameterization gradients triggered the exploration of divergence measures other than the Kullback-Leibler (KL) divergence, such as alpha divergences. These divergences can be tuned to be more mass-covering (preventing overfitting in complex models), but are also often harder to optimize using Monte-Carlo gradients. In this paper, we view BBVI with generalized divergences as a form of biased importance sampling. The choice of divergence determines a bias-variance tradeoff between the tightness of the bound (low bias) and the variance of its gradient estimators. Drawing on variational perturbation theory of statistical physics, we use these insights to construct a new variational bound which is tighter than the KL bound and more mass covering. Compared to alpha-divergences, its reparameterization gradients have a lower variance. We show in several experiments on Gaussian Processes and Variational Autoencoders that the resulting posterior covariances are closer to the true posterior and lead to higher likelihoods on held-out data.

READ FULL TEXT VIEW PDF
POST COMMENT

Comments

There are no comments yet.

Authors

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

Variational inference (VI) (Jordan et al., 1999)

provides a way to convert Bayesian inference to optimization by minimizing a divergence measure. Recent advances of VI have been devoted to scalability 

(Hoffman et al., 2013; Ranganath et al., 2014), divergence measures (Minka, 2005; Li and Turner, 2016; Hernandez-Lobato et al., 2016), and structured variational distributions (Hoffman and Blei, 2015; Ranganath et al., 2016).

While traditional stochastic variational inference (SVI) (Hoffman et al., 2013) was limited to conditionally conjugate Bayesian models, black box variational inference (BBVI) (Ranganath et al., 2014) enables SVI on a large class of models. It expresses the gradient as an expectation, and estimates it by Monte-Carlo sampling. A variant of BBVI uses reparameterized gradients and has lower variance (Salimans and Knowles, 2013; Kingma and Welling, 2014; Rezende et al., 2014; Ruiz et al., 2016). BBVI paved the way for approximate inference in complex and deep generative models (Kingma and Welling, 2014; Rezende et al., 2014; Ranganath et al., 2015; Bamler and Mandt, 2017).

Before the advent of BBVI, divergence measures other than the KL divergence had been of limited practical use due to their complexity in both mathematical derivation and computation (Minka, 2005), but have since then been revisited. Alpha-divergences (Hernandez-Lobato et al., 2016; Dieng et al., 2017; Li and Turner, 2016) achieve a better matching of the variational distribution to different regions of the posterior and may be tuned to either fit its dominant mode or to cover its entire support. The problem with reparameterizing the gradient of the alpha-divergence is, however, that the resulting gradient estimates have large variances. It is therefore desirable to find other divergence measures with low-variance reparameterization gradients.

In this paper, we use concepts from perturbation theory of statistical physics to propose a new family of variational bounds on the marginal likelihood with low-variance reparameterization gradients. The lower bounds are enumerated by an order , which takes odd integer values, and are given by

(1)

Here,

denotes the joint probability density function of the model with observations

and latent variables , is the variational distribution, which depends on variational parameters , and is a reference point for the perturbative expansion, see below. In this paper, we motivate and discuss Eq. 1 (Section 3), and we analyze the properties of the proposed bound experimentally (Section 4). Our contributions are as follows.

  • We establish a view on black box variational inference with generalized divergences as a form of biased importance sampling (Section 3.1). The choice of divergence allows us to trade-off between a low-variance stochastic gradient and loose bound, and a tight variational bound with higher-variance Monte-Carlo gradients. As we explain below, importance sampling and point estimation are at opposite ends of this spectrum.

  • We combine these insights with ideas from perturbation theory of statistical physics to motivate the objective function in Eq. 1 (Section 3.2). We show that, for all odd , is a nontrivial lower bound on the marginal likelihood . Thus, we propose the perturbative black box variational inference algorithm (PBBVI), which maximizes over and

    with stochastic gradient descent (SGD). For

    , our algorithm is equivalent to standard BBVI with the KL-divergence (KLVI). On the variance-bias spectrum, KLVI is on the side of large bias and low gradient variance. Increasing to larger odd integers allows us to gradually trade in some increase in the gradient variance for some reduction of the bias.

  • We evaluate our PBBVI algorithm experimentally for the lowest nonstandard order (Section 4). Compared to KLVI (), our algorithm fits variational distributions that cover more of the mass of the true posterior. Compared to alpha-VI, our experiments confirm that PBBVI uses gradient estimates with lower variance, and converges faster.

2 Related work

Our approach is related to BBVI, VI with generalized divergences, and variational perturbation theory. We thus briefly discuss related work in these three directions.

Black box variational inference (BBVI).

BBVI has already been addressed in the introduction (Salimans and Knowles, 2013; Kingma and Welling, 2014; Rezende et al., 2014; Ranganath et al., 2014; Ruiz et al., 2016); it enables variational inference for many models. Our work builds upon BBVI in that BBVI makes a large class of new divergence measures between the posterior and the approximating distribution tractable. Depending on the divergence measure, BBVI may suffer from high-variance stochastic gradients. This is a practical problem that we aim to improve in this paper.

Generalized divergences measures.

Our work connects to generalized information-theoretic divergences (Amari, 2012). Minka (2005) introduced a broad class of divergences for variational inference, including alpha-divergences. Most of these divergences have been intractable in large-scale applications until the advent of BBVI. In this context, alpha-divergences were first suggested by  Hernandez-Lobato et al. (2016) for local divergence minimization, and later for global minimization by Li and Turner (2016) and Dieng et al. (2017). As we show in this paper, alpha-divergences have the disadvantage of inducing high-variance gradients, since the ratio between posterior and variational distribution enters the bound polynomially instead of logarithmically. In contrast, our approach leads to a more stable inference scheme in high dimensions.

Variational perturbation theory.

Perturbation theory refers to methods that aim to truncate a typically divergent power series to a convergent series. In machine learning, these approaches have been addressed from an information-theoretic perspective by 

Tanaka (1999, 2000). Thouless-Anderson-Palmer (TAP) equations (Thouless et al., 1977) are a form of second-order perturbation theory. They were originally developed in statistical physics to include perturbative corrections to the mean-field solution of Ising models. They have been adopted into Bayesian inference in (Plefka, 1982) and were advanced by many authors (Kappen and Wiegerinck, 2001; Paquet et al., 2009; Opper et al., 2013; Opper, 2015). In variational inference, perturbation theory yields extra terms to the mean-field variational objective which are difficult to calculate analytically. This may be a reason why the methods discussed are not widely adopted by practitioners. In this paper, we emphasize the ease of including perturbative corrections in a black box variational inference framework. Furthermore, in contrast to earlier formulations, our approach yields a strict lower bound to the marginal likelihood which can be conveniently optimized. Our approach is different from the traditional variational perturbation formulation (Kleinert, 2009), which generally does not result in a bound.

3 Method

In this section, we present our main contributions. We first present our view of black box variational inference (BBVI) as a form of biased importance sampling in Section 3.1. With this view, we bridge the gap between variational inference and importance sampling. In Section 3.2, we introduce our family of new variational bounds, and analyze their properties further in Section 3.3.

Figure 1: Different choices for in Eq. 4. KLVI corresponds to (red), and importance sampling to (black). Our proposed PBBVI bound uses (green, Eq. 7), which lies between KLVI and importance sampling (we set and for PBBVI here). Figure 2: Behavior of different VI methods on fitting a univariate Gaussian to a bimodal target distribution (black). PBBVI (proposed, green) covers more of the mass of the entire distribution than the traditional KLVI (red). Alpha-VI is mode seeking for large and mass covering for smaller . Figure 3: Sampling variance of the stochastic gradient (averaged over its components) in the optimum, for alpha-divergences (orange, purple, gray), and the proposed PBBVI (green). The variance grows exponentially with the latent dimension for alpha-VI, and only algebraically for PBBVI.

3.1 Black Box Variational Inference as Biased Importance Sampling

Consider a probabilistic model with data , latent variables

, and joint distribution

. We are interested in the posterior distribution over the latent variables, . This involves the intractable marginal likelihood . In variational inference (Jordan et al., 1999), we instead minimize a divergence measure between a variational distribution and the posterior. Here, are parameters of the variational distribution, and we aim to find the parameters that minimize the distance to the posterior. This is equivalent to maximizing a lower bound on the marginal likelihood.

We call the difference between the log variational distribution and the log joint distribution the interaction energy,

(2)

We use or interchangeably to denote , and to denote , when more convenient. Using this notation, the marginal likelihood is

(3)

We call the importance ratio, since sampling from to estimate the right-hand side of Eq. 3 is equivalent to importance sampling. As importance sampling is inefficient in high dimensions, we resort to variational inference. To this end, let be any concave function defined on the positive reals. We assume furthermore that for all , we have . Applying Jensen’s inequality, we can lower-bound the marginal likelihood,

(4)

Figure 1 shows exemplary choices of . We maximize using reparameterization gradients, where the bound is not computed analytically, but rather its gradients are estimated by sampling from  (Kingma and Welling, 2014). This leads to a stochastic gradient descent scheme, where the noise is a result of the Monte-Carlo estimation of the gradients.

Our approach builds on the insight that black box variational inference is a type of biased importance sampling, where we estimate a lower bound of the marginal likelihood by sampling from a proposal distribution, iteratively improving this distribution. The approach is biased since we do not estimate the exact marginal likelihood but only a lower bound to this quantity. As we argue below, the introduced bias allows us to estimate the bound more easily, because we decrease the variance of this estimator. The choice of the function thereby trades-off between bias and variance in the following way:

  • For being the identity, we obtain importance sampling. (See the black line in Figure 1). In this case, Eq. 4 does not depend on the variational parameters, so there is nothing to optimize and we can directly sample from any proposal distribution . Since the expectation under of the importance ratio gives the exact marginal likelihood, there is no bias. If the model has a large number of latent variables, the importance ratio becomes tightly peaked around the minimum of the interaction energy , resulting in a very high variance of this estimator. Importance sampling is therefore on one extreme end of the bias-variance spectrum.

  • For , we obtain the familiar Kullback-Leibler (KL) bound. (See the pink line in Figure 1; here we add a constant of for comparison, which does not affect the optimization). Since , the bound is

    (5)

    The Monte-Carlo expectation of has a much smaller variance than , implying efficient learning (Bottou, 2010). However, by replacing with we introduce a bias. We can further trade-off less variance for even more bias by dropping the entropy term on the right-hand side of Eq. 5. A flexible enough variational distribution will shrink to zero variance, which completely eliminates the sampling noise. This is equivalent to point-estimation, and is at the opposite end of the bias-variance spectrum.

  • Now, consider any which is between the logarithm and the identity, e.g., the green line in Figure 1 (this is the regularizing function we propose in Section 3.2 below). The more similar is to the identity, the less biased is our estimate of the marginal likelihood, but the larger the variance. Conversely, the more behaves like the logarithm, the easier it is to estimate by sampling, while at the same time the bias grows.

One example of alternative divergences to the KL divergence that have been discussed in the literature are alpha-divergences (Minka, 2005; Hernandez-Lobato et al., 2016; Li and Turner, 2016; Dieng et al., 2017). Up to a constant, they correspond to the following choice of :

(6)

The real parameter determines the distance to the importance sampling case (). As approaches from below, this bound leads to a better-behaved estimation problem of the Monte-Carlo gradient. However, unless taking the limit of (where the objective becomes the KL-bound), still enters exponentially in the bound. As we show, this leads to a high variance of the gradient estimator in high dimensions (see Figure 3 discussed below). The alpha-divergence bound is therefore similarly as hard to estimate as the marginal likelihood in importance sampling.

Our analysis relies on the observation that expectations of exponentials in are difficult to estimate, and expectations of polynomials in are easy to estimate. We derive a family of new variational bounds which are polynomials in , where increasing the order of the polynomial reduces the bias.

3.2 Perturbative Black Box Variational Inference

Perturbative bounds.

We now motivate the family of lower bounds proposed in Eq. 1 in the introduction based on the considerations outlined above. For fixed odd integer and fixed real value , the bound is of the form of Eq. 4 with the following regularizing function :

(7)

Here, the second (equivalent) formulation makes it explicit that is the th order Taylor expansion of its argument in around some reference energy . Figure 1 shows for (red) and (green). The curves are concave and lie below the identity, touching it at . We show in Section 3.3 that these properties extend to every odd and every . Therefore, is indeed a lower bound on the marginal likelihood.

The rationale for the design of the regularizing function in Eq. 7 is as follows. On the one hand, the gradients of the resulting bound should be easy to estimate via the reparameterization approach. We achieve low-variance gradient estimates by making a polynomial in , i.e., in contrast to the alpha-bound, never appears in the exponent.

On the other hand, the regularizing function should be close to the identity function so that the resulting bound has low bias. For , we have . Maximizing over is independent of the value of and equivalent to maximizing the standard KL bound , see Eq. 5, which has low gradient variance and large bias. Increasing the order to larger odd integers makes the Taylor expansion tighter, leading to a bound with lower bias. In fact, in the limit , the right-hand side of Eq. 7 is the series representation of the exponential function, and thus converges pointwise to the identity. In practice, we propose to set to a small odd integer larger than . Increasing

further reduces the bias, but it comes at the cost of increasing the gradient variance because the random variable

appears in higher orders under the expectation in Eq. 4.

As discussed in Section 3.1, the KL bound can be derived from a regularizing function that does not depend on any further parameters like . The derivation of the KL bound therefore does not require the first inequality in Eq. 4, and one directly obtains a bound on the model evidence from the second inequality alone. For , the bound depends nontrivially on , and we have to employ the first inequality in Eq. 4 in order to make the bounded quantity on the left-hand side independent of . This expenses some tightness of the bound but makes the method more flexible by allowing us to optimize over as well, as we describe next.

Input: joint probability ; order of perturbation (odd integer); learning rate schedule ; number of Monte Carlo samples ; number of training iterations ; variational family that allows for reparameterization gradients, i.e., where with a fixed noise distribution and a differentiable reparameterization function .
Output: fitted variational parameters .
1 initialize randomly and ;
2 for  to  do
      3 draw samples from the noise distribution;
       // obtain reparameterization gradient estimates using automatic differentiation:
      4 ;
      5 ];
       // perform variable updates (see second to last paragraph of Section 3.2):
      6 ;
      7 ;
      
end for
Algorithm 1 Perturbative Black Box Variational Inference (PBBVI)

Optimization algorithm.

We now propose the perturbative black box variational inference (PBBVI) algorithm. Since is a lower bound on the marginal likelihood for all and all , we can find the values and for which the bound is tightest by maximizing simultaneously over and . Algorithm 1 summarizes the PBBVI algorithm. We minimize using stochastic gradient descent (SGD) with reparameterization gradients and a learning rate that decreases with the training iteration according to Robbins-Monro bounds (Robbins and Monro, 1951). We obtain unbiased gradient estimators (denoted by “”) using standard techniques: we replace the expectation in Eq. 1 with the empirical average over a fixed number of samples from , and we calculate the reparameterization gradients with respect to and using automatic differentiation.

In practice, we typically discard the value of once the optimization is converged since we are only interested in the fitted variational parameters . However, during the optimization process, is an important auxiliary quantity and the inference algorithm would be inconsistent without an optimization over : if we were to extend the model by an additional observed variable which is statistically independent of the latent variables , then the log joint (as a function of alone) changes by a constant positive prefactor. The posterior remains unchanged by the constant prefactor, and a consistent VI algorithm must therefore produce the same approximate posterior distribution for both models. Optimizing over ensures this consistency since the log joint appears in the lower bound only in the combination . Therefore, a rescaling of the log joint by a constant positive prefactor can be completely absorbed by a change in the reference energy .

We observed in our experiments that the reference energy can become very large (in absolute value) for models with many latent variables. To avoid numerical overflow or underflow from the prefactor , we consider the surrogate objective . The gradients with respect to of and are equal up to a positive prefactor, so we can replace the former with the latter in the update step (line 1 in Algorithm 1). The gradient with respect to is (line 1). Using the surrogate avoids numerical underflow or overflow, as well as exponentially increasing or decreasing gradients.

Mass covering effect.

In Figure 2

, we fit a Gaussian distribution to a one-dimensional bimodal target distribution (black line), using different divergences. Compared to BBVI with the standard KL divergence (KLVI, pink line), alpha-divergences are more mode-seeking (purple line) for large values of

, and more mass-covering (orange line) for small  (Li and Turner, 2016). Our PBBVI bound (, green line) achieves a similar mass-covering effect as in alpha-divergences, but with associated low-variance reparameterization gradients. This is seen in Figure 3, discussed in Section 4.2, which compares the gradient variances of alpha-VI and PBBVI as a function of dimensions.

3.3 Proof of Correctness and Nontriviality of the Bound

To conclude the presentation of the PBBVI algorithm, we prove that the objective in Eq. 1 is indeed a lower bound on the marginal likelihood for all odd orders , and that the bound is nontrivial.

Correctness.

The lower bound results from inserting the regularizing function from Eq. 7 into Eq. 4. For odd , it is indeed a valid lower bound because is concave and lies below the identity. To see this, note that the second derivative is non-positive everywhere for odd . Therefore, the function is concave. Next, consider the function , which has a stationary point at . Since is also concave, is a global maximum, and thus for all , implying that . Thus, for odd , the function satisfies all requirements for Eq. 4, and is a lower bound on the marginal likelihood. Note that an even order does not lead to a valid concave regularizing function.

Nontriviality.

Since the marginal likelihood is always positive, a lower bound would be trivial if it was negative. We show that once the optimization algorithm has converged, the bound at the optimum is always positive. At the optimum, all gradients vanish. By setting the derivative with respect to of the right-hand side of Eq. 1 to zero we find that , where is the variational distribution at the optimum. Thus, the lower bound at the optimum is with , where the sum runs only to because the term with vanishes at . We show that is positive for all . If , then is a positive constant. For , is a polynomial in of even order , whose highest order term has a positive coefficient . Therefore, as , the function goes to positive infinity and it thus has a global minimum at some value . At the global minimum, its derivative vanishes, . Thus, at the global minimum of the polynomial , all terms except the highest order term cancel, and we find , which is nonnegative because is even. The case is achieved if and only if , but this would violate the condition . Therefore, is strictly positive, and since is a global minimum of , we have for all . Inserting into the expression for concludes the proof that the lower bound at the optimum is positive.

4 Experiments

We evaluate PBBVI with different models. First we investigate its behavior in a controlled setup of Gaussian processes on synthetic data (Section 4.1

). We then evaluate PBBVI based on a classification task using Gaussian processes classifiers, where we use data from the UCI machine learning repository (Section

4.2). This is a Bayesian non-conjugate setup where black box inference is required. Finally, we use an experiment with the variational autoencoder (VAE) to explore our approach on a deep generative model (Section 4.3). This experiment is carried out on MNIST data. We use the perturbative order for all experiments with PBBVI. This corresponds to the lowest order beyond standard KLVI, since KLVI is equivalent to PBBVI with , and has to be an odd integer. Across all the experiments, PBBVI demonstrates advantages based on different metrics.

4.1 GP Regression on Synthetic Data

(a) KLVI
(b) PBBVI with
Figure 4:

Gaussian process regression on synthetic data (green dots). Three standard deviations are shown in varying shades of orange. The blue dashed lines show three standard deviations of the true posterior. The red dashed lines show the inferred three standard deviations using KLVI (a) and PBBVI (b). We see that the results from our proposed PBBVI are close to the analytic solution while traditional KLVI underestimates the variances.

Method Avg variances
Analytic 0.0415
KLVI 0.0176
PBBVI 0.0355
Table 1: Average variances across training examples in the synthetic data experiment. The closer to the analytic solution, the better.
Data set Crab Pima Heart Sonar
KLVI 0.22 0.245 0.148 0.212
PBBVI 0.11 0.240 0.1333 0.1731
Table 2: Error rate of GP classification on the test set. The lower the better. Our proposed PBBVI consistently obtains better classification results.

In this section, we inspect the inference behavior using a synthetic data set with Gaussian processes (GP). We generate the data according to a Gaussian noise distribution centered around a mixture of sinusoids, and sample 50 data points (green dots in Figure 4). We then use a GP to model the data, thus assuming the generative process and .

We first compute an analytic solution of the posterior of the GP, (three standard deviations shown in blue dashed lines) and compare it to approximate posteriors obtained by KLVI (Figure 4 (a)) and the proposed PBBVI (Figure 4 (b)). The results from PBBVI are almost identical to the analytic solution. In contrast, KLVI underestimates the posterior variance. This is consistent with Table 2, which shows the average diagonal variances. PBBVI results are much closer to the exact posterior variances.

4.2 Gaussian Process Classification

We evaluate the performance of PBBVI and KLVI on a GP classification task. Since the model is non-conjugate, no analytical baseline is available in this case. We model the data with the following generative process:

Above, is the GP kernel,

indicates the sigmoid function, and

indicates the Bernoulli distribution. We furthermore use the Matern 32 kernel,

Data.

We use four data sets from the UCI machine learning repository, suitable for binary classification: Crab (200 datapoints), Pima (768 datapoints), Heart (270 datapoints), and Sonar (208 datapoints). We randomly split each of the data sets into two halves. One half is used for training and the other half is used for testing. We set the hyper parameters and throughout all experiments, where is the dimensionality of input .

Table 2 shows the classification performance (error rate) for these data sets. Our proposed PBBVI consistently performs better than the traditional KLVI.

Convergence speed comparison.

We also carry out a comparison in terms of speed of convergence, focusing on PBBVI and alpha-divergence VI. Our results indicate that the smaller variance of the reparameterization gradient leads to faster convergence of the optimization algorithm.

We train the GP classifier from Section 4.2 on the Sonar UCI data set using a constant learning rate. Figure 6 shows the test log-likelihood under the posterior mean as a function of training iterations. We split the data set into equally sized training, validation, and test sets. We then tune the learning rate and the number of Monte Carlo samples per gradient step to obtain optimal performance on the validation set after minimizing the alpha-divergence with a fixed budget of random samples. We use here; smaller values of lead to even slower convergence. We optimize the PBBVI lower bound using the same learning rate and number of Monte Carlo samples. The final test error rate is on an approximately balanced data set. PBBVI converges an order of magnitude faster.

Figure 3 in Section 3 provides more insight in the scaling of the gradient variance. Here, we fit GP regression models on synthetically generated data by maximizing the PBBVI lower bound and the alpha-VI lower bound with . We generate a separate synthetic data set for each by drawing random data points around a sinusoidal curve. For each , we fit a one-dimensional GP regression with PBBVI and alpha-VI, respectively, using the same data set for both methods. The variational distribution is a fully factorized Gaussian with latent variables. After convergence, we estimate the sampling variance of the gradient of each lower bound with respect to the posterior mean. We calculate the empirical variance of the gradient based on samples from , and we average over the coordinates. Figure 3 shows the average sampling variance as a function of on a logarithmic scale. The variance of the gradient of the alpha-VI bound grows exponentially in the number of latent variables. By contrast, we find only algebraic growth for PBBVI.

4.3 Variational Autoencoder

Figure 5: Test log-likelihood (normalized by the number of test points) as a function of training iterations using GP classification on the Sonar data set. PBBVI converges faster than alpha-VI even though we tuned the number of Monte Carlo samples per training step () and the constant learning rate () so as to maximize the performance of alpha-VI on a validation set.
Figure 6: Predictive likelihood of a VAE trained on different sizes of the data. The training data are randomly sampled subsets of the MNIST training set. The higher value the better. Our proposed PBBVI method outperforms KLVI mainly when the size of the training data set is small. The fewer the training data, the more advantage PBBVI obtains.

We experiment on Variational Autoencoders (VAEs), and we compare the PBBVI and the KLVI bound in terms of predictive likelihoods on held-out data (Kingma and Welling, 2014)

. Autoencoders compress unlabeled training data into low-dimensional representations by fitting it to an encoder-decoder model that maps the data to itself. These models are prone to learning the identity function when the hyperparameters are not carefully tuned, or when the network is too expressive, especially for a moderately sized training set. VAEs are designed to partially avoid this problem by estimating the uncertainty that is associated with each data point in the latent space. It is therefore important that the inference method does not underestimate posterior variances. We show that, for small data sets, training a VAE by maximizing the PBBVI lower bound leads to higher predictive likelihoods than maximizing the KLVI lower bound.

We train the VAE on the MNIST data set of handwritten digits (LeCun et al., 1998). We build on the publicly available implementation by Burda et al. (2016) and also use the same architecture and hyperparamters, with stochastic layers and samples from the variational distribution per gradient step. The model has latent units in the first stochastic layer and latent units in the second stochastic layer.

The VAE model factorizes over all data points. We train it by stochastically maximizing the sum of the PBBVI lower bounds for all data points using a minibatch size of . The VAE amortizes the gradient signal across data points by training inference networks. The inference networks express the mean and variance of the variational distribution as a function of the data point. We add an additional inference network that learns the mapping from a data point to the reference energy . Here, we use a network with four fully connected hidden layers of , , , and units, respectively.

MNIST contains

training images. To test our approach on smaller-scale data where Bayesian uncertainty matters more, we evaluate the test likelihood after training the model on randomly sampled fractions of the training set. We use the same training schedules as in the publicly available implementation, keeping the total number of training iterations independent of the size of the training set. Different to the original implementation, we shuffle the training set before each training epoch as this turns out to increase the performance for both our method and the baseline.

Figure 6 shows the predictive log-likelihood of the whole test set, where the VAE is trained on random subsets of different sizes of the training set. We use the same subset to train with PBBVI and KLVI for each training set size. PBBVI leads to a higher predictive likelihood than traditional KLVI on subsets of the data. We explain this finding with our observation that the variational distributions obtained from PBBVI capture more of the posterior variance. As the size of the training set grows—and the posterior uncertainty decreases—the performance of KLVI catches up with PBBVI.

As a potential explanation why PBBVI converges to the KLVI result for large training sets, we note that at the optimal variational distribution and reference energy (see Section 3.3). If becomes a symmetric random variable (such as a Gaussian) in the limit of a large training set, then this implies that , and PBBVI reduces to KLVI for large training sets.

5 Conclusion

We first presented a view on black box variational inference as a form of biased importance sampling, where we can trade-off bias versus variance by the choice of divergence. Bias refers to the deviation of the bound from the true marginal likelihood, and variance refers to its reparameterization gradient estimator. We then proposed a family of new variational bounds that connect to variational perturbation theory, and which include corrections to the standard Kullback-Leibler bound. Our proposed PBBVI bound converges to the true marginal likelihood for large order of the perturbative expansion, and we showed both theoretically and experimentally that it has lower-variance reparameterization gradients compared to alpha-VI. In order to scale up our method to massive data sets, future work will explore stochastic versions of PBBVI. Since the PBBVI bound contains interaction terms between all data points, breaking it up into mini-batches is non-straightforward. Besides, while our experiments used a fixed perturbative order of , it could be beneficial to increase the perturbative order at some point during the training cycle once an empirical estimate of the gradient variance drops below a certain threshold. Furthermore, the PBBVI and alpha-bounds can also be combined, such that PBBVI further approximates alpha-VI. This could lead to promising results on large data sets where traditional alpha-VI is hard to optimize due to its variance, and traditional PBBVI converges to KLVI. As a final remark, a tighter variational bound is not guaranteed to always result in a better posterior approximation since the variational family limits the quality of the solution. However, in the context of variational EM, where one performs gradient-based hyperparameter optimization on the log marginal likelihood, our bound gives more reliable results since higher orders of can be assumed to approximate the marginal likelihood better.

References

  • Amari (2012) Amari, S. (2012). Differential-geometrical methods in statistics, volume 28. Springer Science & Business Media.
  • Bamler and Mandt (2017) Bamler, R. and Mandt, S. (2017). Dynamic word embeddings. In ICML.
  • Bottou (2010) Bottou, L. (2010). Large-scale machine learning with stochastic gradient descent. In COMPSTAT. Springer.
  • Burda et al. (2016) Burda, Y., Grosse, R., and Salakhutdinov, R. (2016). Importance weighted autoencoders. In ICLR.
  • Dieng et al. (2017) Dieng, A. B., Tran, D., Ranganath, R., Paisley, J., and Blei, D. M. (2017). Variational inference via upper bound minimization. In ICML.
  • Hernandez-Lobato et al. (2016) Hernandez-Lobato, J., Li, Y., Rowland, M., Bui, T., Hernández-Lobato, D., and Turner, R. (2016). Black-box alpha divergence minimization. In ICML.
  • Hoffman and Blei (2015) Hoffman, M. and Blei, D. (2015). Stochastic Structured Variational Inference. In AISTATS.
  • Hoffman et al. (2013) Hoffman, M. D., Blei, D. M., Wang, C., and Paisley, J. W. (2013). Stochastic variational inference. JMLR, 14(1).
  • Jordan et al. (1999) Jordan, M. I., Ghahramani, Z., Jaakkola, T. S., and Saul, L. K. (1999). An introduction to variational methods for graphical models. Machine learning, 37(2).
  • Kappen and Wiegerinck (2001) Kappen, H. J. and Wiegerinck, W. (2001). Second order approximations for probability models. MIT; 1998.
  • Kingma and Welling (2014) Kingma, D. P. and Welling, M. (2014). Auto-encoding variational Bayes. In ICLR.
  • Kleinert (2009) Kleinert, H. (2009). Path integrals in quantum mechanics, statistics, polymer physics, and financial markets. World scientific.
  • LeCun et al. (1998) LeCun, Y., Bottou, L., Bengio, Y., and Haffner, P. (1998). Gradient-based learning applied to document recognition. volume 86. IEEE.
  • Li and Turner (2016) Li, Y. and Turner, R. E. (2016). Rényi divergence variational inference. In NIPS.
  • Minka (2005) Minka, T. (2005). Divergence measures and message passing. Technical report, Technical report, Microsoft Research.
  • Opper (2015) Opper, M. (2015). Expectation propagation. In Krzakala, F., Ricci-Tersenghi, F., Zdeborova, L., Zecchina, R., Tramel, E. W., and Cugliandolo, L. F., editors, Statistical Physics, Optimization, Inference, and Message-Passing Algorithms, chapter 9, pages 263–292. Oxford University Press.
  • Opper et al. (2013) Opper, M., Paquet, U., and Winther, O. (2013). Perturbative corrections for approximate inference in gaussian latent variable models. JMLR, 14(1).
  • Paquet et al. (2009) Paquet, U., Winther, O., and Opper, M. (2009). Perturbation corrections in approximate inference: Mixture modelling applications. JMLR, 10(Jun).
  • Plefka (1982) Plefka, T. (1982). Convergence condition of the TAP equation for the infinite-ranged ising spin glass model. Journal of Physics A: Mathematical and general, 15(6):1971.
  • Ranganath et al. (2014) Ranganath, R., Gerrish, S., and Blei, D. M. (2014). Black box variational inference. In AISTATS.
  • Ranganath et al. (2015) Ranganath, R., Tang, L., Charlin, L., and Blei, D. (2015). Deep exponential families. In AISTATS.
  • Ranganath et al. (2016) Ranganath, R., Tran, D., and Blei, D. (2016). Hierarchical variational models. In ICML.
  • Rezende et al. (2014) Rezende, D. J., Mohamed, S., and Wierstra, D. (2014).

    Stochastic backpropagation and approximate inference in deep generative models.

    In ICML.
  • Robbins and Monro (1951) Robbins, H. and Monro, S. (1951). A stochastic approximation method. The annals of mathematical statistics.
  • Ruiz et al. (2016) Ruiz, F., Titsias, M., and Blei, D. (2016). The generalized reparameterization gradient. In NIPS.
  • Salimans and Knowles (2013) Salimans, T. and Knowles, D. A. (2013). Fixed-form variational posterior approximation through stochastic line ar regression. Bayesian Analysis, 8(4).
  • Tanaka (1999) Tanaka, T. (1999). A theory of mean field approximation. In NIPS.
  • Tanaka (2000) Tanaka, T. (2000). Information geometry of mean-field approximation. Neural Computation, 12(8).
  • Thouless et al. (1977) Thouless, D., Anderson, P. W., and Palmer, R. G. (1977). Solution of ’solvable model of a spin glass’. Philosophical Magazine, 35(3).