A Jump-Modified Wasserstein Gradient Flow for Maximum Mean Discrepancy with Energy Kernels

Ansh Desai, Jason Liu, Srujan Roplekar, Matthew Rosenzweig
10/21/2026
math.APcs.LGmath.PR

No confident code implementation found for this paper.

Abstract

We study a jump-modified Wasserstein gradient flow of the squared maximum mean discrepancy generated by the energy kernels $K(z)=-|z|^q$, $0<q<2$, in every dimension $d\ge2$ as well as the one-dimensional range $1\le q<2$. Under subcritical density and finite-moment assumptions, we prove global well-posedness. The added jumps transfer mass toward smaller values of the first variation and dissipate the same MMD energy as the transport term. Under our hypotheses on the target and jump rates, every Borel probability measure with finite MMD energy and zero jump dissipation is equal to the target. Combining this rigidity with compactness and a second-moment identity yields convergence to the target from finite-second-moment data, without additional uniform-in-time density or moment assumptions. We also construct a global collision-free particle process whose energy is nonincreasing along each path. A quantitative mean-field estimate applies to general initial empirical approximations, including deterministic configurations; propagation of chaos for exchangeable data and long-time target approximation follow as consequences.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{desai2026jumpmodified,
  title  = {A Jump-Modified Wasserstein Gradient Flow for Maximum Mean Discrepancy with Energy Kernels},
  author = {Ansh Desai and Jason Liu and Srujan Roplekar and Matthew Rosenzweig},
  year   = {2026},
  eprint = {2610.29007},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2610.29007v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2610.29007.svg)](https://codeofpaper.com/paper/2610.29007)

Discussion