Poster
A Combinatorial Algorithm for the Semi-Discrete Optimal Transport Problem
Pankaj Agarwal · Sharath Raghvendra · Pouyan Shirzadian · Keegan Yao
West Ballroom A-D #6007
Abstract:
Optimal Transport (OT, also known as the Wasserstein distance) is a popular metric for comparing probability distributions and has been successfully used in many machine-learning applications.In the semi-discrete -Wasserstein problem, we wish to compute the cheapest way to transport all the mass from a continuous distribution to a discrete distribution in for , where the cost of transporting unit mass between points and is . When both distributions are discrete, a simple combinatorial framework has been used to find the exact solution (see e.g. [Orlin, STOC 1988]). In this paper, we propose a combinatorial framework for the semi-discrete OT, which can be viewed as an extension of the combinatorial framework for the discrete OT but requires several new ideas. We present a new algorithm that given and in and a parameter , computes an -additive approximate semi-discrete transport plan in time (in the worst case), where is the support-size of the discrete distribution and we assume that the mass of inside a triangle can be computed in time. Our algorithm is significantly faster than the known algorithms, and unlike many numerical algorithms, it does not make any assumptions on the smoothness of .As an application of our algorithm, we describe a data structure to store a large discrete distribution (with support size ) using space so that, given a query discrete distribution (with support size ), an -additive approximate transport plan can be computed in time in dimensions. Our algorithm and data structure extend to higher dimensions as well as to -Wasserstein problem for any .
Chat is not available.