Multiscale Strategies for Computing Optimal Transport
arXiv:1708.02469
Abstract
This paper presents a multiscale approach to efficiently compute approximate optimal transport plans between point sets. It is particularly well-suited for point sets that are in high-dimensions, but are close to being intrinsically low-dimensional. The approach is based on an adaptive multiscale decomposition of the point sets. The multiscale decomposition yields a sequence of optimal transport problems, that are solved in a top-to-bottom fashion from the coarsest to the finest scale. We provide numerical evidence that this multiscale approach scales approximately linearly, in time and memory, in the number of nodes, instead of quadratically or worse for a direct solution. Empirically, the multiscale approach results in less than one percent relative error in the objective function. Furthermore, the multiscale plans constructed are of interest by themselves as they may be used to introduce novel features and notions of distances between point sets. An analysis of sets of brain MRI based on optimal transport distances illustrates the effectiveness of the proposed method on a real world data set. The application demonstrates that multiscale optimal transport distances have the potential to improve on state-of-the-art metrics currently used in computational anatomy.
Accepted to JMLR
Cited by in corpus (9)
- DOTmark - A Benchmark for Discrete Optimal Transport
- Learning with minibatch Wasserstein : asymptotic and gradient properties
- Minibatch optimal transport distances; analysis and applications
- Ground Metric Learning on Graphs
- Randomised Wasserstein Barycenter Computation: Resampling with Statistical Guarantees
- Variational Wasserstein Barycenters for Geometric Clustering
- Multiscale semidefinite programming approach to positioning problems with pairwise structure
- Convergence of Gaussian-smoothed optimal transport distance with sub-gamma distributions and dependent samples
- A Linear Transportation Distance for Pattern Recognition