Maximum Mean Discrepancy Gradient Flow
Not on the map. Why?
Abstract (arXiv)
We construct a Wasserstein gradient flow of the maximum mean discrepancy (MMD) and study its convergence properties. The MMD is an integral probability metric defined for a reproducing kernel Hilbert space (RKHS), and serves as a metric on probability measures for a sufficiently rich RKHS. We obtain conditions for convergence of the gradient flow towards a global optimum, that can be related to particle transport when optimizing neural networks. We also propose a way to regularize this MMD flow, based on an injection of noise in the gradient. This algorithmic fix comes with theoretical and empirical evidence. The practical implementation of the flow is straightforward, since both the MMD and its gradient have simple closed-form expressions, which can be easily estimated with samples.
Key facts. Cited by 123 papers in the map (data through 7 Oct 2026). First posted on arXiv 11 Jun 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 14; 2022: 9; 2023: 26; 2024: 25; 2025: 27; 2026: 22.
123 papers.
Sort:
By area:
- 2021Deep Generative Learning via Schr"odinger Bridgecited by 103
- 2021Large-Scale Wasserstein Gradient Flowscited by 76
- 2021Kernel Stein Discrepancy Descentcited by 57
- 2021Optimizing Functionals on the Space of Probabilities with Input Convex Neural Networkscited by 53
- 2022Understanding and Mitigating Overfitting in Prompt Tuning for Vision-Language Modelscited by 40
- 2021KALE Flow: A Relaxed KL Gradient Flow for Probabilities with Disjoint Supportcited by 37
- 2023Generative Sliced MMD Flows with Riesz Kernelscited by 36
- 2022Mirror Descent with Relative Smoothness in Measure Spaces, with application to Sinkhorn and EMcited by 33
- 2023A Rigorous Link between Deep Ensembles and (Variational) Bayesian Methodscited by 31
- 2022Birth-death dynamics for sampling: Global convergence, approximations and their asymptoticscited by 30
- 2023A mean-field games laboratory for generative modelingcited by 28
- 2023Posterior Sampling Based on Gradient Flows of the MMD with Negative Distance Kernelcited by 25
38 more
- 2021A Neural Tangent Kernel Perspective of GANscited by 21
- 2021Efficient Gradient Flows in Sliced-Wasserstein Spacecited by 21
- 2023Unifying GANs and Score-Based Diffusion as Generative Particle Modelscited by 21
- 2023Convergence of mean-field Langevin dynamics: Time and space discretization, stochastic gradient, and variance reductioncited by 19
- 2024Deep MMD Gradient Flow without adversarial trainingcited by 18
- 2023Gaussian Interpolation Flowscited by 17
- 2021Stein Variational Gradient Descent: many-particle and long-time asymptoticscited by 15
- 2021Neural Tangent Kernel Maximum Mean Discrepancycited by 15
- 2026Gradient Flow Drifting: Generative Modeling via Wasserstein Gradient Flows of KDE-Approximated Divergencescited by 15
- 2024Learning Associative Memories with Gradient Descentcited by 14
- 2024A Practical Guide to Sample-based Statistical Distances for Evaluating Generative Models in Sciencecited by 14
- 2023Towards Understanding the Dynamics of Gaussian-Stein Variational Gradient Descentcited by 13
- 2024Nonparametric Teaching of Implicit Neural Representationscited by 13
- 2024Theoretical Guarantees for Variational Inference with Fixed-Variance Mixture of Gaussianscited by 11
- 2024Stein transport for Bayesian inferencecited by 11
- 2025Scale-wise Distillation of Diffusion Modelscited by 11
- 2025Integral Imprecise Probability Metricscited by 11
- 2026Drifting Fields are not Conservativecited by 11
- 2023Dynamic Flows on Curved Space Generated by Labeled Datacited by 10
- 2025Weighted quantization using MMD: From mean field to mean shift via gradient flowscited by 10
- 2023Neural Wasserstein Gradient Flows for Maximum Mean Discrepancies with Riesz Kernelscited by 9
- 2023The Score-Difference Flow for Implicit Generative Modelingcited by 9
- 2023Nonparametric Iterative Machine Teachingcited by 9
- 2023GAD-PVI: A General Accelerated Dynamic-Weight Particle-Based Variational Inference Frameworkcited by 9
- 2025A Dictionary of Closed-Form Kernel Mean Embeddingscited by 9
- 2021Generalization Error of GAN from the Discriminator's Perspectivecited by 8
- 2021Dual Training of Energy-Based Models with Overparametrized Shallow Neural Networkscited by 8
- 2022Lipschitz-regularized gradient flows and generative particle algorithms for high-dimensional scarce datacited by 8
- 2023Coin Sampling: Gradient-Based Bayesian Inference without Learning Ratescited by 8
- 2024Neural Sinkhorn Gradient Flowcited by 8
- 2023Bayesian Posterior Perturbation Analysis with Integral Probability Metricscited by 7
- 2023Nonparametric Teaching for Multiple Learnerscited by 7
- 2024Fast Kernel Summation in High Dimensions via Slicing and Fourier Transformscited by 7
- 2024Interaction-Force Transport Gradient Flowscited by 7
- 2024Particle Semi-Implicit Variational Inferencecited by 7
- 2024Importance Corrected Neural JKO Samplingcited by 7
- 2024Wasserstein Gradient Flows of MMD Functionals with Distance Kernel and Cauchy Problems on Quantile Functionscited by 7
- 2024Fast Summation of Radial Kernels via QMC Slicingcited by 7
50 most cited of 123 All 123 (JSON)
- 2026Improving Function Space Flow Matching with Kernel Optimal Transport29 Sep 2026
- 2026From Scores to Samples: Elastic Forcing for Autoregressive Video Generation28 Sep 2026
- 2026One-Step Generative Modeling via Unbalanced Optimal Transport26 Sep 2026
- 2026Sharp Convergence of Wasserstein Gradient Flows for Spectrally Nonnegative Interaction Energies22 Sep 2026
- 2026Fast Gauss Sums via Flash Attention4 Sep 2026
- 2026Fairness-Aware Network Embeddings: Methods, Applications, and Challenges19 Aug 2026
- 2026Minimax Lower Bounds of Kernel Discrepancy Estimation: MMD, HSIC, KSD27 Jul 2026
- 2026Three-Body Scattering for Generative Modeling20 Jul 2026
- 2026Wasserstein gradient flows for Coulomb discrepancies14 Jul 2026
- 2026A Gradient Flow Perspective on Minimum MMD Estimation4 Jul 2026
- 2026Difference of Convex Programming in the Wasserstein Space with Applications to MMD Optimization26 Jun 2026
- 2026Thinned Mean Field Langevin Dynamics27 May 2026
38 more
- 2026Uniform-in-Time Weak Propagation-of-Chaos in Shallow Neural Networks21 May 2026
- 2026To discretize continually: Mean shift interacting particle systems for Bayesian inference13 May 2026
- 2026Coupling-Informed Transport Maps for Bayesian Filtering in Nonlinear Dynamical Systems13 May 2026
- 2026Sobolev Regularized MMD Gradient Flow12 May 2026
- 2026On the Wasserstein Gradient Flow Interpretation of Drifting Models6 May 2026
- 2026Generalising maximum mean discrepancy: kernelised functional Bregman divergences27 Apr 2026
- 2026Drifting Fields are not Conservative7 Apr 2026
- 2026Gradient Flow Drifting: Generative Modeling via Wasserstein Gradient Flows of KDE-Approximated Divergences11 Mar 2026
- 2026Learning Adaptive Distribution Alignment with Neural Characteristic Function for Graph Domain Adaptation11 Feb 2026
- 2026Variance-Reduced Diffusion Sampling via Target Score Identity4 Jan 2026
- 2025Align & Invert: Solving Inverse Problems with Diffusion and Flow-based Models via Representation Alignment21 Nov 2025
- 2025PRESTO: Preimage-Informed Instruction Optimization for Prompting Black-Box LLMs29 Oct 2025
- 2025Sequence Modeling with Spectral Mean Flows17 Oct 2025
- 2025One-shot Conditional Sampling: MMD meets Nearest Neighbors29 Sep 2025
- 2025Structure-Attribute Transformations with Markov Chain Boost Graph Domain Adaptation25 Sep 2025
- 2025Bilateral Distribution Compression: Reducing Both Data Size and Dimensionality22 Sep 2025
- 2025Hessian-guided Perturbed Wasserstein Gradient Flows for Escaping Saddle Points21 Sep 2025
- 2025E-ROBOT: a dimension-free method for robust statistics and machine learning via Schr"odinger bridge15 Sep 2025
- 2025Cryo-EM as a Stochastic Inverse Problem5 Sep 2025
- 2025Learning Domain- and Class-Disentangled Prototypes for Domain-Generalized EEG Emotion Recognition1 Sep 2025
- 2025Data Generation without Function Estimation11 Jul 2025
- 2025Sinkhorn Normalization of Diffusion Kernels8 Jul 2025
- 2025Kernel Trace Distance: Quantum Statistical Metric between Measures through RKHS Density Operators8 Jul 2025
- 2025A note on the unique properties of the Kullback--Leibler divergence for sampling via gradient flows6 Jul 2025
- 2025Stationary MMD Points27 May 2025
- 2025Integral Imprecise Probability Metrics22 May 2025
- 2025Training Latent Diffusion Models with Interacting Particle Algorithms18 May 2025
- 2025A Dictionary of Closed-Form Kernel Mean Embeddings26 Apr 2025
- 2025Ultra-fast feature learning for the training of two-layer neural networks in the two-timescale regime25 Apr 2025
- 2025Flow Matching Ergodic Coverage24 Apr 2025
- 2025Smoothed Distance Kernels for MMDs and Applications in Wasserstein Gradient Flows10 Apr 2025
- 2025Adversarial Subspace Generation for Outlier Detection in High-Dimensional Data10 Apr 2025
- 2025Accelerating Particle-based Energetic Variational Inference4 Apr 2025
- 2025Scale-wise Distillation of Diffusion Models20 Mar 2025
- 2025DDEQs: Distributional Deep Equilibrium Models through Wasserstein Gradient Flows3 Mar 2025
- 2025Weighted quantization using MMD: From mean field to mean shift via gradient flows14 Feb 2025
- 2025Guiding Time-Varying Generative Models with Natural Gradients on Exponential Family Manifold11 Feb 2025
- 2024Linear convergence of proximal descent schemes on the Wasserstein space22 Nov 2024
50 newest of 123 All 123 (JSON)
- 2023Unifying GANs and Score-Based Diffusion as Generative Particle Modelscited by 21
- 2025Scale-wise Distillation of Diffusion Modelscited by 11
- 2023The Score-Difference Flow for Implicit Generative Modelingcited by 9
- 2025Training Latent Diffusion Models with Interacting Particle Algorithmscited by 4
- 2025Flow Matching Ergodic Coveragecited by 3
- 2023Insights into Closed-form IPM-GAN Discriminator Guidance for Diffusion Modelingcited by 0
- 2026Variance-Reduced Diffusion Sampling via Target Score Identitycited by 0
- 2026Improving Function Space Flow Matching with Kernel Optimal Transportcited by 0
- 2026Improving Function Space Flow Matching with Kernel Optimal Transport29 Sep 2026
- 2026Variance-Reduced Diffusion Sampling via Target Score Identity4 Jan 2026
- 2025Training Latent Diffusion Models with Interacting Particle Algorithms18 May 2025
- 2025Flow Matching Ergodic Coverage24 Apr 2025
- 2025Scale-wise Distillation of Diffusion Models20 Mar 2025
- 2023Insights into Closed-form IPM-GAN Discriminator Guidance for Diffusion Modeling2 Jun 2023
- 2023Unifying GANs and Score-Based Diffusion as Generative Particle Models25 May 2023
- 2023The Score-Difference Flow for Implicit Generative Modeling25 Apr 2023
- 2022Membership Inference Attacks via Adversarial Examplescited by 2
- 2022Membership Inference Attacks via Adversarial Examples27 Jul 2022
- 2021Deep Generative Learning via Schr"odinger Bridgecited by 103
- 2021Large-Scale Wasserstein Gradient Flowscited by 76
- 2021Kernel Stein Discrepancy Descentcited by 57
- 2021Optimizing Functionals on the Space of Probabilities with Input Convex Neural Networkscited by 53
- 2021KALE Flow: A Relaxed KL Gradient Flow for Probabilities with Disjoint Supportcited by 37
- 2023Generative Sliced MMD Flows with Riesz Kernelscited by 36
- 2022Mirror Descent with Relative Smoothness in Measure Spaces, with application to Sinkhorn and EMcited by 33
- 2023A Rigorous Link between Deep Ensembles and (Variational) Bayesian Methodscited by 31
- 2022Birth-death dynamics for sampling: Global convergence, approximations and their asymptoticscited by 30
- 2023A mean-field games laboratory for generative modelingcited by 28
- 2023Posterior Sampling Based on Gradient Flows of the MMD with Negative Distance Kernelcited by 25
- 2021A Neural Tangent Kernel Perspective of GANscited by 21
12 most cited of 113 All 113 (JSON)
- 2026From Scores to Samples: Elastic Forcing for Autoregressive Video Generation28 Sep 2026
- 2026One-Step Generative Modeling via Unbalanced Optimal Transport26 Sep 2026
- 2026Sharp Convergence of Wasserstein Gradient Flows for Spectrally Nonnegative Interaction Energies22 Sep 2026
- 2026Fast Gauss Sums via Flash Attention4 Sep 2026
- 2026Fairness-Aware Network Embeddings: Methods, Applications, and Challenges19 Aug 2026
- 2026Minimax Lower Bounds of Kernel Discrepancy Estimation: MMD, HSIC, KSD27 Jul 2026
- 2026Three-Body Scattering for Generative Modeling20 Jul 2026
- 2026Wasserstein gradient flows for Coulomb discrepancies14 Jul 2026
- 2026A Gradient Flow Perspective on Minimum MMD Estimation4 Jul 2026
- 2026Difference of Convex Programming in the Wasserstein Space with Applications to MMD Optimization26 Jun 2026
- 2026Thinned Mean Field Langevin Dynamics27 May 2026
- 2026Uniform-in-Time Weak Propagation-of-Chaos in Shallow Neural Networks21 May 2026
12 newest of 113 All 113 (JSON)