COFM: Consistent Optimal Transport Flow Matching via Partially Input Convex Neural Networks

Fanghui Song, Zhongjian Wang, Jiebao Sun

Optimal transport (OT) provides a principled framework for learning mappings between probability distributions, and has found broad applications in generative modeling, inverse pro…