GraphGDP: Generative Diffusion Processes for Permutation Invariant Graph Generation

被引:15
作者
Huang, Han [1 ]
Sun, Leilei [1 ]
Du, Bowen [1 ]
Fu, Yanjie [2 ]
Lv, Weifeng [1 ]
机构
[1] Beihang Univ, SKLSDE, Beijing, Peoples R China
[2] Univ Cent Florida, Orlando, FL USA
来源
2022 IEEE INTERNATIONAL CONFERENCE ON DATA MINING (ICDM) | 2022年
基金
中国国家自然科学基金;
关键词
Graph Generation; Generative Diffusion Process; Graph Neural Network;
D O I
10.1109/ICDM54844.2022.00030
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Graph generative models have broad applications in biology, chemistry and social science. However, modelling and understanding the generative process of graphs is challenging due to the discrete and high-dimensional nature of graphs, as well as permutation invariance to node orderings in underlying graph distributions. Current leading autoregressive models fail to capture the permutation invariance nature of graphs for the reliance on generation ordering and have high time complexity. Here, we propose a continuous-time generative diffusion process for permutation invariant graph generation to mitigate these issues. Specifically, we first construct a forward diffusion process defined by a stochastic differential equation (SDE), which smoothly converts graphs within the complex distribution to random graphs that follow a known edge probability. Solving the corresponding reverse-time SDE, graphs can be generated from newly sampled random graphs. To facilitate the reverse-time SDE, we newly design a position-enhanced graph score network, capturing the evolving structure and position information from perturbed graphs for permutation equivariant score estimation. Under the evaluation of comprehensive metrics, our proposed generative diffusion process achieves competitive performance in graph distribution learning. Experimental results also show that GraphGDP can generate high-quality graphs in only 24 function evaluations, much faster than previous autoregressive models.
引用
收藏
页码:201 / 210
页数:10
相关论文
共 59 条
[41]   Joint Optimization of Target Assignment and Resource Allocation for Multi-Target Tracking in Phased Array Radar Network [J].
Shi, Chenguang ;
Ding, Lintao ;
Qiu, Wei ;
Wang, Fei ;
Zhou, Jianjiang .
2020 IEEE RADAR CONFERENCE (RADARCONF20), 2020,
[42]   GraphVAE: Towards Generation of Small Graphs Using Variational Autoencoders [J].
Simonovsky, Martin ;
Komodakis, Nikos .
ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING - ICANN 2018, PT I, 2018, 11139 :412-422
[43]  
Sohl-Dickstein J, 2015, PR MACH LEARN RES, V37, P2256
[44]  
Song J., 2021, ICLR
[45]  
Song Y., 2019, ADV NEUR IN, P1
[46]  
Song Y., 2021, INT C LEARN REPR
[47]  
Srinivasan B, 2020, Arxiv, DOI arXiv:1910.00452
[48]  
Thompson R., 2022, ICLR
[49]  
Uria B, 2014, PR MACH LEARN RES, V32
[50]  
Vaswani A, 2017, ADV NEUR IN, V30