ResearchResearch paperTraining & Scaling · Efficiency & Inference · Reinforcement Learning1 source · Oct 21, 2026

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

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$.

Key points

  • Under subcritical density and finite-moment assumptions, we prove global well-posedness.
  • 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.
  • 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.

Sources (1)

  • [1]A Jump-Modified Wasserstein Gradient Flow for Maximum Mean Discrepancy with Energy Kernels
    arXiv (AI, ML, NLP, CV, robotics, multi-agent) · Oct 21, 12:16 AM
    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.

Extractive summary: sentences quoted from the sources.

Before this

  1. Oct 20, 2026Fine-tuning GPT-OSS-20B for multilingual reasoning on Amazon SageMaker HyperPod
  2. Oct 20, 2026Build a semantic layer for agentic AI on AWS with Denodo and Amazon Bedrock AgentCore
  3. Oct 20, 2026Toward Agentic AI for the 6G Physical Layer: From Algorithm Discovery to Verified Baseband Silicon
  4. Oct 19, 2026How Deepgram optimized voice AI at scale with AWS Trainium
  5. Oct 19, 2026Push-based KV transfer to improve TTFT and TPOT for disaggregated inference in vLLM
  6. Oct 19, 2026jax-ml/jax jax-v0.12.0: JAX v0.12.0

Related