Projection Robust Wasserstein Distance and Riemannian Optimization

06/12/2020
by   Tianyi Lin, et al.
0

Projection robust Wasserstein (PRW) distance, or Wasserstein projection pursuit (WPP), is a robust variant of the Wasserstein distance. Recent work suggests that this quantity is more robust than the standard Wasserstein distance, in particular when comparing probability measures in high-dimensions. However, it is ruled out for practical application because the optimization model is essentially non-convex and non-smooth which makes the computation intractable. Our contribution in this paper is to revisit the original motivation behind WPP/PRW, but take the hard route of showing that, despite its non-convexity and lack of nonsmoothness, and even despite some hardness results proved by <cit.> in a minimax sense, the original formulation for PRW/WPP can be efficiently computed in practice using Riemannian optimization, yielding in relevant cases better behavior than its convex relaxation. More specifically, we provide three simple algorithms with solid theoretical guarantee on their complexity bound (one in the appendix), and demonstrate their effectiveness and efficiency by conducing extensive experiments on synthetic and real data. This paper provides a first step into a computational theory of the PRW distance and provides the links between optimal transport and Riemannian optimization.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
01/21/2020

Bures-Wasserstein Geometry

The Bures-Wasserstein distance is a Riemannian distance on the space of ...
research
01/25/2019

Subspace Robust Wasserstein distances

Making sense of Wasserstein distances between discrete measures in high-...
research
12/09/2020

A Riemannian Block Coordinate Descent Method for Computing the Projection Robust Wasserstein Distance

The Wasserstein distance has become increasingly important in machine le...
research
02/12/2020

Fixed-Support Wasserstein Barycenters: Computational Hardness and Fast Algorithm

We study the fixed-support Wasserstein barycenter problem (FS-WBP), whic...
research
02/12/2020

Computational Hardness and Fast Algorithm for Fixed-Support Wasserstein Barycenter

We study in this paper the fixed-support Wasserstein barycenter problem ...
research
02/05/2021

Projection Robust Wasserstein Barycenter

Collecting and aggregating information from several probability measures...
research
07/21/2023

Convergence of SGD for Training Neural Networks with Sliced Wasserstein Losses

Optimal Transport has sparked vivid interest in recent years, in particu...

Please sign up or login with your details

Forgot password? Click here to reset