Tengyuan Liang

No-regret generative modeling via parabolic Monge–Ampère PDE

We introduce a novel generative modeling framework based on a discretized parabolic Monge–Ampère PDE, which emerges as a continuous limit of the Sinkhorn algorithm commonly used in optimal transport. Our method performs iterative refinement in the space of Brenier maps using a mirror gradient descent step. We establish...