LLM2D
图变换的离散扩散薛定谔桥匹配
Discrete Diffusion Schr\"odinger Bridge Matching for Graph Transformation
作者: Jun Hyeong Kim, Seonghwan Kim, Seokhyun Moon, Hyeongwoo Kim, Jeheon Woo, Woo Youn Kim
发布日期: 10/3/2024
arXiv ID: oai:arXiv.org:2410.01500v1

摘要

在生成式模型中,跨任意分布的传输是一个基本目标。最近提出的扩散桥模型提供了一种潜在的解决方案,但它们依赖于一个在实践中难以获得的联合分布。此外,基于连续域的公式限制了它们在离散域(如图)中的适用性。为了克服这些限制,我们提出了离散扩散薛定谔桥匹配(DDSBM),这是一种利用连续时间马尔可夫链在高维离散状态空间中解决 SB 问题的新框架。我们的方法将迭代马尔可夫拟合扩展到离散域,并且我们已经证明了它收敛到 SB。此外,我们针对图变换调整了我们的框架,并表明我们对节点和边独立修改的底层动力学的設計选择可以被解释为具有由图编辑距离描述的成本函数的最优传输的熵正则化版本。为了证明我们框架的有效性,我们将 DDSBM 应用于化学领域的分子优化。实验结果表明,DDSBM 有效地优化了分子的目标特性,并最小化了图变换,成功地保留了其他特征。