Bayesian posterior approximation via greedy particle optimization

Abstract

In Bayesian inference, the posterior distributions are difficult to obtain analytically for complex models such as neural networks. Variational inference usually uses a parametric distribution for approximation, from which we can easily draw samples. Recently discrete approximation by particles has attracted attention because of its high expression ability. An example is Stein variational gradient descent (SVGD), which iteratively optimizes particles. Although SVGD has been shown to be computationally efficient empirically, its theoretical properties have not been clarified yet and no finite sample bound of the convergence rate is known. Another example is the Stein points (SP) method, which minimizes kernelized Stein discrepancy directly. Although a finite sample bound is assured theoretically, SP is computationally inefficient empirically, especially in high-dimensional problems. In this paper, we propose a novel method named maximum mean discrepancy minimization by the Frank-Wolfe algorithm (MMD-FW), which minimizes MMD in a greedy way by the FW algorithm. Our method is computationally efficient empirically and we show that its finite sample convergence bound is in a linear order in finite dimensions.

References in corpus (10)

• Adam: A Method for Stochastic Optimization
• Stein Variational Gradient Descent as Gradient Flow
• On the Equivalence between Herding and Conditional Gradient Algorithms
• A Linear-Time Kernel Goodness-of-Fit Test
• Provable Bayesian Inference via Particle Mirror Descent
• Frank-Wolfe Bayesian Quadrature: Probabilistic Integration with Theoretical Guarantees
• Parallel and Distributed Block-Coordinate Frank-Wolfe Algorithms
• Sequential Kernel Herding: Frank-Wolfe Optimization for Particle Filtering
• A Unified Optimization View on Generalized Matching Pursuit and Frank-Wolfe
• Boosting Variational Inference: an Optimization Perspective