Amortized Projection Optimization for Sliced Wasserstein Generative Models

03/25/2022
by   Khai Nguyen, et al.
2

Seeking informative projecting directions has been an important task in utilizing sliced Wasserstein distance in applications. However, finding these directions usually requires an iterative optimization procedure over the space of projecting directions, which is computationally expensive. Moreover, the computational issue is even more severe in deep learning applications, where computing the distance between two mini-batch probability measures is repeated several times. This nested-loop has been one of the main challenges that prevent the usage of sliced Wasserstein distances based on good projections in practice. To address this challenge, we propose to utilize the learning-to-optimize technique or amortized optimization to predict the informative direction of any given two mini-batch probability measures. To the best of our knowledge, this is the first work that bridges amortized optimization and sliced Wasserstein generative models. In particular, we derive linear amortized models, generalized linear amortized models, and non-linear amortized models which are corresponding to three types of novel mini-batch losses, named amortized sliced Wasserstein. We demonstrate the favorable performance of the proposed sliced losses in deep generative modeling on standard benchmark datasets.

READ FULL TEXT

page 12

page 13

page 16

page 17

page 18

research
01/10/2023

Markovian Sliced Wasserstein Distances: Beyond Independent Projections

Sliced Wasserstein (SW) distance suffers from redundant projections due ...
research
02/18/2020

Distributional Sliced-Wasserstein and Applications to Generative Modeling

Sliced-Wasserstein distance (SWD) and its variation, Max Sliced-Wasserst...
research
04/04/2022

Revisiting Sliced Wasserstein on Images: From Vectorization to Convolution

The conventional sliced Wasserstein is defined between two probability m...
research
04/30/2023

Control Variate Sliced Wasserstein Estimators

The sliced Wasserstein (SW) distances between two probability measures a...
research
01/12/2023

Self-Attention Amortized Distributional Projection Optimization for Sliced Wasserstein Point-Cloud Reconstruction

Max sliced Wasserstein (Max-SW) distance has been widely known as a solu...
research
04/01/2022

DAG-WGAN: Causal Structure Learning With Wasserstein Generative Adversarial Networks

The combinatorial search space presents a significant challenge to learn...
research
01/28/2022

Wasserstein Iterative Networks for Barycenter Estimation

Wasserstein barycenters have become popular due to their ability to repr...

Please sign up or login with your details

Forgot password? Click here to reset