Course:CPSC522/Bayesian Coresets

From UBC Wiki
Jump to navigation Jump to search

Bayesian Coresets

In order to decouple increasing data sizes from the run-time complexity of Bayesian posterior inference methods like MCMC, Bayesian coresets adaptively subsample and weight observations, thereby constructing a generic approximate sufficient statistic.

Principal Author: Boyan Beronov



As data sizes grow towards the infinite limit, the cost of Bayesian inference becomes increasingly prohibitive for algorithms which don't approximately exhibit linear run-time scaling and constant memory cost. This is a major reason why, despite their inherent bias, variational inference methods have become increasingly popular and relevant in comparison to the well established, but slow MCMC methods which are at least asymptotically unbiased, i.e., converge to the exact value in infinite run-time. Bayesian coresets provide a practical way of scaling up the latter by first summarizing the statistical information present in observations, – i.e., by approximating the full likelihood function at a controllable trade-off between accuracy and memory cost – and then running expensive inference algorithms on the compressed data set. Under the sole assumptions of conditional independence and tractable likelihood, recent work has developed linear cost algorithms for coreset construction with rigorous convergence bounds, which in particular are independent of the internals of the generative model. As a new, generic means for constructing approximate sufficient statistics, these methods might evolve to have significant impact on many problems of interest in probabilistic machine learning and artificial intelligence.

Builds on

Bayesian coresets arise when the idea of coresets, which originated in the context of approximation algorithms in computational geometry, is applied to Bayesian inference. Once constructed, coresets can provide significant speedups to approximate Bayesian inference, e.g., for algorithms in the MCMC and Variational Inference families. Their construction is an interesting computational problem in its own right, and is closely related to Monte Carlo methods and the sparse approximation problem.

Related Pages

Any expensive density estimation problem which tolerates approximate inference algorithms is a potential application of Bayesian coresets. This includes: Graphical Models and Bayesian Networks, Learning Probabilistic Models with Complete Data and Density-Based Unsupervised Learning. Furthermore, as a form of episodic memory, Bayesian coresets might have more general implications for online learning, Continual Learning, Bounded Rationality and Artificial General Intelligence.



Geometric Coresets

Coresets were first developed in computational geometry as a means for approximating the shape of a large point set – see Agarwal et al. [1] for a comprehensive review. If an optimization problem admits an -coreset, then there is a linear or near-linear algorithm for selecting a constant size subset of the original problem, such that the exact solution to the small problem is an approximation to the exact solution of the original problem within a factor of .

One of the original applications of geometric coresets is clustering, via algorithms like -center, -line-center, -median and -means. As the latter two are instances of summed clustering, where the objective is a sum over points, their coreset formulation operates on weighted point sets [1]. In particular, let be a point set, a distance function, a weight function, and the objective function to be minimized over candidate cluster sets of fixed size. Such problems admit -coresets, i.e., subsets and weight functions such that for any of appropriate size,

If the points , the cluster set and the distance function are replaced respectively by observations , parameters and a log-likelihood function , the structural similarity of the clustering problem to maximum likelihood estimation (MLE) becomes apparent. This naturally leads to the question whether the notion of approximation via weighted coresets can be extended to the setting of statistical inference.

Bayesian Coresets

Let a Bayesian model for observations consist of a prior measure and a likelihood with parameters . Define the log-likelihood or potential and the marginal likelihood, evidence, or partition function . Then the posterior measure over parameters is

Under the conditional independence assumption, , the potential decomposes as

In order to reduce the cost of evaluating in inference methods, a Bayesian coreset [2] approximation is introduced as

consisting of subsampled data and a corresponding weight function . The coreset posterior approximation induced by is then

Construction Algorithms

There are several variations of Bayesian coresets, including approaches such as likelihood-based data squashing (LDS) by Madigan et al. [3], which construct pseudo data points for summarization. The following discussion is restricted to methods which only select and weight observed data points.

Sampling Approaches

