Date
Mon, 09 Nov 2026
Time
15:30 - 16:30
Location
L3
Speaker
RenYuan XU
Organisation
Stanford University
Add to calendar

Diffusion models and flow-based methods have achieved strong results in image generation, but often rely on costly iterative sampling. We introduce W-Flow, a framework that compresses a Wasserstein gradient flow into a one-step neural generator. By minimizing the Sinkhorn divergence, W-Flow transports a reference distribution toward the data distribution through efficient optimal-transport updates that capture global distributional discrepancies. We prove that, under suitable assumptions, the finite-sample training dynamics converge to the continuous-time distributional dynamics. Empirically, W-Flow sets a new state of the art in one-step ImageNet generation, with improved mode coverage and domain transfer. We will also discuss extensions using alternative energy functionals and applications to post-training.


 

Last updated on 29 Sep 2026, 10:34am. Please contact us with feedback and comments about this page.