6
8

(f,Γ)(f,Γ)-Divergences: Interpolating between ff-Divergences and Integral Probability Metrics

Abstract

We develop a rigorous and general framework for constructing information-theoretic divergences that subsume both ff-divergences and integral probability metrics (IPMs), such as the 11-Wasserstein distance. We prove under which assumptions these divergences, hereafter referred to as (f,Γ)(f,\Gamma)-divergences, provide a notion of `distance' between probability measures and show that they can be expressed as a two-stage mass-redistribution/mass-transport process. The (f,Γ)(f,\Gamma)-divergences inherit features from IPMs, such as the ability to compare distributions which are not absolutely continuous, as well as from ff-divergences, namely the strict concavity of their variational representations and the ability to control heavy-tailed distributions for particular choices of ff. When combined, these features establish a divergence with improved properties for estimation, statistical learning, and uncertainty quantification applications. Using statistical learning as an example, we demonstrate their advantage in training generative adversarial networks (GANs) for heavy-tailed, not-absolutely continuous sample distributions. We also show improved performance and stability over gradient-penalized Wasserstein GAN in image generation.

View on arXiv
Comments on this paper