Wasserstein Variational Inference
Not on the map. Why?
Abstract (arXiv)
This paper introduces Wasserstein variational inference, a new form of approximate Bayesian inference based on optimal transport theory. Wasserstein variational inference uses a new family of divergences that includes both f-divergences and the Wasserstein distance as special cases. The gradients of the Wasserstein variational loss are obtained by backpropagating through the Sinkhorn iterations. This technique results in a very stable likelihood-free training method that can be used with implicit distributions and probabilistic programs. Using the Wasserstein variational inference framework, we introduce several new forms of autoencoders and test their robustness and performance against existing variational autoencoding techniques.
Key facts. Cited by 17 papers in the map (data through 14 Oct 2026). First posted on arXiv 29 May 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 5; 2022: 2; 2023: 4; 2024: 1; 2025: 3; 2026: 1.
17 papers.
Sort:
By area:
- 2021A Survey on Optimal Transport for Machine Learning: Theory and Applicationscited by 35
- 2021Wasserstein Unsupervised Reinforcement Learningcited by 22
- 2020f-GAIL: Learning f-Divergence for Generative Adversarial Imitation Learningcited by 20
- 2022Sliced Wasserstein Variational Inferencecited by 18
- 2023On the Convergence of Coordinate Ascent Variational Inferencecited by 18
- 2022Gromov-Wasserstein Autoencoderscited by 7
- 2021Bayesian brains and the R'enyi divergencecited by 6
- 2021Evaluating Sensitivity to the Stick-Breaking Prior in Bayesian Nonparametricscited by 4
- 2023Hyperbolic Geometry in Computer Vision: A Surveycited by 4
- 2025Deep Generative Models: Complexity, Dimensionality, and Approximationcited by 4
- 2021Bayesian Distributional Policy Gradientscited by 3
- 2024A Primer on Variational Inference for Physics-Informed Deep Generative Modellingcited by 3
5 more
- 2025Bayesian Experimental Design for Model Discrepancy Calibration: An Auto-Differentiable Ensemble Kalman Inversion Approachcited by 2
- 2023Parameter Estimation in DAGs from Incomplete Data via Optimal Transportcited by 1
- 2023Nonlinear Kalman Filtering with Reparametrization Gradientscited by 0
- 2025Streaming Sliced Optimal Transportcited by 0
- 2026Bayesian Experimental Design for Model Discrepancy Calibration: A Rivalry between Kullback--Leibler Divergence and Wasserstein Distancecited by 0
- 2026Bayesian Experimental Design for Model Discrepancy Calibration: A Rivalry between Kullback--Leibler Divergence and Wasserstein Distance23 Jan 2026
- 2025Streaming Sliced Optimal Transport11 May 2025
- 2025Bayesian Experimental Design for Model Discrepancy Calibration: An Auto-Differentiable Ensemble Kalman Inversion Approach29 Apr 2025
- 2025Deep Generative Models: Complexity, Dimensionality, and Approximation1 Apr 2025
- 2024A Primer on Variational Inference for Physics-Informed Deep Generative Modelling10 Sep 2024
- 2023On the Convergence of Coordinate Ascent Variational Inference1 Jun 2023
- 2023Parameter Estimation in DAGs from Incomplete Data via Optimal Transport25 May 2023
- 2023Hyperbolic Geometry in Computer Vision: A Survey21 Apr 2023
- 2023Nonlinear Kalman Filtering with Reparametrization Gradients8 Mar 2023
- 2022Gromov-Wasserstein Autoencoders15 Sep 2022
- 2022Sliced Wasserstein Variational Inference26 Jul 2022
- 2021Wasserstein Unsupervised Reinforcement Learning15 Oct 2021
5 more
- 2021Bayesian brains and the R'enyi divergence12 Jul 2021
- 2021Evaluating Sensitivity to the Stick-Breaking Prior in Bayesian Nonparametrics8 Jul 2021
- 2021A Survey on Optimal Transport for Machine Learning: Theory and Applications3 Jun 2021
- 2021Bayesian Distributional Policy Gradients20 Mar 2021
- 2020f-GAIL: Learning f-Divergence for Generative Adversarial Imitation Learning2 Oct 2020
- 2021A Survey on Optimal Transport for Machine Learning: Theory and Applicationscited by 35
- 2021Wasserstein Unsupervised Reinforcement Learningcited by 22
- 2022Sliced Wasserstein Variational Inferencecited by 18
- 2023On the Convergence of Coordinate Ascent Variational Inferencecited by 18
- 2022Gromov-Wasserstein Autoencoderscited by 7
- 2021Bayesian brains and the R'enyi divergencecited by 6
- 2021Evaluating Sensitivity to the Stick-Breaking Prior in Bayesian Nonparametricscited by 4
- 2023Hyperbolic Geometry in Computer Vision: A Surveycited by 4
- 2025Deep Generative Models: Complexity, Dimensionality, and Approximationcited by 4
- 2021Bayesian Distributional Policy Gradientscited by 3
- 2024A Primer on Variational Inference for Physics-Informed Deep Generative Modellingcited by 3
- 2025Bayesian Experimental Design for Model Discrepancy Calibration: An Auto-Differentiable Ensemble Kalman Inversion Approachcited by 2
12 most cited of 16 All 16 (JSON)
- 2026Bayesian Experimental Design for Model Discrepancy Calibration: A Rivalry between Kullback--Leibler Divergence and Wasserstein Distance23 Jan 2026
- 2025Streaming Sliced Optimal Transport11 May 2025
- 2025Bayesian Experimental Design for Model Discrepancy Calibration: An Auto-Differentiable Ensemble Kalman Inversion Approach29 Apr 2025
- 2025Deep Generative Models: Complexity, Dimensionality, and Approximation1 Apr 2025
- 2024A Primer on Variational Inference for Physics-Informed Deep Generative Modelling10 Sep 2024
- 2023On the Convergence of Coordinate Ascent Variational Inference1 Jun 2023
- 2023Parameter Estimation in DAGs from Incomplete Data via Optimal Transport25 May 2023
- 2023Hyperbolic Geometry in Computer Vision: A Survey21 Apr 2023
- 2023Nonlinear Kalman Filtering with Reparametrization Gradients8 Mar 2023
- 2022Gromov-Wasserstein Autoencoders15 Sep 2022
- 2022Sliced Wasserstein Variational Inference26 Jul 2022
- 2021Wasserstein Unsupervised Reinforcement Learning15 Oct 2021
12 newest of 16 All 16 (JSON)