Improving GANs Using Optimal Transport
Not on the map. Why?
Abstract (arXiv)
We present Optimal Transport GAN (OT-GAN), a variant of generative adversarial nets minimizing a new metric measuring the distance between the generator distribution and the data distribution. This metric, which we call mini-batch energy distance, combines optimal transport in primal form with an energy distance defined in an adversarially learned feature space, resulting in a highly discriminative distance function with unbiased mini-batch gradients. Experimentally we show OT-GAN to be highly stable when trained with large mini-batches, and we present state-of-the-art results on several popular benchmark problems for image generation.
Key facts. Cited by 122 papers in the map (data through 7 Oct 2026). First posted on arXiv 15 Mar 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 2; 2021: 27; 2022: 27; 2023: 28; 2024: 17; 2025: 17; 2026: 4.
122 papers.
Sort:
By area:
- 2021Cross-Modal Contrastive Learning for Text-to-Image Generationcited by 259
- 2023SE(3)-Stochastic Flow Matching for Protein Backbone Generationcited by 147
- 2023Ambiguous Medical Image Segmentation using Diffusion Modelscited by 92
- 2021Proximal Optimal Transport Modeling of Population Dynamicscited by 91
- 2023Simulation-free Schr"odinger bridges via score and flow matchingcited by 81
- 2022PLOT: Prompt Learning with Optimal Transport for Vision-Language Modelscited by 64
- 2021Minibatch optimal transport distances; analysis and applicationscited by 62
- 2022Generative Multiplane Images: Making a 2D GAN 3D-Awarecited by 57
- 2021Don't Generate Me: Training Differentially Private Generative Models with Sinkhorn Divergencecited by 54
- 2021Rates of Estimation of Optimal Transport Maps using Plug-in Estimators via Barycentric Projectionscited by 52
- 2023Effective Neural Topic Modeling with Embedding Clustering Regularizationcited by 46
- 2022Unbalanced Optimal Transport, from Theory to Numericscited by 45
38 more
- 2021Improving Mini-batch Optimal Transport via Partial Transportationcited by 40
- 2023SoK: Privacy-Preserving Data Synthesiscited by 37
- 2021A Survey on Optimal Transport for Machine Learning: Theory and Applicationscited by 35
- 2021Data Efficient Language-supervised Zero-shot Recognition with Optimal Transport Distillationcited by 35
- 2024FASTopic: Pretrained Transformer is a Fast, Adaptive, Stable, and Transferable Topic Modelcited by 34
- 2023The Monge Gap: A Regularizer to Learn All Transport Mapscited by 32
- 2020Imitation Learning with Sinkhorn Distancescited by 30
- 2022A Unified Framework for Implicit Sinkhorn Differentiationcited by 28
- 2024AWT: Transferring Vision-Language Models via Augmentation, Weighting, and Transportationcited by 27
- 2025Distributional Diffusion Models with Scoring Rulescited by 25
- 2022A Survey on Training Challenges in Generative Adversarial Networks for Biomedical Image Analysiscited by 24
- 2022Hierarchical Sliced Wasserstein Distancecited by 23
- 2023Quasi-Monte Carlo for 3D Sliced Wassersteincited by 22
- 2022Optimal transport map estimation in general function spacescited by 20
- 2023Sampling via Gradient Flows in the Space of Probability Measurescited by 20
- 2024Tyche: Stochastic In-Context Learning for Medical Image Segmentationcited by 20
- 2026Sinkhorn-Drifting Generative Modelscited by 20
- 2022AADG: Automatic Augmentation for Domain Generalization on Retinal Image Segmentationcited by 18
- 2024On the Affinity, Rationality, and Diversity of Hierarchical Topic Modelingcited by 18
- 2021Input Convex Gradient Networkscited by 17
- 2022Amortized Projection Optimization for Sliced Wasserstein Generative Modelscited by 17
- 2022Taming Hyperparameter Tuning in Continuous Normalizing Flows Using the JKO Schemecited by 17
- 2026Generative Drifting is Secretly Score Matching: a Spectral and Variational Perspectivecited by 17
- 2021Large-scale optimal transport map estimation using projection pursuitcited by 16
- 2023Monge, Bregman and Occam: Interpretable Optimal Transport in High-Dimensions with Feature-Sparse Mapscited by 16
- 2023Performance Scaling via Optimal Transport: Enabling Data Selection from Partially Revealed Sourcescited by 16
- 2023OST: Refining Text Knowledge with Optimal Spatio-Temporal Descriptor for General Video Recognitioncited by 16
- 2021On Transportation of Mini-batches: A Hierarchical Approachcited by 15
- 2023Improving Cross-lingual Information Retrieval on Low-Resource Languages via Optimal Transport Distillationcited by 15
- 2024Sliced Wasserstein with Random-Path Projecting Directionscited by 15
- 2022Rethinking Initialization of the Sinkhorn Algorithmcited by 14
- 2021Entropic Gromov-Wasserstein between Gaussian Distributionscited by 12
- 2021Near-optimal estimation of smooth transport maps with kernel sums-of-squarescited by 12
- 2023Markovian Sliced Wasserstein Distances: Beyond Independent Projectionscited by 11
- 2023Flow Factorized Representation Learningcited by 11
- 2023Sample Complexity Bounds for Estimating Probability Divergences under Invariancescited by 11
- 2021Re-evaluating Word Mover's Distancecited by 10
- 2021Entropy Regularized Optimal Transport Independence Criterioncited by 9
50 most cited of 122 All 122 (JSON)
- 2026RW-Flow: One-Step Generation on Compact Manifolds via Wasserstein Gradient Flows30 Sep 2026
- 2026One-Step Generative Modeling via Unbalanced Optimal Transport26 Sep 2026
- 2026Sinkhorn-Drifting Generative Models12 Mar 2026
- 2026Generative Drifting is Secretly Score Matching: a Spectral and Variational Perspective10 Mar 2026
- 2025A Multi-domain Image Translative Diffusion StyleGAN for Iris Presentation Attack Detection16 Oct 2025
- 2025Federated Flow Matching25 Sep 2025
- 2025Generic Event Boundary Detection via Denoising Diffusion16 Aug 2025
- 2025SoK: Can Synthetic Images Replace Real Data? A Survey of Utility and Privacy of Synthetic Image Generation24 Jun 2025
- 2025Joint stochastic localization and applications19 May 2025
- 2025Fair Clustering via Alignment14 May 2025
- 2025Tree-Sliced Wasserstein Distance with Nonlinear Projection2 May 2025
- 2025Embedding Empirical Distributions for Computing Optimal Transport Maps24 Apr 2025
38 more
- 2025Collective Learning Mechanism based Optimal Transport Generative Adversarial Network for Non-parallel Voice Conversion18 Apr 2025
- 2025Towards generalizable single-cell perturbation modeling via the Conditional Monge Gap11 Apr 2025
- 2025FACT: Multinomial Misalignment Classification for Point Cloud Registration9 Apr 2025
- 2025Distance-Based Tree-Sliced Wasserstein Distance14 Mar 2025
- 2025Linear Bandits beyond Inner Product Spaces, the case of Bandit Optimal Transport11 Feb 2025
- 2025PINS: Proximal Iterations with Sparse Newton and Sinkhorn for Optimal Transport6 Feb 2025
- 2025Distributional Diffusion Models with Scoring Rules4 Feb 2025
- 2025A User's Guide to Sampling Strategies for Sliced Optimal Transport4 Feb 2025
- 2025Enhancing Unsupervised Graph Few-shot Learning via Set Functions and Optimal Transport10 Jan 2025
- 2024Auditing and Enforcing Conditional Fairness via Optimal Transport17 Oct 2024
- 2024Optimal Transport for Probabilistic Circuits16 Oct 2024
- 2024Explanatory Model Monitoring to Understand the Effects of Feature Shifts on Performance24 Aug 2024
- 2024Plug-in estimation of Schr"odinger bridges21 Aug 2024
- 2024AWT: Transferring Vision-Language Models via Augmentation, Weighting, and Transportation5 Jul 2024
- 2024Tree-Sliced Wasserstein Distance: A Geometric Perspective19 Jun 2024
- 2024FASTopic: Pretrained Transformer is a Fast, Adaptive, Stable, and Transferable Topic Model28 May 2024
- 2024A New Robust Partial p-Wasserstein-Based Metric for Comparing Distributions6 May 2024
- 2024Gaussian-Smoothed Sliced Probability Divergences4 Apr 2024
- 2024A Sinkhorn-type Algorithm for Constrained Optimal Transport8 Mar 2024
- 2024Align Your Intents: Offline Imitation Learning via Optimal Transport20 Feb 2024
- 2024PQMass: Probabilistic Assessment of the Quality of Generative Models using Probability Mass Estimation6 Feb 2024
- 2024Stochastic Modified Flows for Riemannian Stochastic Gradient Descent2 Feb 2024
- 2024Sliced Wasserstein with Random-Path Projecting Directions29 Jan 2024
- 2024On the Affinity, Rationality, and Diversity of Hierarchical Topic Modeling25 Jan 2024
- 2024Tyche: Stochastic In-Context Learning for Medical Image Segmentation24 Jan 2024
- 2024Accelerating Sinkhorn Algorithm with Sparse Newton Iterations20 Jan 2024
- 2023OST: Refining Text Knowledge with Optimal Spatio-Temporal Descriptor for General Video Recognition30 Nov 2023
- 2023A High-Quality Robust Diffusion Framework for Corrupted Dataset28 Nov 2023
- 2023Structured Transforms Across Spaces with Cost-Regularized Optimal Transport9 Nov 2023
- 2023Sample Complexity Bounds for Estimating Probability Divergences under Invariances6 Nov 2023
- 2023The SVHN Dataset Is Deceptive for Probabilistic Generative Models Due to a Distribution Mismatch30 Oct 2023
- 2023A Note on Generalization in Variational Autoencoders: How Effective Is Synthetic Data & Overparameterization?30 Oct 2023
- 2023A Specialized Semismooth Newton Method for Kernel-Based Optimal Transport21 Oct 2023
- 2023On Unsupervised Image-to-image translation and GAN stability18 Oct 2023
- 2023Sampling via Gradient Flows in the Space of Probability Measures5 Oct 2023
- 2023SE(3)-Stochastic Flow Matching for Protein Backbone Generation3 Oct 2023
- 2023Structural Adversarial Objectives for Self-Supervised Representation Learning30 Sep 2023
- 2023Flow Factorized Representation Learning22 Sep 2023
50 newest of 122 All 122 (JSON)
- 2023SE(3)-Stochastic Flow Matching for Protein Backbone Generationcited by 147
- 2023Ambiguous Medical Image Segmentation using Diffusion Modelscited by 92
- 2023Simulation-free Schr"odinger bridges via score and flow matchingcited by 81
- 2025Distributional Diffusion Models with Scoring Rulescited by 25
- 2025Generic Event Boundary Detection via Denoising Diffusioncited by 2
- 2025Federated Flow Matchingcited by 0
- 2025Federated Flow Matching25 Sep 2025
- 2025Generic Event Boundary Detection via Denoising Diffusion16 Aug 2025
- 2025Distributional Diffusion Models with Scoring Rules4 Feb 2025
- 2023SE(3)-Stochastic Flow Matching for Protein Backbone Generation3 Oct 2023
- 2023Simulation-free Schr"odinger bridges via score and flow matching7 Jul 2023
- 2023Ambiguous Medical Image Segmentation using Diffusion Models10 Apr 2023
- 2022PLOT: Prompt Learning with Optimal Transport for Vision-Language Modelscited by 64
- 2024AWT: Transferring Vision-Language Models via Augmentation, Weighting, and Transportationcited by 27
- 2023OST: Refining Text Knowledge with Optimal Spatio-Temporal Descriptor for General Video Recognitioncited by 16
- 2024AWT: Transferring Vision-Language Models via Augmentation, Weighting, and Transportation5 Jul 2024
- 2023OST: Refining Text Knowledge with Optimal Spatio-Temporal Descriptor for General Video Recognition30 Nov 2023
- 2022PLOT: Prompt Learning with Optimal Transport for Vision-Language Models3 Oct 2022
- 2024Align Your Intents: Offline Imitation Learning via Optimal Transport20 Feb 2024
- 2020Imitation Learning with Sinkhorn Distances20 Aug 2020
- 2020Nested-Wasserstein Self-Imitation Learning for Sequence Generation20 Jan 2020
- 2021Cross-Modal Contrastive Learning for Text-to-Image Generationcited by 259
- 2021Proximal Optimal Transport Modeling of Population Dynamicscited by 91
- 2021Minibatch optimal transport distances; analysis and applicationscited by 62
- 2022Generative Multiplane Images: Making a 2D GAN 3D-Awarecited by 57
- 2021Don't Generate Me: Training Differentially Private Generative Models with Sinkhorn Divergencecited by 54
- 2021Rates of Estimation of Optimal Transport Maps using Plug-in Estimators via Barycentric Projectionscited by 52
- 2023Effective Neural Topic Modeling with Embedding Clustering Regularizationcited by 46
- 2022Unbalanced Optimal Transport, from Theory to Numericscited by 45
- 2021Improving Mini-batch Optimal Transport via Partial Transportationcited by 40
- 2023SoK: Privacy-Preserving Data Synthesiscited by 37
- 2021A Survey on Optimal Transport for Machine Learning: Theory and Applicationscited by 35
- 2021Data Efficient Language-supervised Zero-shot Recognition with Optimal Transport Distillationcited by 35
12 most cited of 110 All 110 (JSON)
- 2026RW-Flow: One-Step Generation on Compact Manifolds via Wasserstein Gradient Flows30 Sep 2026
- 2026One-Step Generative Modeling via Unbalanced Optimal Transport26 Sep 2026
- 2026Sinkhorn-Drifting Generative Models12 Mar 2026
- 2026Generative Drifting is Secretly Score Matching: a Spectral and Variational Perspective10 Mar 2026
- 2025A Multi-domain Image Translative Diffusion StyleGAN for Iris Presentation Attack Detection16 Oct 2025
- 2025SoK: Can Synthetic Images Replace Real Data? A Survey of Utility and Privacy of Synthetic Image Generation24 Jun 2025
- 2025Joint stochastic localization and applications19 May 2025
- 2025Fair Clustering via Alignment14 May 2025
- 2025Tree-Sliced Wasserstein Distance with Nonlinear Projection2 May 2025
- 2025Embedding Empirical Distributions for Computing Optimal Transport Maps24 Apr 2025
- 2025Collective Learning Mechanism based Optimal Transport Generative Adversarial Network for Non-parallel Voice Conversion18 Apr 2025
- 2025Towards generalizable single-cell perturbation modeling via the Conditional Monge Gap11 Apr 2025
12 newest of 110 All 110 (JSON)