In the realm of randomized approximation algorithms, the first improvement over trivial uniform sampling is importance sampling, in which subsampling is performed according to some distribution that correlates with the likelihood of data points. Variations of this approach are taken by most existing literature, both for geometric and for Bayesian coresets, including Feldman et al. [4][5][6], Zhang et al. [7], Huggins et al. [2], Campbell et al. [8] and Lucic et al. [9][10]

Optimization Approaches: Hilbert Coresets

In order to improve the theoretical guarantees of Bayesian coresets over randomized constructions, Campbell et al. [8] introduced a new family of generic, deterministic algorithms. When the function space of is a Hilbert space, i.e., when it admits an inner product, Hilbert coresets can be computed by solving suitable optimization problems formulated only in terms of inner products and Hilbert norms of the (function) vectors and . Instances of this family of methods include:

  • Quadratic optimization under a polytope constraint using the Frank-Wolfe algorithm [8],
  • Manifold optimization on the unit hyper-sphere using iterative geodesic search (generalization of line search) [11], maximizing the directional alignment between the vectors of the full () and coreset () potentials.

Both methods are greedy and iterative, achieving the linear run-time required by the coreset problem formulation. The main contribution of the latter algorithm is the optimal scaling of the coreset potential, which significantly reduces approximation errors especially in the empirically relevant regime of small coresets. Regarding the choice of Hilbert norms, several options have been suggested, including a weighted norm and a weighted Fisher information metric. In general, such norms require a cheap approximation to the true posterior as a weighting measure.

Convergence Analysis

Several convergence results exist for model-specific coreset constructions, in particular for mixture models and logistic regression as in [2][4][10]. Furthermore, adaptations of classical optimization methods have been proposed for the weighted likelihood setting, e.g., weighted EM [10].

For the generic case, Huggins et al. [12] provide a non-asymptotic approach for controlling the error of finite-size Bayesian posterior approximations, by means of a two-step bound using the Fisher information and Wasserstein metrics. This enables the full characterization of the statistical error in the iterative constructions of [8] and [11], including the effect of the weighting distribution. Specifically, these approaches achieve a decay of the Wasserstein posterior distance which is exponential in , the coreset size.

Streaming and Parallel Computation

An important property of many coreset constructions is their compositionality, which enables their construction to be performed sequentially or in parallel on small batches of data, with a subsequent efficient merge operation preserving theoretical approximation guarantees [1]. In particular, the following properties hold for many geometric coresets:

  1. If are -coresets for , then is an -coreset for .
  2. If is an -coreset for , and is an -coreset for , then is an -coreset for with .

In both the sequential and the parallel setting, an optimal strategy for merging batch-coresets is then an alternation between (1) and (2) in a tree of depth .

Analogous properties hold for several Bayesian coreset constructions, e.g., in [4] and in [8] when the approximation error is measured in the weighted uniform norm . But more generally, the posterior approximation bounds from [12] provide a formalism better suited for characterizing the compositional guarantees of Bayesian coreset constructions.

Related Numerical Problems

The Hilbert coreset formulation is related to the sparse approximation problem, for which a wide range of numerical methods have been studied. See [8][11] for a discussion of the particular requirements in the coreset setting.

Relevance for Continual Learning

A long-standing challenge for machine learning and AI systems is the problem of Continual Learning or lifelong learning, i.e., the ability of an agent to learn over long time periods by incorporating new knowledge while retaining past experiences. The main obstacle for computational models of continual learning is the phenomenon of catastrophic forgetting , whereby new information interferes destructively with previously acquired knowledge representations. In a recent comprehensive review from the perspective of Artificial Neural Networks, Parisi et al. [13] describe the influence of the Complementary Learning Systems theory [14][15] on various deep learning architectures, in particular the use of dual memory and experience replay. The essential idea, motivated by a broad range of anatomical and cognitive studies, is an intricate interplay of two learning systems with complementary roles via storage, retrieval and goal-dependent replay:

  • Semantic memory: responsible for learning general regularities/statistics of the environment; slow learning rate and overlapping, structured representations; associated with the neo-cortex,
  • Episodic memory: responsible for learning arbitrary specifics of the environment; rapid learning rate and sparse representations; associated with the hippocampus.

