Feb, 2024

潜在图扩散:一种在图上生成和预测的统一框架

TL;DR我们提出了第一个框架,允许使用一个模型解决所有级别(节点、边和图)和所有类型(生成、回归和分类)的图学习任务。我们首先提出了潜在图扩散(LGD),它是一个生成模型,可以同时生成所有类别的节点、边和图级特征。我们通过将图结构和特征嵌入到潜在空间中,并利用一个强大的编码器对其进行解码,然后在潜在空间中训练扩散模型来实现这个目标。LGD还能够通过一个特殊设计的交叉注意机制进行条件生成。然后,我们将包括回归和分类在内的预测任务制定为(条件)生成,使我们的LGD能够解决所有级别和所有类型的任务,并提供可证明的保证。通过广泛的实验证明了我们框架的有效性,其中我们的模型在生成和回归任务中达到了最先进或极具竞争力的结果。