每日 Arxiv

本页面整理了世界各地发表的人工智能相关论文。
本页面使用 Google Gemini 汇总而成,并以非盈利为基础运营。
论文版权归作者及相关机构所有,分享时请注明出处。

用于图形生成的 Bures-Wasserstein 流匹配

Created by
  • Haebom

作者

蒋克跃、崔家豪、董晓文、劳拉·托尼

大纲

图生成已成为药物研发和电路设计等各个领域的一项关键挑战。现有方法,尤其是基于扩散和流的模型,通过构建在参考分布和数据分布之间进行插值的概率路径来实现稳健的图生成性能。然而,这些方法通常分别对单个节点和边的演化进行建模,并使用线性插值来构建路径。这种单独的插值会破坏图的互连模式,使构建的概率路径变得不规则且不平滑,从而导致学习动态性能差和采样收敛性差。为了解决这个问题,本文提出了一个理论上合理的框架,用于在图生成模型中构建概率路径。具体而言,我们将图表示为一个由马尔可夫随机场 (MRF) 参数化的连通系统,以模拟节点和边的共同演化。然后,我们利用 MRF 对象之间的最优传输位移来设计平滑的概率路径,以确保图组件的共同演化。在此基础上,我们提出了 BWFlow,这是一个流一致的图生成框架,它利用导出的最优概率路径来改进学习和采样算法的设计。在通用图生成和分子生成方面的实验评估验证了 BWFlow 的有效性,展现了其优异的性能、更好的学习收敛性和高效的采样能力。

Takeaways, Limitations

Takeaways:
图生成模型中构建概率路径的理论框架。
使用 MRF 对节点和边的共同进化进行建模。
使用最优运输位移的平滑概率路线设计
BWFlow,一个提议的流匹配框架
表现出竞争性的表现、更好的学习融合和高效的采样。
Limitations:
论文中没有明确提及所提出的框架和 BWFlow 的具体 Limitations(需要进一步研究)
需要有关实验评估的具体细节和结果的更多信息。
👍