Title: AUXILIARY VARIATIONAL MCMC

Abstract: We introduce Auxiliary Variational MCMC, a novel framework for learning MCMC kernels that combines recent advances in variational inference with insights drawn from traditional auxiliary variable MCMC methods such as Hamiltonian Monte Carlo. Our framework exploits low dimensional structure in the target distribution in order to learn a more efficient MCMC sampler. The resulting sampler is able to suppress random walk behaviour and mix between modes efficiently, without the need to compute gradients of the target distribution. We test our sampler on a number of challenging distributions, where the underlying structure is known, and on the task of posterior sampling in Bayesian logistic regression. Code to reproduce all experiments is available at https://github.com/AVMCMC.

1. INTRODUCTION:

Markov Chain Monte Carlo (MCMC) and Variational Inference (VI) are well-established approaches to approximating expectations under the complex distributions p that frequently arise in machine learning and statistics (Wainwright & Jordan, 2008; Brooks et al., 2011) . VI is usually fast and cheap but often places strong restrictions on the class of approximating distributions leading to irreducible bias. MCMC on the other hand is asymptotically exact but may require an infeasible amount of computation to converge. Given the complimentary strengths and weaknesses of the two methods it is natural to wish to combine them (De Freitas et al., 2001; Salimans et al., 2015) .

The most naive pairing of the two methods, simply using a variational approximation q as the proposal distribution in a Metropolis-Hastings sampler (see for example Gamerman & Lopes (2006) or De Freitas et al. (2001) ), is known to scale poorly with the dimension of the target distribution (De Freitas et al., 2001) and mix inefficiently when p is multi-modal.

In this paper we suggest an alternative approach inspired both by the successes of black box variational inference (Ranganath et al., 2014) and auxiliary variable MCMC methods, such as Hamiltonian Monte Carlo (HMC) (Duane et al., 1987; Girolami & Calderhead, 2011) .

The key contributions of our work are :

• A general framework for marrying variational inference with Markov Chain Monte Carlo in a way likely to produce efficient samplers. • The use of the auxiliary variational method to capture latent low-dimensional structure in our target distributions and exploit this structure to suppress random walk behaviour. • The extension of the Metropolis-Hastings algorithm to continuous mixture proposals.

• The introduction and demonstration of a specific instance of our framework, the Auxiliary Variational Sampler (AVS). Our sampler takes advantage of flexible distributions parameterized by neural networks that can be trained in a fully black-box manner.

2. AUXILIARY VARIATIONAL MCMC:

The key idea behind Auxiliary Variational MCMC is to exploit structure present in the target distribution p(x) by first fitting a parameterized variational approximation in an augmented space. This allows the sampler to leverage learned low-dimensional structure. In the subsequent sections we introduce the auxiliary variational method and describe how it can be combined with a carefully chosen class of proposal distributions to construct an efficient sampler.

2.1. MIXTURE PROPOSAL MCMC:

To develop a valid MCMC algorithm we need to construct an ergodic Markov chain whose stationary distribution is our target distribution p(x). In order to do this we introduce a Metropolis-Hastings (Gamerman & Lopes, 2006) like algorithm with a specially chosen form of proposal distribution. This proposal can be naturally combined with the auxiliary variational method which we will introduce in section 2.2. We first consider a mixture proposal distribution of the following form1 :

EQUATION

We prove in supplement (7) that the following forms a valid MCMC sampling step:

1. Sample a from q(a|x) 2. Sample x from q(x |a) 3. Accept the candidate sample x with probability min 1, q(x|a)q(a|x )p(x ) q(x |a)q(a|x)p(x)

(2)

otherwise reject x and define the new sample as a copy of the current x, namely x = x.

It is worth noting that the above procedure is not equivalent to simply performing Metropolis-Hastings in the joint (x, a) space, but is a sampler in x alone, specified by the marginalized proposal distribution given by (1). The two become equivalent only in the case that q(a|x) is constant, see supplement ( 7).

