Wasserstein Gradient

Wasserstein gradient flows are a powerful tool for optimizing over probability distributions, offering a geometrically informed approach to problems where traditional gradient descent methods are insufficient. Current research focuses on applying these flows in diverse areas, including generative modeling (e.g., using diffusion models and adversarial networks), variational inference (with Gaussian mixtures and particle-based methods), and robust machine learning (e.g., for handling noisy data and outliers). This framework's significance lies in its ability to address challenges in high-dimensional spaces and complex data structures, leading to improved performance in various applications such as image generation, causal inference, and Bayesian inference.

Papers