Scholion.ai

Wasserstein generative adversarial networks are minimax optimal distribution estimators

Arthur Stéphanovitch, Eddie Aamari, Clément Levrard

The Annals of Statistics · 2024 · Volume 52 · Issue 5

DOI: 10.1214/24-AOS2430

Abstract

We provide nonasymptotic rates of convergence of the Wasserstein Generative Adversarial networks (WGAN) estimator. We build neural networks classes representing the generators and discriminators which yield a GAN that achieves the minimax optimal rate for estimating a certain probability measure μ with support in Rp. The probability μ is considered to be the push forward of the Lebesgue measure on the d-dimensional torus Td by a map g⋆:Td→Rp of smoothness β+1. Measuring the error with the γ-Hölder Integral Probability Metric (IPM), we obtain up to logarithmic factors, the minimax optimal rate O(n−β+γ2β+d∨n−12) where n is the sample size, β determines the smoothness of the target measure μ, γ is the smoothness of the IPM (γ=1 is the Wasserstein case) and d≤p is the intrinsic dimension of μ. In the process, we derive a sharp interpolation inequality between Hölder IPMs. This novel result of theory of functions spaces generalizes classical interpolation inequalities to the case where the measures involved have densities on different manifolds.

Keywords

62E17, 62G05, Distribution estimation, Generative model, interpolation inequality, Manifold, Minimax rate

Enable JavaScript for citation tools and additional paper interactions.