T RAINING G ENERATIVE L ATENT M ODELS BY V ARIATIONAL f -D IVERGENCE M INIMIZATION
Anonymous authors
Paper under double-blind review
A BSTRACT
Probabilistic models are often trained by maximum likelihood, which corresponds to minimizing a speciﬁc form of f -divergence between the model and data distribution.
We derive an upper bound that holds for all f -divergences, showing the intuitive result that the divergence between two joint distributions is at least as great as the divergence between their corresponding marginals.
Additionally, the f -divergence is not formally deﬁned when two distributions have different supports.
We thus propose a noisy version of f -divergence which is well deﬁned in such situations.
We demonstrate how the bound and the new version of f -divergence can be readily used to train complex probabilistic generative models of data and that the ﬁtted model can depend signiﬁcantly on the particular divergence used.
I NTRODUCTION
Probabilistic modelling generally deals with the task of trying to ﬁt a model p θ ( x ) parameterized by θ to a given distribution p ( x ) .
To ﬁt the model we often wish to minimize some measure of difference between p θ ( x ) and p ( x ) .
A popular choice is the class of f -divergences 1 (see for example Sason &
Verdú (2015)) which, for two distributions p ( x ) and q ( x ) , is deﬁned by
[Equation]
(1) where f ( x ) is a convex function with f (1) = 0 .
Many of the standard divergences correspond to simple choices of the function f , see table 1.
For example, for f ( u ) =
u log u we have the "forward" KL divergence KL ( p ( x ) ||
q
( x )) ; setting f ( u )
= − log u gives the "reverse" KL divergence KL ( q ( x ) || p ( x )) .
The divergence D f ( p ( x ) || q ( x )) is zero if and only if p ( x ) =
q ( x ) .
However, for a constrained model p θ ( x ) ﬁtted to a distribution p ( x ) by minimizing D f ( p θ ( x ) || p
( x )) , the resulting optimal θ can be heavily dependent on the choice of the divergence function f (Minka, 2005).
Whilst there is signiﬁcant recent interest in using f -divergences to train complex probabilistic models (Nowozin et al., 2016; Goodfellow et al., 2014), the f -divergence is generally computationally intractable for such complex models.
The main contribution of our paper is the introduction of an upper bound on the f -divergence.
We show how this bound can be readily applied to training complex latent variable generative models based on only a modest departure from the standard Variational
Autoencoder (Kingma & Welling, 2013).
B ACKGROUND 2. M AXIMUM L IKELIHOOD
For data x 1 , . . .
, x N drawn independently and identically from some unknown distribution, ﬁtting an approximating distribution p θ ( x ) by minimizing the forward KL between p θ ( x ) and the empirical distribution ˆ p ( x ) of the data we have
[Table]
[Table]
Minimizing KL (ˆ p ( x ) || p
θ ( x ))
w.r.t.
θ is therefore equivalent to maximizing the likelihood of the data, p θ ( x ) .
Given the asymptotic guarantees of the efﬁciency of maximum likelihood (Wolfowitz, 1965)
[Table]
the forward KL is the standard divergence used in statistics and machine learning
[Table]
2. F ORWARD VERSUS REVERSE KL
It is interesting to compare models trained by the forward and reverse KL divergences.
For example, when p θ is Gaussian with parameters θ = (cid:0) µ, σ 2 (cid:1) , then minimizing the forward KL gives
[Equation]
so that the optimal setting is for µ to be the mean of p ( x ) and σ 2 the variance.
For an "under-powered" model (a model which is not rich enough to have a small divergence) p θ ( x ) and multi-modal p ( x ) this could result in p θ ( x ) placing signiﬁcant mass on low probability regions in p ( x ) .
This is the so-called "mean matching" behavior of KL ( p ( x ) ||
p
θ ( x ))
that has been suggested as a possible explanation for the poor ﬁdelity of images generated by models p θ ( x ) trained by forward KL minimization (Goodfellow, 2016).
Conversely, when using the reverse KL objective KL ( p θ ( x ) || p
( x )) , for a
Gaussian p θ ( x ) and multi-modal p ( x ) with well separated modes, optimally µ and σ 2 ﬁt one of the local modes.
This behavior is illustrated in ﬁgure 1 and is the so-called "mode matching" behavior of
KL ( p θ ( x ) || p
( x )) .
For this reason, the reverse KL objective has been suggested to be useful if high quality data samples are preferable to coverage of the dataset (Goodfellow, 2016).
This highlights the potentially signiﬁcant difference in the resulting model that is ﬁtted to the data, depending on the choice of divergence (Minka, 2005).
In this sense, it is of interest to explore ﬁtting generative models p θ ( x ) to a data distribution p ( x ) using f -divergences other than the forward KL divergence (maximum likelihood).
2. L ATENT GENERATIVE MODELS
For the model p θ ( x ) to have the power to generate complex datasets, we introduce a latent variable z , i.e. p θ ( x ) =
(cid:82) p θ ( x | z ) p ( z ) dz .
Following standard practice we do not place any parameters on the prior for the latent z , though this would be straightforward.
A standard approach to ﬁtting a latent generative model to data x 1 , . .
, x N is maximum likelihood.
[Equation] [Equation]
In all but simple cases the integral above over the latent z is intractable and a lower bound is used instead
[Equation]
where the so-called variational distribution q φ ( z | x ) is chosen such that the bound (and its gradient) is either computationally tractable or can be readily estimated by sampling (Kingma & Welling, 2013).
The parameters φ of the variational distribution q φ ( z | x ) and parameters θ of the model p θ ( x ) are jointly optimized to increase the log-likelihood
lower bound L ( θ, φ ) .
This lower bound on the likelihood corresponds to an upper bound on the forward divergence KL (ˆ p ( x ) || p
θ ( x )) , where ˆ p ( x ) is the empirical data distribution.
Our interest is to train latent generative models using a different divergence from the forward KL.
Whilst the above upper bound (5) on the forward divergence KL (ˆ p ( x ) || p θ ( x )) is well known, an upper bound on other f -divergences seems to be unfamiliar (Sason & Verdú, 2015) and we are unaware of any upper bound on general f -divergences that has been used within the machine learning community.
Recently a lower bound on the f -divergence was introduced in Nowozin et al.
(2016) by the use of the Fenchel conjugate.
The resulting training algorithm is a form of minimax in which the parameters φ that tighten the bound are adjusted so as to push up the bound towards the true divergence, whilst the model parameters θ are adjusted to lower the bound.
In Nowozin et al.
the authors were then able to relate the Generative Adversarial Network (GAN) (Goodfellow, 2016) training algorithm to the Fenchel conjugate lower bound on a corresponding f -divergence, see table 1.
In contrast, if the interest is purely on minimizing an f -divergence, it is arguably preferable to have an upper bound on the divergence since then standard optimization methods can be applied, resulting in a stable optimization procedure, see ﬁgure 2.
2. f - DIVERGENCE BETWEEN DISTRIBUTIONS WITH DISJOINT SUPPORTS
The f -divergence between two distributions, D f ( p ( x ) || p θ ( x )) , is well deﬁned when p ( x ) and p θ ( x ) have the same support.
Many datasets that we are interested in, e.g. images, are generally believed to lie on low dimension manifolds embedded in high-dimensional space (Narayanan & Mitter (2010)).
Moreover, if our model is a mixture of delta functions, then the support of p θ ( x ) is contained in a countable union of manifolds of dimension at most dim Z .
In this case, the supports of the data distribution and the model distribution are disjoint, so the f -divergence is not formally deﬁned, we refer readers to Arjovsky & Bottou (2017) for details.
To solve the problem, Sønderby et al.
(2017) proposed the instance noise trick and applied it to stabilize the training of the discriminator in GANs.
Barber et al.
(2018) introduced and discussed the relevant properties of a divergence which is well deﬁned on distributions with different supports.
We extend this idea to construct a surrogate of f -divergence in section 3.1 and show how to train the new divergence using the auxiliary upper bound.
T HE f - DIVERGENCE UPPER BOUND
A central contribution of our paper is the following upper bound on D f
( p ( x ) || q ( x )) between any two distributions p ( x ) and q ( x )
[Equation]
(6) where p ( x, z ) is a distribution with marginal (cid:82) p ( x, z ) dz = p ( x ) and similarly (cid:82) q ( x, z ) dz = q ( x ) .
The bound corresponds to a generalization of the auxiliary variational method (Agakov & Barber,
and follows from a straightforward application of Jensen’s inequality:
[Equation]
The result states that the divergence between two joint distributions is no less than the divergence between their marginals.
Additional properties of the auxiliary f -divergence are given in section A of the supplementary material.
We show that the bound is tight and reduces to D f ( p ( x ) || q ( x )) when performing a full unconstrained minimization of the bound with respect to p ( z | x ) (keeping q ( x, z ) ﬁxed).
3. S PREAD f - DIVERGENCE
Recently, Barber et al.
(2018) proposed the spread divergence.
We give a brief introduction and show how to apply the spread divergence to a f -divergence in order to alleviate the problem of disjoint support between the data and model distributions, as discussed in section 2.4.
For q ( x ) and p ( x ) which have disjoint supports, we deﬁne new distributions q ( y ) and p ( y ) that have the same support.
We let
[Equation]
where p ( y | x ) is a “noise" process designed such that p ( y ) and q ( y ) have the same support.
For example, if we use a Gaussian p ( y | x ) = N (cid:0) y x, σ 2 (cid:1) , then p ( y ) and q ( y ) both have support R .
We thus deﬁne the spread f -divergence
[Equation]
This satisﬁes the requirements of a divergence, that is D (cid:48)
f
( q ( x ) || p ( x )) ≥ 0 and D (cid:48)
f
( q ( x ) || p ( x ))
=
if and only if q ( x ) = p ( x ) .
The auxiliary upper bound can be easily applied to the spread f -divergence
[Equation]
3. T RAINING LATENT GENERATIVE MODELS
The bound (9) can be directly applied to form an upper bound on f -divergences for training latent generative models, including implicit models with deterministic output p θ ( x, z ) =
p θ ( x | z ) p ( z ) =
δ ( x − µ θ ( z )) p ( z ) .
The key idea is that, in situations in which the divergence D f ( p ( x ) || p
θ ( x )) is intractable, the spread divergence of the joint D f ( p ( y, z ) ||
p θ ( y, z )) may be tractable.
We introduce a variational distribution q φ ( z | y ) to express the joint p ( y, z ) =
q φ ( z | y ) p ( y ) .
Our generative model factorizes p θ ( y, z )
= p θ ( y | z ) p ( z ) as before.
The bound is then
[Equation]
Once the model has been trained we can recover the model on x by inverting the noise process.
We will use a Gaussian noise process, p ( y | x ) =
N (cid:0) y x, σ 2 (cid:1) , and a Gaussian output for the generative model, p θ ( y | z ) =
N (cid:0) y µ θ ( z ) , σ 2 (cid:1) , with both set to use the same ﬁxed variance (which allows us to invert the noise process straightforwardly).
So in practice, we learn a generative model on the y -space, p θ ( y ) , by minimizing the f -divergence to the noise corrupted data distribution p ( y ) .
After training we then recover the generative model on x -space, p θ ( x ) , by taking the mean of our generative y model as the output.
Note that this model p θ ( x ) is still stochastic (i.e. not deterministic) because of the stochasticity of the latent variable p θ ( x )
=
(cid:82) δ ( x − µ θ ( z ))
p ( z ) dz .
By using the spread divergence we have ensured both that our
[Figure]
divergence is properly deﬁned and that we have a way to invert the noise process in a theoretically grounded manner.
Similar to standard treatments in variational inference (see for example Kingma & Welling (2013);
Rezende et al.
(2014)), the variational distribution q φ ( z | y ) is only being used to tighten the resulting bound and is not a component of the generative model, although it can be used to infer structure in the latent space.
As an example, the above provides the following upper bound on the reverse KL divergence
[Equation]
The upper bound (11) can now be estimated through sampling and minimized with respect to θ and φ by using the reparameterization trick (Kingma & Welling (2013)) and taking gradients.
This results in a tractable procedure to minimize KL (
p θ ( y ) || p ( y )) 3 .
Note that this will require the estimation of the gradient (cid:79) θ (cid:82) p θ ( y | z ) p ( z ) log p ( y ) dxdz , for which we propose an efﬁcient (approximately unbiased) gradient estimator in the supplementary materials section B.
For f -divergences other than the reverse KL, since the joint f -divergence D f ( p ( y, z ) || p θ ( y, z )) is expressed as an expectation over p θ ( y | z ) p ( z ) , we can use a similar procedure to optimize the upper bound (10).
Namely we can generate ( y, z ) samples from these distributions and then estimate the bound and take gradients.
E XPERIMENTS
In the following experiments our interest is to demonstrate the applicability of the f -divergence upper bound.
The main focus is on training with the reverse KL divergence since this provides a natural "opposite" to training with the forward KL divergence .
Throughout, the data is continuous and we use a Gaussian noise process with width σ for p ( y | x ) .
We take p ( z ) to be a standard zero mean unit covariance Gaussian (thus with no trainable parameters).
Similar to standard VAE training, we use deep networks to parameterize the Gaussian model p θ ( y | z ) =
N (cid:0) y µ θ ( z ) , σ 2 (cid:1) and Gaussian variational distribution q φ ( z | y )
=
N ( z µ φ ( y ) , Σ φ ( y )) for diagonal Σ φ ( y ) .
Experimentally, we found that running several optimizer steps on φ whilst keeping θ ﬁxed is also useful to ensure that the bound is tight when adjusting θ .
We therefore use this strategy throughout training. The result that
[Figure]
optimizing the auxiliary bound with respect to only q φ ( z | y ) tightens the bound (towards the marginal divergence) is shown in the supplementary materials section A.
4. T OY PROBLEM : F OWARD KL, R EVERSE KL AND JS TRAINING
The toy dataset, as described by Roth et al.
(2017), is a mixture of seven two-dimensional Gaussians arranged in a circle and embedded in three dimensional space, see ﬁgures: ﬁgure 3, ﬁgure 6.
We use 5 hidden layers of 400 units and relu activation function for the mean and variance parameterization in q φ ( z | y ) and mean parameterization in p θ ( y | z ) with a two dimensional latent space z ∈ R 2 .
We use the KL (moment matching) objective KL ( p ( y ) ||
p θ ( y )) , reverse KL (mode seeking) objective
KL ( p θ ( y ) ||
p
( y ))
objective, and JS divergence (balance between KL and reverse KL) objective
JS ( p θ ( y ) ||
p
( y ))
to train the model.
We minimize the corresponding auxiliary f -divergence bounds by gradient descent using the Adam optimizer (Kingma & Ba, 2014) with learning rate 10 − 3 .
To evaluate the bound in each iteration we use a minibatch of size 100 to calculate p ( B ) ( y ) .
For each minibatch we draw 100 samples from p ( z ) and subsequently draw 10 samples from p θ ( y | z ) for each drawn z to generate ( y, z ) samples.
To facilitate training, we anneal the width, σ , of the spread divergence throughout the optimization process.
This enables the model to feel the presence of other distant modes (high mass regions of p ( y ) ), allowing the method to overcome any poor initialization of ( θ, φ ) .
In this experiment, σ is annealed from 1 . 0 to 0 .
1 using the formula σ = 1 . 0 ∗ (0 . 1 currentstepstotalsteps ) .
We can see in ﬁgure 3 that the model trained by JS and reverse KL divergence converge to cover each mode in the true generating distribution, and exhibits good separation of the seven modes in the latent space.
Even though the reverse KL tends to collapse a model to a single mode, provided the model p θ ( y | z ) is sufﬁciently powerful, it can correctly capture all the 7 modes.
[Figure]
4. MNIST : FORWARD AND REVERSE KL TRAINING
To model the standard MNIST handwritten character dataset (LeCun & Cortes (2010)), we parametrize the mean of p θ ( y | z ) by a neural network and the variance of the spread divergence is ﬁxed to 0.5 for both RKL training and forward KL training.
For q φ ( z | y )
we use a Gaussian with mean and isotropic covariance parameterized by a neural network.
Both networks contain 5 layers, each layer withunits and leaky-relu as activation function.
The latent z has dimension 20.
We train both forward KL and reverse KL divergence using the f -divergence upper bound.
For the reverse KL experiment, we ﬁrst initialize the model by training using the forward KL objective for 10 epochs to prevent getting a sub-optimal solution (e.g. mode collapsing).
After that, we train the model using the reverse KL objective for additional 10 epochs.
We interleave each θ update (learning rate 10 − 4 with SGD) with 20 φ updates (learning rate 10 − 3 with Adam).
We use the Monte Carlo estimation of the gradient from log p ( y ) (discussed in section B).
We also train the model by forward KL for 20 epochs to make a comparison.
All the networks are trained using the Adam optimizer with learning rate 10 − 4 .
In ﬁgure 4 we show the samples from two models.
As we can see, model trained by reverse KL generates sharper images than the model trained by forward KL.
4. C ELEB A : FORWARD AND REVERSE KL TRAINING
We pre-process CelebA
(Liu et al., 2015) images by ﬁrst taking 140x140 center crops and then resizing to 64x64.
Pixel values were then rescaled to lie in [0 , 1] .
The architectures of the convolutional encoder q φ ( z | y ) and deconvolutional decoder p θ ( y | z ) (with ﬁxed variance 0 . 5 ) are given in the supplementary material, section D. The standard deviation of the spread divergence is 0.5.
We train a VAE for 2 epochs as initialization and then train for one additional epoch for both pure forward
KL and reverse KL.
In order to ensure that the bound remained tight, we interleave each θ update (learning rate 10 − 5 with SGD) with 20 φ updates (learning rate 10 − 5 with Adam), training used in both cases.
In ﬁgure 5 we show samples from the trained models.
As we can see, the impact of the reverse
KL term in training is signiﬁcant, resulting in less variability in pose, but sharper images.
This is consistent with the "mode-seeking" behavior of the reverse KL objective.
[Figure]
4. f -GAN COMPARISON
As discussed in section 2.3, a different approach to minimizing the f -divergence is used in Nowozin et al.
(2016), utilizing a variational lower bound to the f -divergence:
[Equation]
Here f ∗ is the Fenchel conjugate and T is any class of functions that respects the domain of f ∗ .
After parameterizing T =
g f ( V φ ) (where g f : R → dom f ∗ and V φ is an unconstrained parametric function) and p θ ( x ) , the optimization scheme is then to alternately tighten (i.e. increase) the bound through changes to φ and then lower the bound through changes to θ , see ﬁgure 2.
This is of interest because the GAN objective (Goodfellow et al., 2014) can be seen as a speciﬁc instance of this scheme.
We acknowledge that Nowozin et al.
(2016) principally grounds GANs in a wider class of techniques, and is not necessarily intended as a scheme for minimizing an f -divergence.
However, it is natural to ask whether our auxiliary upper bound or the Fenchel-conjugate lower bound give different results when used to minimize the f -divergence for a similar complexity of parameter space ( θ, φ ) .
To compare the two methods we ﬁt a univariate Gaussian p θ ( x ) to data generated from a mixture of two Gaussians through the minimization of various f -divergences.
See the supplementary material for details.
For the f -GAN lower bound we use a network with two hidden layers of size 64 for
V φ ( x ) .
For our upper bound we use a network with two hidden layers of size 50 to parameterize q φ ( z | x ) and set p θ ( x, z ) to be a bivariate Gaussian, so that it marginalizes to a univariate Gaussian as required.
The upper and lower bound methods have a similar number of free parameters ( q φ has fewer hidden units but more outputs than V φ ).
The two methods result in broadly similar Gaussian ﬁts, see table 2.
In general, minimizing the upper bound results in a slightly superior ﬁt compared to the f -GAN method (Nowozin et al., 2016) in terms of proximity to the true minimal f -divergence ﬁt and proximity of the bound value to the true divergence.
Additionally we ﬁnd that minimizing our upper bound is computationally more stable than the optimization procedure required for f -GAN training (simultaneous tightening and lowering of the bound – see supplementary material E).
R ELATED WORK
The Auxiliary Variational Method (Agakov & Barber, 2004) uses an auxiliary space to minimize the joint KL divergence in order to minimize the marginal KL divergence.
We extend this method to the more general class of f -divergences.
[Table]
Variational Auto-Encoders (Kingma & Welling, 2013)
Our method is a way to train an identical class of generative and variational models
[Table]
but with a class of different optimization objectives based on f -divergences.
Since the VAE optimization scheme is a variational method of maximizing the likelihood, it is similar to our scheme with the choice of minimizing the forward KL divergence, which is also a variational form of maximum likelihood.
Both methodologies use sampling to estimate a variational bound which can be differentiated through the use of the reparameterization trick.
In Rényi divergence variational inference (Li & Turner, 2016), a variational lower bound of loglikelihood is proposed based on the Rényi divergence.
However, our joint upper bound is an estimator of f -divergence in marginal data space, it only relates to maximum likelihood learning when we use
KL divergence.
In Auxiliary Deep Generative Models (Maaløe et al., 2016), a VAE is extended with an auxiliary space.
This allows a richer variational distribution to be learned, with the correlation between latent variables being pushed to the auxiliary space to keep the calculation tractable.
This, similarly to our method, utilizes the general auxiliary variational method Agakov & Barber (2004), but is focused on making VAEs more powerful rather than providing different optimization schemes.
In the f -GAN (Nowozin et al., 2016) methodology, an interesting connection is made between the
GAN training objective and a lower bound on the f -divergence.
The authors conclude that using different divergences leads to largely similar results, and that the divergence only has a large impact when the model is "under-powered".
However, that conclusion is somewhat at odds with our own, in which we ﬁnd that the (upper bound on) different divergences gives very different model ﬁts.
Indeed, others have reached a similar conclusion: the reverse KL divergence is optimized as a GAN objective in Sønderby et al.
(2017), demonstrating that it is effective in the task of image super-resolution.
A variety of different generator objectives for GANs are used in Poole et al.
(2016), with some divergence objectives exhibiting the "mode-seeking" behavior we have observed.
In Mohamed & Lakshminarayanan (2016), the authors demonstrate an alternative approach to train f -divergence D f ( p ( x ) || q ( x ))
= (cid:82) q ( x ) f (cid:16) p ( x ) q ( x ) (cid:17) dx by directly estimating the density ratio p ( x ) q ( x ) .
This method makes a connection to GANs: the discriminator is trained to approximate the ratio and the generator loss is designed based upon different choices of f -divergence (see the supplementary material section F for details).
We thereby recognize there are three different tractable estimations of the f -divergence: 1.
ratio estimation in the marginal space 2.
Fenchel conjugate lower bound ( f -GAN) and 3.
the variational joint upper bound (introduced by our paper).
Ratio estimation by classiﬁcation has also been extended to minimize the KL-divergence in the joint space (Huszár, 2017).
Similarly, Bi-directional GAN (Donahue et al., 2016) and Ali-GAN (Dumoulin et al., 2016) augment the GAN generator with an additional inference network.
Although these models focus on similar training objectives to our own, the purpose of using the joint space is different to that of our approach.
Our method uses the joint distribution to create an upper bound in order to estimate the f -divergence in the marginal space; the latent representation is automatically achieved.
In contrast, all three methods mentioned above expand the original space to a joint space just for learning the latent representation, the divergence is estimated by either ratio estimation or GAN approaches.
Additionally, they only minimize the target divergence only at the limit of an optimal discriminator (or in the nonparametric limit, see Goodfellow et al.
(2014) and Mescheder et al.
(2017)), which may cause instability in the GAN training process (Arjovsky & Bottou, 2017).
C ONCLUSION
We introduced an upper bound on f -divergences, based on an extension of the auxiliary variational method.
The approach allows variational training of latent generative models in a much broader set of divergences than previously considered.
We showed that the method requires only a modest change to the standard VAE training algorithm but can result in a qualitatively very different ﬁtted model.
For our low dimensional toy problems, both the forward KL and reverse KL can be effective in learning the model.
However, for higher dimensional image generation, compared to standard forward KL training (VAE), training with the reverse KL tends to focus much more on ensuring that data is generated with high ﬁdelity around a smaller number of modes.
The central contribution of our work is to facilitate the application of more general f -divergences to training of probabilistic generative models with different divergences potentially giving rise to very different learned models.
S UPPLEMENTARY M ATERIAL A P ROPERTIES OF THE A UXILIARY V ARIATIONAL M ETHOD
Here we give a property of the auxiliary bound for f -divergences with differentiable f ; this covers most f of interest, and the argument extends to those f which are piecewise differentiable.
Then for the particular case of the reverse Kl divergence we give a simpler proof of this property as well as two additional properties (which do not hold for general f ).
A. D f ( p ( x ) || q ( x ))
For differentiable f we claim that when we fully optimize the auxiliary f -divergence w.r.t p ( z | x ) , this is the same as minimizing the f -divergence in the x space alone.
Let’s ﬁrst
ﬁx q ( x, z ) and ﬁnd the optimal p ( z | x ) by taking the functional derivative of the auxiliary f -divergence
[Equation]
At the minimum this will be equal to 0 (plus a constant Lagrange multiplier that comes from the constraint that p ( z | x ) is normalized).
Since f (cid:48) is not constant (if it is then the f -divergence is a constant), this then implies that the argument of f (cid:48) must be constant in z .
This implies that optimally p ( z | x ) =
q ( z | x ) .
Plugging this back into the f -divergence, it reduces to simply D f ( p ( x ) || q ( x ))
Hence, we have shown
[Equation]
Since the assumption is that D f ( p ( x ) || q ( x )) is not computationally tractable, this means that, in practice, we need to use a suboptimal p ( z | x ) , restricting p ( z | x ) to a family p θ ( z | x ) such that the joint f -divergence is computationally tractable.
A. R ELATION TO KL ( q ( x ) || p ( x ))
For the particular case of the reverse KL divergence we also provide this more straightforward proof.
Again, the claim is that when we fully optimize the auxiliary KL divergence w.r.t p ( z | x ) , this is the same as minimizing the KL in the x space alone.
Let’s ﬁrst
ﬁx q ( x, z ) and ﬁnd the optimal p ( z | x ) .
The divergence is
[Equation]
Since we are taking a positive combination of KL divergences, this means that, optimally, p ( z | x ) =
q ( z | x ) .
Plugging this back into the KL divergence, the KL reduces to simply
KL ( q (
[Equation]
Hence, we have shown
[Equation]
A. I NDEPENDENCE p ( z | x ) = p ( z )
Also for the particular case of the reverse KL divergence we can derive a result from the assumption that the auxiliary variables are independent of the observations and the prior p ( z | x ) = p ( z ) .
We have
[Equation]
Optimally, therefore, we set p ( z ) =
q ( z ) , which gives the resulting expression
[Equation]
Since we are still free to set q ( z ) , we should optimally set q ( z ) to place all its mass on the z that minimizes
[Equation]
In other words, the assumption of independence p ( z | x ) = p ( z ) implies that method is no better than computing each KL ( q ( x | z ) || p ( x )) and then choosing the single best model q ( x | z ) .
A. F ACTORIZING q ( x, z ) = q ( x ) q ( z )
Again for the reverse KL divergence, under the independence assumption q ( x, z ) =
q ( x ) q ( z ) , it is straightforward to show that
[Equation]
In the case that q ( x ) for example is a simple Gaussian distribution, this means that the independence assumption does not help enrich the complexity of the approximating distribution.
A. R ELATION TO THE ELBO
The reverse KL divergence in joint space: KL ( q ( x, z ) || p ( x, z )) is equivalent to using the ELBO to lower bound log p ( x ) in KL ( q ( x ) || p ( x )) :
[Equation]
B A PPROXIMATING GRADIENTS OF log p ( y )
For the reverse KL upper bound, we need to take calculate (cid:79) θ (cid:82) p θ ( y | z ) p ( z ) log p ( y ) dxdz , where p ( y )
=
N − 1 (cid:80) m p ( y | x ( n ) ) , i.e. a sum of delta functions (the data distribution) corrupted with a noise process as described in section 3.1.
Clearly, summing over all points in the dataset to calculate p ( y ) is computationally burdensome.
Naively using a minibatch to estimate p ( y ) inside the log results in a biased estimator, (cid:79) x log (cid:0) M − 1 (cid:80) n p (cid:0) y | x ( m ) (cid:1)(cid:1) , which we have found to be detrimental to the optimization procedure in our experiments.
To proceed, let us assume that p θ ( y | z ) and p ( y | x ( n ) ) are spherical Gaussians with the same variance σ 2 (which is the case in our experiments)
[Equation]
Where we have used the reparametrization y
= µ θ ( z ) + σ(cid:15) and p ( (cid:15) )
=
N (0 , I ) , and then noticed that we can deﬁne
[Equation]
which is a softmax over the square distance (with a scaling) between y and x ( n ) .
We can now get an unbiased estimator for this gradient (29) if we can generate samples from p ( n | (cid:15), z ) .
The computationally expensive part of calculating p ( n | (cid:15), z ) is the normalizer, which requires summing over all data points.
Given that typically the x -space will be high dimensional in practice we consider a dimensionality reduction technique to speed up computing the square distance between y and x ( n ) .
We use Principal Components Analysis (PCA) to project the x -space to a much lower dimensional space.
PCA is an appropriate choice as it maximizes the variance preserved by the lower dimensional projections, whilst minimizing the square distance between the reconstructions and the original data.
Note that the PCA projection matrix, U , is learned once on the input data { x ( n ) } .
So we approximate
[Equation]
We can now get an (approximate) unbiased estimator for (29) by sampling (cid:15), z and then n ∼ q ( n | (cid:15), z ) .
[Equation] [Equation]
Where z ( s ) ∼ p
( z ) , (cid:15) ( s ) ∼ p ( (cid:15) ) , and for each s we sample n ( t ) s ∼ q ( n | (cid:15) ( s ) , z ( s ) ) .
In the experiments on MNIST/CelebA, we sample T = 10 to form the approximation, and the PCA dimension is 50.
We also ﬁnd that scaling the contribution from this term to the gradient can sometimes improve optimization.
Using this approximation, the computation cost of the normalizer in (32) scales with the number of the data points, but we have found this not to be an issue in practice when using the PCA projection.
For a very large dataset this could be problematic though.
In this case other minibatch methods could be used to approximate this normalizer, such as Ruiz et al.
(2018) and Botev et al.
(2017), which we leave to further work.
C T ARGET DISTRIBUTION OF THE TOY PROBLEM
We train on the toy dataset described by Roth et al.
(2017), which is a mixture of seven two-dimensional Gaussians arranged in a circle and embedded in three dimensional space, see ﬁgure 6.
The standard deviation of the each Gaussian is 0.05.
D N ETWORK A RCHITECTURE
Both encoder and decoder used fully convolutional architectures with 5x5 convolutional ﬁlters and used vertical and horizontal stridesexcept the last deconvolution layer we used stride 1.
Here Conv k stands for a convolution with k ﬁlters, DeConv k for a deconvolution with k ﬁlters, BN for the batch normalization Ioffe & Szegedy (2015),
ReLU for the rectiﬁed linear units, and FC k for the fully connected layer mapping to R k .
[Equation]
E f -GAN COMPARISON
The mixture of Gaussians we attempt to ﬁt a univariate Gaussian to is plotted in Figure 7.
[Figure]
. 1) and
N 2 = N ( µ 2 = 2 , σ 2 = 0 . 5)
We plot the lower and upper bounds during training in Figure 8.
We can see the upper bound is generally faster to converge and less noisy.
It also a consistently decreasing objective, whereas the variational lower bound ﬂuctuates higher and lower in value throughout the training process
[Figure]
F C LASS P ROBABILITY E STIMATION
In Mohamed & Lakshminarayanan (2016), two ratio estimation techniques, class probability estimation and ratio matching, are discussed.
We brieﬂy show how to use the class probability estimation technique to estimate f -divergence, and refer readers to the original paper Mohamed
The density ratio can be computed by building a classiﬁer to distinguish between training data and the data generated by the model.
This ratio is p ( x ) q θ ( x ) = p ( x | y =1) p ( x | y =0) , where label y = 1 represents samples from p and y = 0 represents samples from q .
By using Bayes rule and assuming that we have the same number of samples from both p and q , we have p ( x ) q θ ( x ) =
p
( x | y =1) p ( x | y =0)
=
p ( y =1 | x ) p ( x ) p ( y =1) / p ( y =
| x ) p ( x ) p ( y =0) =
p ( y =1 | x ) p ( y =
| x ) .
We can then set the discriminator output to be D φ ( x ) = p ( y = 1 | x ) , so the ratio can be written as p ( x ) q θ ( x )
= p ( y =1 | x ) 1 − p ( y =1 | x ) =
D φ ( x ) 1 −D φ ( x ) .
The generator loss corresponding to an f -divergence can then be designed as D f ( p ( x ) ||
q θ ( x ))
= (cid:82)
q θ ( x ) f (cid:16) p ( x ) q θ ( x ) (cid:17) dx = (cid:82)
q θ ( x ) f
( D φ ( x ) 1 −D φ ( x ) ) =
E q ( z ) [ f ( D ( G θ ( z ))
1 −D ( G θ ( z )) )] .