The mixture proposal can be extended to an arbitrary number of auxiliary variables as long as the acceptance ratio is adjusted accordingly. For example with two auxiliary variables, the proposal becomes: q(x |x) = q(x |a )q(a |a)q(a|x)dada

where q(a |a) is another arbitrary proposal distribution. The acceptance probability is now given by: min 1, p(x )q(x|a )q(a |a)q(a|x ) p(x)q(x |a)q(a|a )q(a |x)

For a proposal q(a |a) that is symmetric, q(a |a) = q(a|a ), this simplifies to:

EQUATION

We now show how the mixture proposal can be naturally combined with the auxiliary variational method.

2.2. THE AUXILIARY VARIATIONAL METHOD:

The auxiliary variational method (Agakov & Barber, 2004; Ranganath et al., 2016) is a strategy for creating more expressive families of approximating distributions for use in variational inference. Instead of minimizing the Kullback-Leibler divergence between our approximating distribution q(x) and our target p(x), we instead minimize the divergence in an augmented space (x, a) with additional auxiliary variables a. In doing this, we first define a joint q φ (x, a) in the augmented space and a joint p(x, a) = p θ (a|x)p(x) where θ and φ are parameters. Marginalizing p(x, a) over a recovers p(x) by construction. We are free to decide on the dimension and form (e.g. continuous, discrete or mixed) for a and our objective is the Kullback-Leibler divergence between the joint approximation q and joint target p:

EQUATION

Algorithm 1: Auxiliary variational sampler Initialize φ and θ

Fit the variational distribution while Not Converged do {a1, . . . , aN } ∼ q φ (a) {x1, . . . , xN } ∼ q φ (x|an)

re-parameterized L ← 1 N n log q φ (xn|an)q φ (an) p(xn)p θ (an|xn) φ ← φ -η∇ φ L θ ← θ -η∇ θ L end a ∼ q φ (a) Mixture-Model MCMC sampling x0 ∼ q φ (x|a) for t = 0, . . . , T do a ∼ p θ (a|xt) a ∼ N a a, σ 2 a I x ∼ q φ (x|a ) A ← p(x )p θ (a |x )q φ (x t |a) p(x t )p θ (a|x t )q φ (x |a ) acceptance ratio u ∼ U [0, 1] sample u from uniform distribution if u ≤ A then xt+1 = x else xt+1 = xt end end

The Auxiliary Variational Sampler. The algorithm begins by fitting a mixture variational distribution to the target p(x) by stochastic gradient descent, based on the auxiliary variational method.

After the auxiliary variational model is fitted to p(x) the MCMC sampling phase begins and is a modified form of Metropolis-Hastings sampling.

Moving to the joint space allows us to tractably learn complex approximating distributions q φ (x, a) whose marginal, q φ (x) = q φ (x|a)q φ (a)da, may be intractable.

Whilst sampling in high-dimensional spaces is difficult, in many cases of interest the high probability or typical regions lie close to much lower dimensional manifolds. A key point of fitting an auxiliary variational distribution is that, by constraining a to have dimension much lower than x, we can then exploit this low dimensional manifold to form a more efficient proposal distribution, staying close to the manifold of significant probability.

2.3. COMBINING AUXILIARY VARIATIONAL INFERENCE AND MCMC:

After fitting our variational approximation to p(x), we will have three variational distributions: q * (x|a), q * (a) and p * (a|x) that are the solution of the optimization problem given in (6). These distributions approximately satisfy the following relationships:

EQUATION

which become exact iff the divergence in (6) becomes zero. Here p * (a|x) is a learned stochastic mapping from the high-dimensional target space to the low-dimensional auxiliary space and q * (x|a) is a mapping in the opposite direction. These learned mappings can be composed in a variety of ways to form marginal proposal distributions of the kind discussed in section 2.1 and also joint proposals.

We now discuss some natural combinations and argue that these may form good proposal distributions in practice.

