Saved in:
Bibliographic Details
Main Authors: Chen, Naitong, Campbell, Trevor
Format: Preprint
Published: 2023
Subjects:
Online Access:https://arxiv.org/abs/2310.17063
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909133156909056
author Chen, Naitong
Campbell, Trevor
author_facet Chen, Naitong
Campbell, Trevor
contents A Bayesian coreset is a small, weighted subset of data that replaces the full dataset during inference in order to reduce computational cost. However, state of the art methods for tuning coreset weights are expensive, require nontrivial user input, and impose constraints on the model. In this work, we propose a new method -- Coreset MCMC -- that simulates a Markov chain targeting the coreset posterior, while simultaneously updating the coreset weights using those same draws. Coreset MCMC is simple to implement and tune, and can be used with any existing MCMC kernel. We analyze Coreset MCMC in a representative setting to obtain key insights about the convergence behaviour of the method. Empirical results demonstrate that Coreset MCMC provides higher quality posterior approximations and reduced computational cost compared with other coreset construction methods. Further, compared with other general subsampling MCMC methods, we find that Coreset MCMC has a higher sampling efficiency with competitively accurate posterior approximations.
format Preprint
id arxiv_https___arxiv_org_abs_2310_17063
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Coreset Markov Chain Monte Carlo
Chen, Naitong
Campbell, Trevor
Computation
Machine Learning
A Bayesian coreset is a small, weighted subset of data that replaces the full dataset during inference in order to reduce computational cost. However, state of the art methods for tuning coreset weights are expensive, require nontrivial user input, and impose constraints on the model. In this work, we propose a new method -- Coreset MCMC -- that simulates a Markov chain targeting the coreset posterior, while simultaneously updating the coreset weights using those same draws. Coreset MCMC is simple to implement and tune, and can be used with any existing MCMC kernel. We analyze Coreset MCMC in a representative setting to obtain key insights about the convergence behaviour of the method. Empirical results demonstrate that Coreset MCMC provides higher quality posterior approximations and reduced computational cost compared with other coreset construction methods. Further, compared with other general subsampling MCMC methods, we find that Coreset MCMC has a higher sampling efficiency with competitively accurate posterior approximations.
title Coreset Markov Chain Monte Carlo
topic Computation
Machine Learning
url https://arxiv.org/abs/2310.17063