MINIMAX ESTIMATION OF SMOOTH DENSITIES IN WASSERSTEIN DISTANCE

Jonathan Niles-Weed, Quentin Berthet

Research output: Contribution to journalArticlepeer-review

Abstract

We study nonparametric density estimation problems where error is measured in the Wasserstein distance, a metric on probability distributions popular in many areas of statistics and machine learning. We give the first minimax-optimal rates for this problem for general Wasserstein distances for two classes of densities: smooth probability densities on [0,1]d bounded away from 0, and sub-Gaussian densities lying in the Hölder class Cs, s ∈ (0,1). Unlike classical nonparametric density estimation, these rates depend on whether the densities in question are bounded below, even in the compactly supported case. Motivated by variational problems involving the Wasserstein distance, we also show how to construct discretely supported measures, suitable for computational purposes, which achieve the minimax rates. Our main technical tool is an inequality giving a nearly tight dual characterization of the Wasserstein distances in terms of Besov norms.

Original languageEnglish (US)
Pages (from-to)1519-1540
Number of pages22
JournalAnnals of Statistics
Volume50
Issue number3
DOIs
StatePublished - Jun 2022

Keywords

  • Wasserstein distance
  • density estimation
  • high-dimensional statistics
  • optimal transport

ASJC Scopus subject areas

  • Statistics and Probability
  • Statistics, Probability and Uncertainty

Fingerprint

Dive into the research topics of 'MINIMAX ESTIMATION OF SMOOTH DENSITIES IN WASSERSTEIN DISTANCE'. Together they form a unique fingerprint.

Cite this