2.3.1. NAIVE PROPOSAL DISTRIBUTIONS:

A simple proposal constructed from our variational distribution is to perform Metropolis-Hastings in the joint (x, a) space with an independent proposal given by:

EQUATION

If the variational approximation is accurate such that q(x, a) ≈ p(x)p(a|x) then one might expect this to have high acceptance probability. A potential downside of this scheme is that proposals are independent between time-steps and in high dimensions this may cause the acceptance ratio to become impractically low.

p(x) "manifold" p(x) "manifold" auxiliary space q(a|x)

x q(x |a)

x

Figure 1 : The auxiliary random walk proposal. The variational approximation allows for small steps in the auxiliary space to correspond to large steps in the target space. Our initial point starts on a manifold of high probability in the target space, is mapped down to the low-dimension auxiliary space, perturbed, and then mapped back up to the high probability manifold. Unlike a random-walk in the x-space, our random perturbations correspond to moves along the high probability manifold.

Another natural MCMC proposal distribution to consider is:

EQUATION

where we have replaced the arbitrary proposal distribution of (1) with the optimal variational distributions learned by minimizing the joint divergence given in ( 6). We might expect this to be a promising proposal distribution, with high acceptance probability, as it already approximately satisfies the stationarity criterion of our MCMC chain. That is

q(x |x)p(x)dx = q * (x |a)p * (a|x)p(x)dxda ≈ q * (x |a)q * (x|a)q * (a)dxda = q * (x |a)q * (a)da ≈ p * (a|x )p * (a)da = p(x )

However, if the variational distribution truly captures underlying structure in p(x), it has been our experience that p * (a|x) and q * (x|a) become approximate inverses causing q(x |x) to resemble the identity mapping and thereby slowing mixing in the chain.

2.3.2. AN AUXILIARY RANDOM WALK PROPOSAL DISTRIBUTION:

To avoid the above identity mapping issue and encourage the sampler to take large steps, we introduce an additional random perturbation in the auxiliary a-space. That is we use a proposal distribution:

EQUATION

where q(a |a) = N a a, σ 2 a I is a Gaussian proposal with mean a and isotropic covariance σ 2 a I. The overall algorithm can be viewed as mapping from the high-dimensional x to the low-dimensional a, performing a random walk in the low dimensional auxiliary a-space and subsequently mapping back up to the high-dimensional target space x, see figure (1).

In high dimensions our target distribution is likely to have high probability only close to some lowdimensional manifold. The traditional Random-Walk-Metropolis Algorithm proposes new samples x by perturbing the most recent sample in an arbitrary direction, usually based on a Gaussian proposal q(x |x) = N x x, σ 2

x I ; however, in high-dimensional spaces almost all directions will correspond to steps off the manifold of high probability and thus out of the typical set. In our case, we perform the random perturbation in the low-dimensional auxiliary space and our variational distribution q * (x |a ) ensures that this move remains within the manifold of high density.

We depict this process in figure (1) and provide explicit examples of the learned latent structure of real distributions in section 3. Much like HMC, we are able to exploit the addition of auxiliary variables to encourage large moves of high probability in the target space but unlike HMC we do so by explicitly modelling the structure of the target distribution and do not require gradients of the target density. We are able to take large steps because small perturbations in a can correspond to large perturbations within the manifold of high probability; unlike methods that adapt to the local geometry of the probability manifold (Girolami & Calderhead, 2011; Strathmann et al., 2015) , our variational fit allows for larger, non-local moves in the x-space. Note that, unlike HMC, in general there is no requirement that (x, a) are continuous random variables.

2.4. CHOOSING THE VARIATIONAL FAMILY:

Within the above framework, we still have to decide on the structure of the variational distributions, q(a, x) and p(a|x). We take inspiration from recent successes in the generative modeling of complex data distributions (Kingma & Welling, 2014) , and propose to parameterize our variational distributions using deep neural networks. For continuous x we choose the following structure for each of our approximating distributions:

