Discrete Wasserstein Flows for One-Step Generative Modeling

Excerpt

arXiv:2610.01355v1 Announce Type: new Abstract: We introduce a new framework for one-step generative modelling on finite state spaces. To extend drifting beyond continuous domains, we use discrete Wasserstein geometry to define a target-relative KL gradient flow over the transitions of a reversible Markov kernel. We realize this probability flow at the particle level through Markov jumps and amortize the resulting transport updates into a latent-conditioned generator, so that the iterative dynam