arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2412.03487cs.LGcs.AI
收藏

基于通用离散路径的 Flow Matching:动力学最优视角

Flow Matching with General Discrete Paths: A Kinetic-Optimal Perspective

Weizmann Institute of Science(魏茨曼科学研究所);Meta FAIR;MIT CSAIL(麻省理工学院计算机科学与人工智能实验室)

Neta Shaul, Itai Gat, Marton Havasi, Daniel Severo, Anuroop Sriram, Peter Holderrieth, Brian Karrer, Yaron Lipman, Ricky T. Q. Chen

AI总结:

本文从对称动能优化出发,为连续时间 Markov 链离散流模型提出适配任意概率路径的速度公式,并证明混合路径的动力学最优性,在多模态生成中优于 mask 构造。

摘要:

与连续空间中的扩散或流生成模型相比,离散空间中这类模型的设计空间仍远未得到充分理解,许多工作只关注简单的 mask 构造。本文旨在以整体性方法构建基于连续时间 Markov 链的离散生成模型,并首次允许使用任意离散概率路径,通俗而言即任意损坏过程。通过优化对称动能的视角,我们提出了可应用于任意给定概率路径的速度公式,使概率与速度完全解耦,从而让用户能够依据特定数据领域的专家知识指定任何理想的概率路径。此外,我们发现一种特殊的混合概率路径构造能够在离散情形下优化对称动能。我们在文本生成、无机材料生成和图像生成等多种模态上通过实验验证了这一新设计空间的实用性。结果表明,即使在文本任务中,采用动力学最优的混合路径也能优于 mask 构造;而在视觉领域,则可利用面向特定领域的概率路径构造。

英文摘要:

The design space of discrete-space diffusion or flow generative models are significantly less well-understood than their continuous-space counterparts, with many works focusing only on a simple masked construction. In this work, we aim to take a holistic approach to the construction of discrete generative models based on continuous-time Markov chains, and for the first time, allow the use of arbitrary discrete probability paths, or colloquially, corruption processes. Through the lens of optimizing the symmetric kinetic energy, we propose velocity formulas that can be applied to any given probability path, completely decoupling the probability and velocity, and giving the user the freedom to specify any desirable probability path based on expert knowledge specific to the data domain. Furthermore, we find that a special construction of mixture probability paths optimizes the symmetric kinetic energy for the discrete case. We empirically validate the usefulness of this new design space across multiple modalities: text generation, inorganic material generation, and image generation. We find that we can outperform the mask construction even in text with kinetic-optimal mixture paths, while we can make use of domain-specific constructions of the probability path over the visual domain.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