arXiv stat.ML· Soon Hoe Lim, Aditi Gupta, Michael W. Mahoney, N. Benjamin Erichson·· 16 小时前AI 评分34
面向 Flow Matching 无展开后训练的粗粒化最优传输
Coarse-Graining Optimal Transport for Rollout-Free Post-Training of Flow Matching
AI 导读
研究团队提出 ReCouple 框架，通过构建粗粒化最优传输（OT）耦合，实现 Flow Matching 模型无需模拟展开（rollout-free）的后训练。该框架进一步拓展出逼近半离散 OT 的 ReCouple+，以及加入空间正则化的 ReCouple++。实验表明，该方法有效优化了低函数评估次数（NFE）下的生成质量与速度场。
来源：arXiv stat.ML · arxiv.org