Timezone: »

Generalized Sliced Wasserstein Distances
Soheil Kolouri · Kimia Nadjahi · Umut Simsekli · Roland Badeau · Gustavo Rohde

Tue Dec 10 10:45 AM -- 12:45 PM (PST) @ East Exhibition Hall B + C #76

The Wasserstein distance and its variations, e.g., the sliced-Wasserstein (SW) distance, have recently drawn attention from the machine learning community. The SW distance, specifically, was shown to have similar properties to the Wasserstein distance, while being much simpler to compute, and is therefore used in various applications including generative modeling and general supervised/unsupervised learning. In this paper, we first clarify the mathematical connection between the SW distance and the Radon transform. We then utilize the generalized Radon transform to define a new family of distances for probability measures, which we call generalized sliced-Wasserstein (GSW) distances. We further show that, similar to the SW distance, the GSW distance can be extended to a maximum GSW (max-GSW) distance. We then provide the conditions under which GSW and max-GSW distances are indeed proper metrics. Finally, we compare the numerical performance of the proposed distances on the generative modeling task of SW flows and report favorable results.

Author Information

Soheil Kolouri (HRL Laboratories LLC)

** Soheil Kolouri ** is a research scientist and principal investigator at HRL Laboratories, Malibu, CA. His research lies in the intersection of applied mathematics, machine learning, and computer vision. He is currently the PI on DARPA Learning with Less Labels (LwLL) and the Co-PI on DARPA Lifelong Learning Machines (L2M) programs. Before joining HRL, he was a post-doctoral researcher at Carnegie Melon University. He received his Ph.D. in Biomedical Engineering from Carnegie Mellon University, where he was the recipient of Bertucci Fellowship Award for outstanding graduate students from the College of Engineering in 2014, and the Outstanding Dissertation Award from the Biomedical Engineering Department in 2015.

Kimia Nadjahi (Télécom ParisTech)
Umut Simsekli (Institut Polytechnique de Paris/ University of Oxford)
Roland Badeau (Télécom ParisTech)
Gustavo Rohde (University of Virginia)

More from the Same Authors