English

Variational Representations and Neural Network Estimation of R\'enyi Divergences

Machine Learning 2021-07-21 v4 Information Theory Machine Learning math.IT Probability

Abstract

We derive a new variational formula for the R\'enyi family of divergences, Rα(QP)R_\alpha(Q\|P), between probability measures QQ and PP. Our result generalizes the classical Donsker-Varadhan variational formula for the Kullback-Leibler divergence. We further show that this R\'enyi variational formula holds over a range of function spaces; this leads to a formula for the optimizer under very weak assumptions and is also key in our development of a consistency theory for R\'enyi divergence estimators. By applying this theory to neural-network estimators, we show that if a neural network family satisfies one of several strengthened versions of the universal approximation property then the corresponding R\'enyi divergence estimator is consistent. In contrast to density-estimator based methods, our estimators involve only expectations under QQ and PP and hence are more effective in high dimensional systems. We illustrate this via several numerical examples of neural network estimation in systems of up to 5000 dimensions.

Keywords

Cite

@article{arxiv.2007.03814,
  title  = {Variational Representations and Neural Network Estimation of R\'enyi Divergences},
  author = {Jeremiah Birrell and Paul Dupuis and Markos A. Katsoulakis and Luc Rey-Bellet and Jie Wang},
  journal= {arXiv preprint arXiv:2007.03814},
  year   = {2021}
}

Comments

24 pages, 2 figures