English

Joint control variate for faster black-box variational inference

Machine Learning 2024-03-11 v4 Machine Learning

Abstract

Black-box variational inference performance is sometimes hindered by the use of gradient estimators with high variance. This variance comes from two sources of randomness: Data subsampling and Monte Carlo sampling. While existing control variates only address Monte Carlo noise, and incremental gradient methods typically only address data subsampling, we propose a new "joint" control variate that jointly reduces variance from both sources of noise. This significantly reduces gradient variance, leading to faster optimization in several applications.

Cite

@article{arxiv.2210.07290,
  title  = {Joint control variate for faster black-box variational inference},
  author = {Xi Wang and Tomas Geffner and Justin Domke},
  journal= {arXiv preprint arXiv:2210.07290},
  year   = {2024}
}

Comments

Published in the 27th International Conference on Artificial Intelligence and Statistics (AISTATS 2024)

R2 v1 2026-06-28T03:35:18.738Z