Variational inference approximates the posterior distribution in probabilistic models. Given observed variables we would like to know the underlying phenomenon , defined probabilistically as . Variational inference approximates through a simpler distribution . The approximation is defined through a distance/divergence, often the Kullback-Leibler divergence:
It is interesting to see that this deterministic strategy does not require Monte Carlo updates. It can be seen as a deterministic optimization problem. However, it is definitely possible to solve this deterministic problem stochastically as well! We can formulate it as a stochastic optimization problem!
There are two main strategies:
- the reparametrization trick
- the log-derivate trick
The log-derivate trick is quite general but still suffers from high variance. Henceforth, so-called control variates have been introduced that reduce variance. We will spend quite a bit of time to clarify what a control variate is. The last section describes modern approaches that combine features from both strategies.
The reparametrization trick
The reparametrization trick introduces auxiliary random variables that are stochastic such that the parameters to be optimized over are only occuring in deterministic functions. This is convenient because it can reduce variance and sometimes the derivatives of the probability density functions do not exist in closed-form (which means no autodifferentation). See the Inference in deep learning post.
The log-derivative trick
The log-derivative trick is also called the score function method, REINFORCE, or black-box variational inference. The term black-box variational inference reveals that this trick is completely general. It can be applied to any model. For instance, models that have both continuous and discrete latent variables. The joint distribution does not need to be differentiable either.
It uses the following identity:
This identity is just obtained by differentiating using and applying the chain rule . Let’s subsequently rewrite this identity as a product:
The expected costs we want to minimize:
We can use Leibniz’s integral rule (differentiation under the integral sign) to shift the differential operator into the integral. To recall the rule:
In our case:
Using the log identity:
Now we can use Monte Carlo to estimate:
Here $x_s \sim p_\phi(x)$ i.i.d. This is general estimator: $f_\theta(x)$ does not need to be differentiable or continuous with respect to $x$. Note that $\log p_\phi(x)$ needs to be differentiable with respect to $\phi$.
We should show that the variance is actually reduced… However, let us first explain something that you will find time after time. Namely the notion of control variates…
Let us estimate the expectation over a function given a function . The Monte Carlo estimator is of the form with . We can introduce a control variate to reduce the variance:
The parameter can be chosen to minimize the variance, which turns out to be optimally:
More information can be found at Wikipedia. The final variance will be something along the lines:
Here and . So, how we can explain this best?
Assume we have to sum over the function with , then if we sample uniformly random values between and we will have results between and . We would like to transform this function in such way that these results are closer to each other. The values at should be going to the mean, and the values at as well. At wikipedia they give the example of the covariate (this could have just been ). By adding and subtracting the average (in this case ) we make the function flatter with picking , in other words we reduced the variance. We sample 100 values uniformly and demonstate in the following graph that the function using the covariate is indeed flatter.
Another covariate could be . We then have to subtract the expectation of that function, namely . This function is even flatter and has an even smaller variance. You can see that in the graph above. We have picked a value for . The covariate which would make the compound function completely flat would be , which is mirrored over the range from . However, this would of course render the Monte Carlo sampling redundant, because we would need the expectation over which is in this case just as hard as that over .
Recent approaches (and combinations)
The log-derivative trick (or the score function estimator) still suffers from high variance. Common techniques to reduce variance is by introducing baselines. Examples of unbiased single sample gradient estimators, are NVIL (Mnih and Gregor, 2014) and MuProp (Gu et al., 2015). An example of an unbiased multisample case is VIMCO (Mnih and Rezende, 2016).
Examples of biased single sample gradient estimators, are Gumbel-Softmax (Jang et al., 2016) and Concrete relaxiations (Maddison et al., 2017), independent researchers coming to the same strategy. The family of concrete distributions (Maddison et al, 2017) has closed-form densities and a simple reparametrization. The concrete distributions can replace discrete distributions on training so all gradients can properly be calculated. During training the concrete distributions can be replaced by discrete distributions.
REBAR (Tucker et al., 2017) is a new approach that uses a novel control variate to make the Concrete relaxation approach unbiased again.
- Reparametrization Trick (Huang, 2018, blog post)
- The Generalized Reparameterization Gradient (Ruiz et al., 2016)
- Local Expectation Gradients for Doubly Stochastic Variational Inference (Titsias, 2015)
- Neural Variational Inference and Learning in Belief Networks (Mnih, Gregor, 2014)
- MuProp: Unbiased Backpropagation for Stochastic Neural Networks, (Gu et al, 2016)
- Variational Inference for Monte Carlo Objectives (Mnih, Rezende, 2016)
- Categorical Reparameterization with Gumbel-Softmax (Jang et al., 2016)
- The Concrete Distribution: A Continuous Relaxation of Discrete Random Variables (Maddison et al., 2017)
- REBAR: Low-Variance, Unbiased Gradient Estimates for Discrete Latent Variable Models (Tucker et al., 2017)