In the context of Bayesian continual learning, the semantic and episodic memories could be interpreted as roughly corresponding to a generative model and a coreset – see, e.g., the continual learning model of Nguyen et al. [16]

Annotated Bibliography

  1. 1.0 1.1 1.2 Agarwal, Pankaj K., Sariel Har-Peled, and Kasturi R. Varadarajan. "Geometric approximation via coresets." Combinatorial and computational geometry 52 (2005): 1-30.
  2. 2.0 2.1 2.2 Huggins, Jonathan, Trevor Campbell, and Tamara Broderick. "Coresets for scalable Bayesian logistic regression." Advances in Neural Information Processing Systems. 2016.
  3. Madigan, David, et al. "Likelihood-based data squashing: A modeling approach to instance construction." Data Mining and Knowledge Discovery 6.2 (2002): 173-190.
  4. 4.0 4.1 4.2 Feldman, Dan, Matthew Faulkner, and Andreas Krause. "Scalable training of mixture models via coresets." Advances in neural information processing systems. 2011.
  5. Feldman, Dan, Melanie Schmidt, and Christian Sohler. "Turning big data into tiny data: Constant-size coresets for k-means, pca and projective clustering." Proceedings of the twenty-fourth annual ACM-SIAM symposium on Discrete algorithms. Society for Industrial and Applied Mathematics, 2013.
  6. Braverman, Vladimir, Dan Feldman, and Harry Lang. "New frameworks for offline and streaming coreset constructions." arXiv preprint arXiv:1612.00889 (2016).
  7. Zhang, Min, et al. "Computational efficient Variational Bayesian Gaussian Mixture Models via Coreset." Computer, Information and Telecommunication Systems (CITS), 2016 International Conference on. IEEE, 2016.
  8. 8.0 8.1 8.2 8.3 8.4 8.5 Campbell, Trevor, and Tamara Broderick. "Automated Scalable Bayesian Inference via Hilbert Coresets." arXiv preprint arXiv:1710.05053 (2017).
  9. Lucic, Mario, Olivier Bachem, and Andreas Krause. "Strong Coresets for Hard and Soft Bregman Clustering with Applications to Exponential Family Mixtures." Artificial Intelligence and Statistics. 2016.
  10. 10.0 10.1 10.2 Lucic, Mario, et al. "Training Gaussian mixture models at scale via coresets." The Journal of Machine Learning Research 18.1 (2017): 5885-5909.
  11. 11.0 11.1 11.2 Campbell, Trevor, and Tamara Broderick. "Bayesian coreset construction via greedy iterative geodesic ascent." arXiv preprint arXiv:1802.01737 (2018).
  12. 12.0 12.1 Huggins, Jonathan H., et al. "Practical bounds on the error of Bayesian posterior approximations: A nonasymptotic approach." arXiv preprint arXiv:1809.09505 (2018).
  13. Parisi, German I., Ronald Kemker, Jose L. Part, Christopher Kanan, and Stefan Wermter. "Continual lifelong learning with neural networks: A review." Neural Networks (2019).
  14. McClelland, James L., Bruce L. McNaughton, and Randall C. O'reilly. "Why there are complementary learning systems in the hippocampus and neocortex: insights from the successes and failures of connectionist models of learning and memory." Psychological review 102.3 (1995): 419.
  15. Kumaran, Dharshan, Demis Hassabis, and James L. McClelland. "What learning systems do intelligent agents need? Complementary Learning Systems theory updated." Trends in cognitive sciences 20.7 (2016): 512-534.
  16. Nguyen, Cuong V., et al. "Variational continual learning." arXiv preprint arXiv:1710.10628 (2017).

To Add

Some rights reserved
Permission is granted to copy, distribute and/or modify this document according to the terms in Creative Commons License, Attribution-NonCommercial-ShareAlike 3.0. The full text of this license may be found here: CC by-nc-sa 3.0