Variational Representations and Neural Network Estimation of R\'enyi Divergences
Abstract
We derive a new variational formula for the R\'enyi family of divergences, , between probability measures and . 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 and 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