LLM2D
可扩展的无模拟熵不平衡最优传输
Scalable Simulation-free Entropic Unbalanced Optimal Transport
作者: Jaemoo Choi, Jaewoong Choi
发布日期: 10/4/2024
arXiv ID: oai:arXiv.org:2410.02656v1

摘要

最优传输 (OT) 问题研究连接两个分布的传输映射,同时最小化给定的成本函数。找到这样的传输映射在机器学习中具有广泛的应用,例如生成式建模和图像到图像的转换。本文提出了一种可扩展且无模拟的方法来解决熵均衡最优传输 (EUOT) 问题。我们推导出 EUOT 问题的动力学形式,它是薛定谔桥 (SB) 问题的推广。在此基础上,我们从随机最优控制解释推导出 EUOT 问题的对偶公式和最优性条件。利用这些性质,我们提出了一种无模拟算法来解决 EUOT,称为无模拟 EUOT (SF-EUOT)。虽然现有的 SB 模型在训练和评估过程中需要昂贵的模拟成本,但我们的模型通过利用互易性实现了无模拟训练和一步生成。与之前的 SB 方法相比,我们的模型在生成式建模和图像到图像的转换任务中展现出显著的扩展性提升。