From optimal transport to generative modeling: the VEGAN cookbook
Not on the map. Why?
Abstract (arXiv)
We study unsupervised generative modeling in terms of the optimal transport (OT) problem between true (but unknown) data distribution PX and the latent variable model distribution PG. We show that the OT problem can be equivalently written in terms of probabilistic encoders, which are constrained to match the posterior and prior distributions over the latent space. When relaxed, this constrained optimization problem leads to a penalized optimal transport (POT) objective, which can be efficiently minimized using stochastic gradient descent by sampling from PX and PG. We show that POT for the 2-Wasserstein distance coincides with the objective heuristically employed in adversarial auto-encoders (AAE) (Makhzani et al., 2016), which provides the first theoretical justification for AAEs known to the authors. We also compare POT to other popular techniques like variational auto-encoders (VAE) (Kingma and Welling, 2014). Our theoretical results include (a) a better understanding of the commonly observed blurriness of images generated by VAEs, and (b) establishing duality between Wasserstein GAN (Arjovsky and Bottou, 2017) and POT for the 1-Wasserstein distance.
Key facts. Cited by 41 papers in the map (data through 7 Oct 2026). First posted on arXiv 22 May 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 7; 2022: 10; 2023: 9; 2024: 7; 2025: 7; 2026: 1.
41 papers.
Sort:
By area:
- 2021Don't Generate Me: Training Differentially Private Generative Models with Sinkhorn Divergencecited by 54
- 2021Adversarial Intrinsic Motivation for Reinforcement Learningcited by 40
- 2023The Monge Gap: A Regularizer to Learn All Transport Mapscited by 32
- 2023Scalable Optimal Transport Methods in Machine Learning: A Contemporary Surveycited by 27
- 2021Disentangled Recurrent Wasserstein Autoencodercited by 22
- 2024Neural Optimal Transport with Lagrangian Costscited by 16
- 2022An Optimal Transport Perspective on Unpaired Image Super-Resolutioncited by 12
- 2022Bottlenecks CLUB: Unifying Information-Theoretic Trade-offs Among Complexity, Leakage, and Utilitycited by 12
- 2022Continual Variational Autoencoder Learning via Online Cooperative Memorizationcited by 12
- 2021Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, and Morecited by 10
- 2022Review of Disentanglement Approaches for Medical Applications -- Towards Solving the Gordian Knot of Generative Models in Healthcarecited by 9
- 2022InfoOT: Information Maximizing Optimal Transportcited by 9
29 more
- 2024Accelerating Sinkhorn Algorithm with Sparse Newton Iterationscited by 9
- 2021InvGAN: Invertible GANscited by 7
- 2022Shedding a PAC-Bayesian Light on Adaptive Sliced-Wasserstein Distancescited by 7
- 2024A Review of Bayesian Uncertainty Quantification in Deep Probabilistic Image Segmentationcited by 6
- 2023Differentially Private Gradient Flow based on the Sliced Wasserstein Distancecited by 4
- 2024Imitation from Diverse Behaviors: Wasserstein Quality Diversity Imitation Learning with Single-Step Archive Explorationcited by 4
- 2021Bayesian Distributional Policy Gradientscited by 3
- 2022Challenges in creative generative models for music: a divergence maximization perspectivecited by 3
- 2023Approximation properties of slice-matching operatorscited by 3
- 2024A Sinkhorn-type Algorithm for Constrained Optimal Transportcited by 3
- 2025Equivariant Flow Matching for Symmetry-Breaking Bifurcation Problemscited by 3
- 2023Investigating and Improving Latent Density Segmentation Models for Aleatoric Uncertainty Quantification in Medical Imagingcited by 2
- 2024Rapid Review of Generative AI in Smart Medical Applicationscited by 2
- 2021Momentum Contrastive Autoencoder: Using Contrastive Learning for Latent Space Distribution Matching in WAEcited by 1
- 2025Sparse Partial Optimal Transport via Quadratic Regularizationcited by 1
- 2026Learning to Orchestrate Agents under Uncertaintycited by 1
- 2022Unified Perspective on Probability Divergence via Maximum Likelihood Density Ratio Estimation: Bridging KL-Divergence and Integral Probability Metricscited by 0
- 2022Multisymplectic Formulation of Deep Learning Using Mean--Field Type Control and Nonlinear Stability of Training Algorithmcited by 0
- 2022Auto-Encoding Goodness of Fitcited by 0
- 2023Wasserstein Auto-encoded MDPs: Formal Verification of Efficiently Distilled RL Policies with Many-sided Guaranteescited by 0
- 2023On Certified Generalization in Structured Predictioncited by 0
- 2023Joint Distributional Learning via Cramer-Wold Distancecited by 0
- 2023Balanced Marginal and Joint Distributional Learning via Mixture Cramer-Wold Distancecited by 0
- 2024Sum-of-Squares Programming for Ma-Trudinger-Wang Regularity of Optimal Transport Mapscited by 0
- 2025Cloud Model Characteristic Function Auto-Encoder: Integrating Cloud Model Theory with MMD Regularization for Enhanced Generative Modelingcited by 0
- 2025CbLDM: A Diffusion Model for recovering nanostructure from atomic pair distribution functioncited by 0
- 2025Bispectral OT: Dataset Comparison using Symmetry-Aware Optimal Transportcited by 0
- 2025MELCOT: A Hybrid Learning Architecture with Marginal Preservation for Matrix-Valued Regressioncited by 0
- 2025Merging without Forgetting: Continual Fusion of Task-Specific Models via Optimal Transportcited by 0
- 2026Learning to Orchestrate Agents under Uncertainty26 May 2026
- 2025Merging without Forgetting: Continual Fusion of Task-Specific Models via Optimal Transport24 Nov 2025
- 2025MELCOT: A Hybrid Learning Architecture with Marginal Preservation for Matrix-Valued Regression27 Sep 2025
- 2025Bispectral OT: Dataset Comparison using Symmetry-Aware Optimal Transport25 Sep 2025
- 2025Equivariant Flow Matching for Symmetry-Breaking Bifurcation Problems3 Sep 2025
- 2025CbLDM: A Diffusion Model for recovering nanostructure from atomic pair distribution function1 Sep 2025
- 2025Sparse Partial Optimal Transport via Quadratic Regularization11 Aug 2025
- 2025Cloud Model Characteristic Function Auto-Encoder: Integrating Cloud Model Theory with MMD Regularization for Enhanced Generative Modeling6 Aug 2025
- 2024Sum-of-Squares Programming for Ma-Trudinger-Wang Regularity of Optimal Transport Maps17 Dec 2024
- 2024A Review of Bayesian Uncertainty Quantification in Deep Probabilistic Image Segmentation25 Nov 2024
- 2024Imitation from Diverse Behaviors: Wasserstein Quality Diversity Imitation Learning with Single-Step Archive Exploration11 Nov 2024
- 2024Rapid Review of Generative AI in Smart Medical Applications8 Jun 2024
29 more
- 2024Neural Optimal Transport with Lagrangian Costs1 Jun 2024
- 2024A Sinkhorn-type Algorithm for Constrained Optimal Transport8 Mar 2024
- 2024Accelerating Sinkhorn Algorithm with Sparse Newton Iterations20 Jan 2024
- 2023Differentially Private Gradient Flow based on the Sliced Wasserstein Distance13 Dec 2023
- 2023Balanced Marginal and Joint Distributional Learning via Mixture Cramer-Wold Distance6 Dec 2023
- 2023Joint Distributional Learning via Cramer-Wold Distance25 Oct 2023
- 2023Approximation properties of slice-matching operators16 Oct 2023
- 2023Investigating and Improving Latent Density Segmentation Models for Aleatoric Uncertainty Quantification in Medical Imaging31 Jul 2023
- 2023On Certified Generalization in Structured Prediction15 Jun 2023
- 2023Scalable Optimal Transport Methods in Machine Learning: A Contemporary Survey8 May 2023
- 2023Wasserstein Auto-encoded MDPs: Formal Verification of Efficiently Distilled RL Policies with Many-sided Guarantees22 Mar 2023
- 2023The Monge Gap: A Regularizer to Learn All Transport Maps9 Feb 2023
- 2022Challenges in creative generative models for music: a divergence maximization perspective16 Nov 2022
- 2022Auto-Encoding Goodness of Fit12 Oct 2022
- 2022InfoOT: Information Maximizing Optimal Transport6 Oct 2022
- 2022Continual Variational Autoencoder Learning via Online Cooperative Memorization20 Jul 2022
- 2022Bottlenecks CLUB: Unifying Information-Theoretic Trade-offs Among Complexity, Leakage, and Utility11 Jul 2022
- 2022Multisymplectic Formulation of Deep Learning Using Mean--Field Type Control and Nonlinear Stability of Training Algorithm7 Jul 2022
- 2022Shedding a PAC-Bayesian Light on Adaptive Sliced-Wasserstein Distances7 Jun 2022
- 2022Review of Disentanglement Approaches for Medical Applications -- Towards Solving the Gordian Knot of Generative Models in Healthcare21 Mar 2022
- 2022An Optimal Transport Perspective on Unpaired Image Super-Resolution2 Feb 2022
- 2022Unified Perspective on Probability Divergence via Maximum Likelihood Density Ratio Estimation: Bridging KL-Divergence and Integral Probability Metrics31 Jan 2022
- 2021InvGAN: Invertible GANs8 Dec 2021
- 2021Don't Generate Me: Training Differentially Private Generative Models with Sinkhorn Divergence1 Nov 2021
- 2021Momentum Contrastive Autoencoder: Using Contrastive Learning for Latent Space Distribution Matching in WAE19 Oct 2021
- 2021Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, and More14 Jul 2021
- 2021Adversarial Intrinsic Motivation for Reinforcement Learning27 May 2021
- 2021Bayesian Distributional Policy Gradients20 Mar 2021
- 2021Disentangled Recurrent Wasserstein Autoencoder19 Jan 2021
- 2021Don't Generate Me: Training Differentially Private Generative Models with Sinkhorn Divergencecited by 54
- 2023The Monge Gap: A Regularizer to Learn All Transport Mapscited by 32
- 2023Scalable Optimal Transport Methods in Machine Learning: A Contemporary Surveycited by 27
- 2021Disentangled Recurrent Wasserstein Autoencodercited by 22
- 2024Neural Optimal Transport with Lagrangian Costscited by 16
- 2022An Optimal Transport Perspective on Unpaired Image Super-Resolutioncited by 12
- 2022Bottlenecks CLUB: Unifying Information-Theoretic Trade-offs Among Complexity, Leakage, and Utilitycited by 12
- 2022Continual Variational Autoencoder Learning via Online Cooperative Memorizationcited by 12
- 2021Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, and Morecited by 10
- 2022Review of Disentanglement Approaches for Medical Applications -- Towards Solving the Gordian Knot of Generative Models in Healthcarecited by 9
- 2022InfoOT: Information Maximizing Optimal Transportcited by 9
- 2024Accelerating Sinkhorn Algorithm with Sparse Newton Iterationscited by 9
12 most cited of 37 All 37 (JSON)
- 2026Learning to Orchestrate Agents under Uncertainty26 May 2026
- 2025Merging without Forgetting: Continual Fusion of Task-Specific Models via Optimal Transport24 Nov 2025
- 2025MELCOT: A Hybrid Learning Architecture with Marginal Preservation for Matrix-Valued Regression27 Sep 2025
- 2025Bispectral OT: Dataset Comparison using Symmetry-Aware Optimal Transport25 Sep 2025
- 2025Sparse Partial Optimal Transport via Quadratic Regularization11 Aug 2025
- 2025Cloud Model Characteristic Function Auto-Encoder: Integrating Cloud Model Theory with MMD Regularization for Enhanced Generative Modeling6 Aug 2025
- 2024Sum-of-Squares Programming for Ma-Trudinger-Wang Regularity of Optimal Transport Maps17 Dec 2024
- 2024A Review of Bayesian Uncertainty Quantification in Deep Probabilistic Image Segmentation25 Nov 2024
- 2024Rapid Review of Generative AI in Smart Medical Applications8 Jun 2024
- 2024Neural Optimal Transport with Lagrangian Costs1 Jun 2024
- 2024A Sinkhorn-type Algorithm for Constrained Optimal Transport8 Mar 2024
- 2024Accelerating Sinkhorn Algorithm with Sparse Newton Iterations20 Jan 2024
12 newest of 37 All 37 (JSON)