Course:CPSC522/Bayesian Coresets
Bayesian Coresets
In order to decouple increasing data sizes from the runtime 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
Collaborators:
Abstract
As data sizes grow towards the infinite limit, the cost of Bayesian inference becomes increasingly prohibitive for algorithms which don't approximately exhibit linear runtime 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 runtime. 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 tradeoff 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 DensityBased 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.
Content
Definition
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 nearlinear 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, linecenter, 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 loglikelihood 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 loglikelihood 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 likelihoodbased 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 FrankWolfe algorithm ^{[8]},
 Manifold optimization on the unit hypersphere 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 runtime 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 modelspecific 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 nonasymptotic approach for controlling the error of finitesize Bayesian posterior approximations, by means of a twostep 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:
 If are coresets for , then is an coreset for .
 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 batchcoresets 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 longstanding 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 goaldependent replay:
 Semantic memory: responsible for learning general regularities/statistics of the environment; slow learning rate and overlapping, structured representations; associated with the neocortex,
 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.0} ^{1.1} ^{1.2} Agarwal, Pankaj K., Sariel HarPeled, and Kasturi R. Varadarajan. "Geometric approximation via coresets." Combinatorial and computational geometry 52 (2005): 130. http://library.msri.org/books/Book52/files/01agar.pdf
 ↑ ^{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. http://papers.nips.cc/paper/6485coresetsforscalablebayesianlogisticregression
 ↑ Madigan, David, et al. "Likelihoodbased data squashing: A modeling approach to instance construction." Data Mining and Knowledge Discovery 6.2 (2002): 173190. https://link.springer.com/article/10.1023/A:1014095614948
 ↑ ^{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. http://papers.nips.cc/paper/4363scalabletrainingofmixturemodelsviacoresets
 ↑ Feldman, Dan, Melanie Schmidt, and Christian Sohler. "Turning big data into tiny data: Constantsize coresets for kmeans, pca and projective clustering." Proceedings of the twentyfourth annual ACMSIAM symposium on Discrete algorithms. Society for Industrial and Applied Mathematics, 2013. https://dl.acm.org/citation.cfm?id=2627920
 ↑ Braverman, Vladimir, Dan Feldman, and Harry Lang. "New frameworks for offline and streaming coreset constructions." arXiv preprint arXiv:1612.00889 (2016). https://arxiv.org/abs/1612.00889
 ↑ 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. https://ieeexplore.ieee.org/abstract/document/7546405/
 ↑ ^{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). https://arxiv.org/abs/1710.05053
 ↑ 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. http://proceedings.mlr.press/v51/lucic16.html
 ↑ ^{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): 58855909. http://www.jmlr.org/papers/v18/15506.html
 ↑ ^{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). https://arxiv.org/abs/1802.01737
 ↑ ^{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). https://arxiv.org/abs/1809.09505
 ↑ Parisi, German I., Ronald Kemker, Jose L. Part, Christopher Kanan, and Stefan Wermter. "Continual lifelong learning with neural networks: A review." Neural Networks (2019). https://www.sciencedirect.com/science/article/pii/S0893608019300231
 ↑ 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. https://pdfs.semanticscholar.org/57f1/16f3e6780424463cc8416ce755a72f873aa9.pdf
 ↑ 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): 512534. https://www.sciencedirect.com/science/article/pii/S1364661316300432
 ↑ Nguyen, Cuong V., et al. "Variational continual learning." arXiv preprint arXiv:1710.10628 (2017). https://arxiv.org/abs/1710.10628
To Add
