AlignFlow: Improving Flow-based Generative Models with Semi-Discrete Optimal Transport

Lingkai Kong, Molei Tao, Yang Liu +4

Flow-based Generative Models (FGMs) effectively transform noise into complex data distributions. Incorporating Optimal Transport (OT) to couple noise and data during FGM training h…