Sparse Variational Inference: Bayesian Coresets from Scratch

Abstract

The proliferation of automated inference algorithms in Bayesian statistics has provided practitioners newfound access to fast, reproducible data analysis and powerful statistical models. Designing automated methods that are also both computationally scalable and theoretically sound, however, remains a significant challenge. Recent work on Bayesian coresets takes the approach of compressing the dataset before running a standard inference algorithm, providing both scalability and guarantees on posterior approximation error. But the automation of past coreset methods is limited because they depend on the availability of a reasonable coarse posterior approximation, which is difficult to specify in practice. In the present work we remove this requirement by formulating coreset construction as sparsity-constrained variational inference within an exponential family. This perspective leads to a novel construction via greedy optimization, and also provides a unifying information-geometric view of present and past methods. The proposed Riemannian coreset construction algorithm is fully automated, requiring no problem-specific inputs aside from the probabilistic model and dataset. In addition to being significantly easier to use than past methods, experiments demonstrate that past coreset constructions are fundamentally limited by the fixed coarse posterior approximation; in contrast, the proposed algorithm is able to continually improve the coreset, providing state-of-the-art Bayesian dataset summarization with orders-of-magnitude reduction in KL divergence to the exact posterior.

arXiv commentIn Advances in Neural Information Processing Systems, 2019

References in corpus (12)

• Approximation and learning by greedy algorithms
• Frequentist Consistency of Variational Bayes
• Austerity in MCMC Land: Cutting the Metropolis-Hastings Budget
• Streaming Variational Bayes
• Bayesian Posterior Sampling via Stochastic Gradient Fisher Scoring
• Automated Scalable Bayesian Inference via Hilbert Coresets
• Firefly Monte Carlo: Exact MCMC with Subsets of Data
• Consistency of Variational Bayes Inference for Estimation and Model Selection in Mixtures
• Random projections for Bayesian regression
• Optimally-Weighted Herding is Bayesian Quadrature
• The Fundamental Incompatibility of Hamiltonian Monte Carlo and Data Subsampling
• Variational consensus Monte Carlo

Cited by in corpus (1)

• Fitting large mixture models using stochastic component selection