Discrete Wasserstein Flows for One-Step Generative Modeling
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…
We haven't written up this one. arXiv cs.AI has the full story — the link below goes straight to it.