EQUATION

Where q φ (x|a) and p θ (a|x) are both diagonal Gaussian distributions whose means and covariances are parameterized by neural networks with parameters shared between the mean and covariance. In some experiments we also choose p θ (a|x) to be a mixture of diagonal Gaussians (see the supplementary material). Key to the flexibility of the auxiliary variational method is that whilst q(a, x) can be evaluated point-wise, the marginal q(x) is a much richer approximating distribution whose density we typically cannot evaluate point-wise. Whilst we can thus evaluate our joint approximating density we still can't compute the objective:

EQUATION

However, we recognize that an unbiased estimator of the KL divergence can be obtained by sampling from q φ (x, a). We use the standard re-parameterization trick of Kingma & Welling (2014) to reduce the variance in the corresponding gradient estimator. We refer to this version of the algorithm as the Auxiliary Variational Sampler (AVS). The full algorithm is given in algorithm (1). Reparameterisation requires the gradient of log p(x); if this is prohibitively expensive then the scorefunction estimator may be used (Kleijnen & Rubinstein, 1996) . Within this framework, different mixture proposals could be considered, based on the fitted variational distribution, but our experience is that the Auxiliary Random Walk proposal is effective in our experiments.

3. EXPERIMENTS:

To demonstrate the benefit of fitting an auxiliary variational approximation we first test our sampler on a number of distributions with known low-dimensional structure. We show how AVS is both able to recover the latent structure and exploit it for the purposes of sampling. We then demonstrate the sampler on more realistic problems.

Ring density:

The first distribution is in two dimensions x = (x 1 , x 2 ). The distribution is a ring of high probability, centered at a fixed distance from the origin, figure (2b). We construct the ring density to have explicit latent structure by defining the target distribution as: Since this ring distribution is highly constrained, pure random walk sampling in x will not efficiently move around the ring. However, provided we can find a variational fit to capture the ring structure, we expect our AVS sampler to be able to move more efficiently. To train our AVS sampler, we fit an auxiliary variational approximation with a 1 dimensional continuous mixture (see supplementary for details). We can examine the latent structure learned by the variational distribution by drawing samples from p * (a|x) as we vary x = (cos(θ), sin(θ)) for θ ∈ [0, 2π]. Looking at figure (3), we can see that the variational distribution has successfully captured the latent structure, automatically learning to map points of high probability in x onto distinct auxiliary variables. Used with the mixture proposal (10) this results in an effective sampler that is able to take much larger step sizes than random walk Metropolis (RWM) as shown in figure (2).

EQUATION

Mixtures of Gaussians and Student T:

The second pair of densities we wish to sample from is a two dimensional mixture of Gaussians with highly separated means (with a distance of 20 between the means and a standard deviation of 1), figure (2), and a two dimensional mixture of Student T distributions. These distributions have discrete latent structure and are challenging both for random walk Metropolis (RWM) (Gamerman & Lopes, 2006) and more advanced samplers such as HMC, which fail to find more than one mode in any reasonable amount of time or can struggle with the heavy tails. Our sampler has no problem finding both modes and, as shown by the plot of consecutive samples in figure (4), is able to hop between modes in single steps. We use a 1-dimensional continuous auxiliary variable.

Posterior sampling in Bayesian models The low-dimensional distributions above demonstrate the ability of our sampler to learn and exploit latent structure. Here we analyze the performance of our sampler on a more realistic problem: posterior sampling in Bayesian logistic regression. We use the heart data-set used by Song et al. (2017) , which has 13 covariates and 270 data-points. We tune all algorithms to maximize effective sample size (see supplement ( 9)) at convergence. In the case of HMC we use a number of initial runs to select an appropriate step-size and then tune the number of leapfrog steps. In all cases, we run a chain for 10000 burn-in steps and then draw 20000 samples to calculate diagnostics. In order to assess convergence and ensure the validity of the results, we monitor the Gelman-Rubin statistic (Gelman & Rubin, 1992) and visually inspect trace-plots of parameters.

