Skip to main content
U.S. Department of Energy
Office of Scientific and Technical Information

Potential Flow Generator With L2 Optimal Transport Regularity for Generative Models

Journal Article · · IEEE Transactions on Neural Networks and Learning Systems
Here, we propose a potential flow generator with $$L_{2}$$ optimal transport regularity, which can be easily integrated into a wide range of generative models, including different versions of generative adversarial networks (GANs) and normalizing flow models. With only a slight augmentation to the original generator loss functions, our generator not only tries to transport the input distribution to the target one but also aims to find the one with minimum $$L_{2}$$ transport cost. We show the effectiveness of our method in several 2-D problems and illustrate the concept of “proximity” due to the $$L_{2}$$ optimal transport regularity. Subsequently, we demonstrate the effectiveness of the potential flow generator in image translation tasks with unpaired training data from the MNIST data set and the CelebA data set with a comparison against vanilla Wasserstein GAN with gradient penalty (WGAN-GP) and CycleGAN.
Research Organization:
Brown Univ., Providence, RI (United States)
Sponsoring Organization:
USDOE
Grant/Contract Number:
SC0019453
OSTI ID:
2281642
Alternate ID(s):
OSTI ID: 1980528
Journal Information:
IEEE Transactions on Neural Networks and Learning Systems, Journal Name: IEEE Transactions on Neural Networks and Learning Systems Journal Issue: 2 Vol. 33; ISSN 2162-237X
Publisher:
IEEE Computational Intelligence SocietyCopyright Statement
Country of Publication:
United States
Language:
English

References (15)

Polar factorization and monotone rearrangement of vector-valued functions journal June 1991
The geometry of optimal transportation journal January 1996
A computational fluid mechanics solution to the Monge-Kantorovich mass transfer problem journal January 2000
{Euclidean, metric, and Wasserstein} gradient flows: an overview journal March 2017
Natural gradient via optimal transport journal November 2018
DGM: A deep learning algorithm for solving partial differential equations journal December 2018
Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations journal February 2019
Artificial neural networks for solving ordinary and partial differential equations journal January 1998
Image-to-Image Translation with Conditional Adversarial Networks conference July 2017
Generative Modeling Using the Sliced Wasserstein Distance conference June 2018
Deep Learning Face Attributes in the Wild conference December 2015
Unpaired Image-to-Image Translation Using Cycle-Consistent Adversarial Networks conference October 2017
DualGAN: Unsupervised Dual Learning for Image-to-Image Translation conference October 2017
Inverting the Generator of a Generative Adversarial Network journal July 2019
Viscous Fingering: An Optimal Bound on the Growth Rate of the Mixing Zone journal August 1997