Current browse context:
stat.ML
Change to browse by:
References & Citations
Statistics > Machine Learning
Title: Near-optimal estimation of smooth transport maps with kernel sums-of-squares
(Submitted on 3 Dec 2021 (v1), last revised 29 Dec 2021 (this version, v2))
Abstract: It was recently shown that under smoothness conditions, the squared Wasserstein distance between two distributions could be efficiently computed with appealing statistical error upper bounds. However, rather than the distance itself, the object of interest for applications such as generative modeling is the underlying optimal transport map. Hence, computational and statistical guarantees need to be obtained for the estimated maps themselves. In this paper, we propose the first tractable algorithm for which the statistical $L^2$ error on the maps nearly matches the existing minimax lower-bounds for smooth map estimation. Our method is based on solving the semi-dual formulation of optimal transport with an infinite-dimensional sum-of-squares reformulation, and leads to an algorithm which has dimension-free polynomial rates in the number of samples, with potentially exponentially dimension-dependent constants.
Submission history
From: Boris Muzellec [view email][v1] Fri, 3 Dec 2021 13:45:36 GMT (127kb,D)
[v2] Wed, 29 Dec 2021 10:26:50 GMT (126kb,D)
Link back to: arXiv, form interface, contact.