3.1. EVALUATING PERFORMANCE:

Evaluation of MCMC methods is notoriously difficult (Gelman & Rubin, 1992) . Even diagnosing convergence is challenging and there is no clear agreement on the "correct" metric of performance.

In practice there are two factors that are of primary concern to an end-user: How easy is the sampler to implement and tune? and how computationally efficient is the resulting sampler? We evaluate our performance by comparison to RWM, HMC and two recently proposed neural adaptive samplers: A-NICE-MC (Song et al., 2017) and L2HMC (Levy et al., 2018) , discussed further in section 4. For learned samplers there is an additional cost involved in training the sampler that may be offset by improved performance. As a baseline, we include the independent proposal given by equation ( 8).

We choose here to focus our quantitative evaluation on the effective Sample Size (ESS), which measures how many independent samples would be equivalent to an MCMC sample, and the trainingtime for the learned samplers. We choose not to focus our investigation on burn-in time since AVS, being initialized by the variational distribution, often has short burn-in relative to competitive methods. For ESS calculations, we use the batch-means estimator applied to a single long chain as it has been shown to be more reliable than many alternatives (Thompson, 2010) , exact details are given in the supplement. For the experiments with mixtures of Gaussians we do not present ESS for HMC and RWM as both algorithms clearly fail to converge, getting trapped in one of the local modes. For all distributions except the Mixture of Gaussians, the independent sampler also failed, getting quickly stuck in a region of high-probability where the acceptance rate dropped to near zero.

To ensure consistency in timing, all methods were implemented using Tensorflow 1.10 ( Abadi & Agarwal, 2015) and run on a single Tesla K80 GPU. Code to reproduce all experiments can be found at github.com/AVMCMC. For L2HMC and A-NICE-MCMC, we perform a random search over hyper-parameters ensuring we include the parameters used in the original papers wherever possible. The figures reported are the mean of 10 independent runs alongside their standard deviations. 

4. DISCUSSION AND RELATED WORK:

The results demonstrate that Auxiliary MCMC can be used to construct a practical sampler that is able to exploit low dimensional structure and mix between modes. AVS produces competitive ESS, and initializing from a variational approximation reduces the need to burn-in. AVS also offers a straightforward method to interpolate between pure VI and MCMC. We found that the naive baseline, independent Metropolis-Hastings in the joint auxiliary-target space, failed to produce reasonable samples for all but the Mixture of Gaussians. Though the more recent neural adaptive samplers have higher effective sample sizes, this comes at a high cost in training time that, in our moderately sized problems, overwhelms their benefit. It was also our experience that training L2HMC is very sensitive to correct tuning of many hyper-parameters. In a search over 42 training configurations only 2 configurations converged to samplers able to mix between modes in the Mixture of Gaussian experiments and none for the Mixture of T. The difficulty of tuning is hard to quantify and will no doubt vary with the experience of the practitioner. We nonetheless see this as a possible barrier to the adoption of more complex adaptive methods and a possible explanation for the popularity of RWM. We found qualitatively that AVS and A-NICE-MC were more straightforwards to tune with good performance from many hyper-parameter configurations. When training time is included, the neural samplers often under-perform HMC but we expect this effect to diminish with data-size.

The work most similar to ours is Variational MCMC (De Freitas et al., 2001) , in which a variational approximation is used as a proposal distribution in independent Metropolis-Hastings. To overcome the poor scaling with dimension of the independent Metropolis algorithm, the authors interleave their variational proposal distribution with traditional RWM and make block proposals. Although their use of a variational distribution aids the convergence of MCMC, their method still struggles to mix efficiently even in relatively low dimensional problems. By introducing low dimensional auxiliary variables, we are able both to fit a more accurate approximating distribution and to leverage the learned low dimensional structure to take very large steps in the target space, even hopping between modes. In Salimans et al. (2015) the authors also consider a combination of MCMC and auxiliary variational inference but take a different approach, choosing to use MCMC kernels to design more flexible variational distributions rather than constructing an MCMC sampler at all.

