This model is trained on ImageNet-32x32 text-to-image with conditional minibatch optimal transport data coupling.

Paper: The Curse of Conditions: Analyzing and Improving Optimal Transport for Conditional Flow-Based Generation Project Page: https://hkchengrex.github.io/C2OT Code: https://github.com/hkchengrex/C2OT

C²OT is an algorithm for computing prior-to-data couplings for flow-matching-based generative models during training. Our goal is to achieve straighter flows, enabled by optimal transport (OT) couplings, while mitigating the test-time degradation that OT encounters in the conditional setting. The key idea is that OT samples from a condition-skewed prior distribution at test time, whereas C²OT unskews the prior by incorporating a condition-dependent term into the OT cost.

See the GitHub repo: https://github.com/hkchengrex/C2OT

If you use C$^2$OT in your research, please cite the original paper: