【ICML2024】双曲几何潜在扩散模型用于图生成

数据派THU

共 636字,需浏览 2分钟

 ·

2024-05-15 17:00

   
来源:专知
本文为论文介绍,建议阅读5分钟
我们提出了一种新颖的几何潜在扩散框架 HypDiff。

扩散模型在计算机视觉领域取得了重大进展,最近在社区内引发了对其在图生成中的应用的广泛兴趣。现有的离散图扩散模型表现出较高的计算复杂性和较低的训练效率。直接在潜在空间中扩散图是一种更优越且自然的方式。然而,由于图的非欧几里得结构在潜在空间中并不各向同性,现有的潜在扩散模型难以有效捕获和保留图的拓扑信息。
为了解决上述挑战,我们提出了一种新颖的几何潜在扩散框架 HypDiff。具体而言,我们首先基于双曲几何建立一个具有可解释性度量的几何潜在空间,以定义图的各向异性潜在扩散过程。然后,我们提出了一种受径向和角度几何属性约束的几何潜在扩散过程,从而确保在生成的图中保留原始拓扑属性。大量实验结果证明了 HypDiff 在生成具有不同拓扑结构的图方面的卓越效果。


浏览 16
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报