There have also been other recent attempts to parameterize MCMC kernels with neural networks that don't start from variational inference. A-NICE-MCMC (Song et al., 2017) , attempts to learn MCMC transition kernels parameterized by volume preserving flows (Dinh et al., 2015) . They craft an adversarial objective to train their kernels and use a discriminator that examines pairs of samples, in order to encourage fast mixing. To ensure they have a valid sampler, they use a bootstrap procedure starting from an existing MCMC algorithm. L2HMC (Levy et al., 2018) is an extension of HMC, that parameterizes a scaling and shift of Hamiltonian dynamics with neural networks and trains this parameterization to optimize the expected squared step-size. They show that the introduction of flexible neural networks to HMC vastly improves the ability to mix between modes but still require access to the gradient of the target density.

For any adaptive MCMC method, care has to be taken to preserve the ergodicity of the chain during adaptation (Andrieu & Thoms, 2008; Roberts & Rosenthal, 2009) . To overcome this, all of the above methods, including ours, stop adaptation before collecting any samples. A potential problem with this strategy is that if the learned sampler is specialized to only a part of the distribution or the variational approximation has entirely missed regions of high probability, then such regions are unlikely to be explored. A possible avenue for future work would be to investigate iterative improvement of our variational approximation with samples drawn from the true distribution rather than our variational approximation. Another potential avenue for ensuring good coverage of the variational approximation, would be to combine our reverse KL objective, ( 14), with the KL divergence in the forward direction as this is known to encourage moment matching rather than mode seeking behavior (Minka, 2005) .

5. CONCLUSION:

We introduced a novel framework for combining MCMC and Variational Inference that makes use of the auxiliary variational method to capture low-dimensional latent structure. We have explored a particular black-box instance of this framework and demonstrated that it can be used to create a fast mixing sampler without the need to take gradients of the target distribution. The method is competitive with other recent geometry-learning approaches and opens up additional avenues for exploring how to combine the best of variational inference and sampling.

8. EXACT PARAMETERIZATION OF THE VARIATIONAL DISTRIBUTIONS:



8.1. LOW DIMENSIONAL EXAMPLES:

For the mixtures of Gaussians and ring density experiments we used an auxiliary dimension of 1 and the target dimension was 2. For the mixture of Gaussians, the variational distributions had the following form:

EQUATION

Where the µ φ (x) and Σ φ (a) are 3 layer feed-forward neural networks with 10 neurons in each layer, all but the last layer were shared between them. Similarly µ θ (x) and Σ θ (x) were also 3 layer feed-forward neural networks with all but the last layer shared. We used tanh non-linearities in all but the final layer which was simply linear.

In the case of the ring density experiments p θ (a|x) = Σ k π k N a µ k θ (x), Σ k θ (x) was a mixture of 2 Gaussians, each parameterized as above. The mixture weights π k were also learned and were paramterized as π 1 = σ(τ ), π 2 = 1 -π 1 to allow for unconstrained optimization.

8.2. REGRESSION EXAMPLES:

For Bayesian logistic regression, the target dimension was 14, the auxiliary dimension used was 2 dimensional and the number of hidden units was 300. The structure was otherwise the same as for the low dimensional experiments.

9. CALCULATION OF THE EFFECTIVE SAMPLE SIZE:

The effective sample size is calculated as the reciprocal of the auto-correlation-time. It is intended to represent the number of truly independent samples that would be equivalent to a correlated sample drawn using MCMC, in terms of the variance of estimated quantities. One definition of the auto-correlation-time, for a 1-dimensional chain, is:

EQUATION

where C τ = E [x t , x t+τ ] is the lag-τ auto-correlation of the stationary converged MCMC chain. Though multivariate definitions of the auto-correlation exist, it is common practice to report the lowest effective-sample-size across all dimensions. For ease of comparison, we adopt this practice.

There are numerous methods for estimating ρ (Thompson, 2010) . It is worth noting however that simply estimating C τ from multiple chains and substituting the estimates into the above formula does not yield a consistent estimator except in very simple cases when the auto-correlations are guaranteed to be positive. In general however, the auto-correlations can be of both negative and positive variance.

The estimator formed by substituting the empirically estimated auto-correlations does not have a variance that goes to 0 as the sequence length goes to infinityThompson (2010). Instead we use the batch-means estimator.

9.1. BATCH-MEANS ESTIMATOR:

An estimator of the auto-correlation time is computed using "batch-means" (Thompson, 2010) . A single long sequence is split into m sub-sequences and the mean of each sub-sequence is calculated. The auto-correlation-time is then estimated by using the ratio of the variance of the batch-means to the variance of the overall sequence. The estimator is given by:

ρ = m s 2 m s 2 (29)

where s 2 m is the variance of the batch-means and s 2 is the variance of the entire chain. 

:

In general we use q to denote a proposal distribution and q to denote a variational distribution.

Figures and Tables:

Figure 2: Target densities with a high degree of latent structure. (a) A ring of high density centered at the origin. (b) Mixture of Two Gaussians with highly separated means.

Figure3: (a) Samples from the learned variational distribution p θ (a|x) for x on the ring of high probability. The auxiliary space is plotted on the vertical axis, with the target space plotted in the plane. We can see that the auxiliary variational approximation has recovered the low dimensional structure of the target distribution, so that random perturbations in the auxiliary space will mostly correspond to proposals within the region of high target probability and large moves. (b): Samples from the trained Auxiliary Variational Sampler (AVS) algorithm (1). (c,d) show the distance moved per sampling step for RWM and AVS for a ring of radius 5, thus demonstrating the benefit of exploiting structure.

Figure 4: Traceplot showing 20000 consecutive samples of x 1 (blue), x 2 (red), drawn from the mixture of Gaussians, figure 2, using (a) AVS given in algorithm 1, (b) RWM. AVS can easily move between the modes at x 1 = 10 and x 1 = -10, whilst RWM get stuck in one mode.

ESS calculated using the batch-means estimator.

Table content: <table><tr><td/><td>AVS</td><td>AVS-IND</td><td>HMC</td><td>RWM</td><td>ANICE</td><td>L2HMC</td></tr><tr><td>Ring</td><td colspan="2">0.176 ± 0.005 NA</td><td>0.612 ± 0.120</td><td colspan="3">0.024 ± 0.005 0.541 ± 0.105 0.247 ± 0.062</td></tr><tr><td>Mix. of Gauss.</td><td colspan="3">0.178 ± 0.042 0.009 ± 0.004 NA</td><td>NA</td><td colspan="2">0.322 ± 0.103 0.170 ± 0.10</td></tr><tr><td colspan="3">Mix. of Student T. 0.047 ± 0.026 NA</td><td colspan="4">0.0020 ± 0.0002 0.002 ± 0.001 0.071 ± 0.026 NA</td></tr><tr><td>Log. regression</td><td colspan="2">0.066 ± 0.027 NA</td><td>0.070 ± 0.015</td><td colspan="3">0.013 ± 0.002 0.257 ± 0.035 0.562 ± 0.071</td></tr></table>

ESS/s taking into account the total time including both training time and sampling time for 20000 samples (left) and taking account only the sampling time (right). Absolute times are provided in the appendix.

Table content: <table><tr><td/><td>AVS</td><td>ANICE L2HMC</td><td/><td>AVS</td><td colspan="3">ANICE L2HMC HMC</td><td>RWM</td></tr><tr><td>Ring</td><td colspan="2">5.30e-3 1.25e-3 7.54e-5</td><td>Ring</td><td colspan="2">35.42 807.42</td><td>6.34</td><td>624.97 688.32</td></tr><tr><td>Mix. of Gauss.</td><td colspan="2">7.64e-2 2.50e-4 2.88e-5</td><td>Mix. of Gauss.</td><td colspan="2">66.08 288.10</td><td>3.24</td><td>NA</td><td>NA</td></tr><tr><td colspan="3">Mix. of Student T 1.45e-2 2.17e-4 NA</td><td colspan="3">Mix. of Student T 74.19 45.42</td><td>NA</td><td>1.16</td><td>6.12</td></tr><tr><td>Log. regression</td><td colspan="2">6.55e-2 4.30e-4 6.24e-5</td><td>Log. regression</td><td colspan="3">48.12 2660.35 87.11</td><td>180.43 1858.15</td></tr></table>

Absolute training times in seconds for each of the learned samplers for the experiments given in section 3, rounded to the nearest second.

Table content: <table><tr><td colspan="4">10 FURTHER EXPERIMENTAL DETAILS</td></tr><tr><td colspan="4">10.1 ABSOLUTE TRAINING TIMES</td></tr><tr><td/><td colspan="3">AVS ANICE L2HMC</td></tr><tr><td>Ring</td><td>70</td><td>120</td><td>3310</td></tr><tr><td>Mix. of Gauss.</td><td>2</td><td>1289</td><td>2619</td></tr><tr><td colspan="2">Mix. of Student T. 3</td><td>290</td><td>NA</td></tr><tr><td>Log. regression</td><td>2</td><td>598</td><td>4590</td></tr></table>

Peer review (iclr2019):
 This paper proposes a clever and sensible approach to using the structure learned by the auxiliary variational method to accelerate random-walk MCMC. The idea is to learn a low-dimensional latent space that explains much of the variation in the original parameter space, then do random-walk sampling in that space (while also updating a state variable in the original state, which is necessary to ensure correctness). I like this idea and think the paper merits acceptance, although there are some important unanswered questions. For example: - How does the method work on higher-dimensional target distributions? I would think it would be hard for a low-dimensional auxiliary space to have high mutual information with a much higher-dimensional space. In principle neural networks can do all sorts of crazy things, but phenomena like VAEs with low-dimensional latent spaces generating blurry samples make me suspect that auxiliary dimension should be important. - How does the method work with hierarchical models, heavy-tailed models, etc.? Rings, MoGs, and flat logistic regressions are already pretty easy targets. - Is it really so valuable to not need gradients? High-quality automatic differentiation systems are widely available, and variational inference on discrete parameters with neural nets remains a pretty hard problem in general. Some other comments: * It’s probably worth citing Ranganath et al. (2015; “Hierarchical Variational Models”), who combine the auxiliary variational method with modern stochastic VI. Also, I wonder if there are connections to approximate Bayesian computation (ABC). * I think you could prove the validity of the procedure in section 2.1 more succinctly by interpreting it as alternating a Gibbs sampling update for “a” with a Metropolis-Hastings update for “x”. If we treat “a” as an auxiliary variable such that p(a | x) = .tilde q(a | x) p(x | a) .propto p(x) .tilde q(a | x) then the equation (2) is the correct M-H acceptance probability for the proposal .tilde q(a’, x’) = δ(a’-a) .tilde q(x’ | a). Alternating between this proposal and a Gibbs update for “a” yields the mixture proposal in section 2.1. * It’s also possibly worth noting that this procedure will have a strictly lower acceptance rate than the ideal procedure of using the marginal .tilde q(x’|x) as a M-H proposal directly. Unfortunately that marginal density usually can’t be computed, which makes this ideal procedure impractical. It might be interesting to try to say something about how large this gap is for the proposed method. * "We choose not to investigate burn-in since AVS is initialized by the variational distribution and therefore has negligible if any burn-in time.” This claim seems unjustified to me. It’s only true insofar as the variational distribution is an excellent approximation to the posterior (in which case why use MCMC at all?). It’s easy to find examples where an MCMC chain initialized with a sample from a variational distribution takes quite a while to burn in.