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 条
[1]   Statistical mechanics of complex networks [J].
Albert, R ;
Barabási, AL .
REVIEWS OF MODERN PHYSICS, 2002, 74 (01) :47-97
[2]  
Anderson BD., 1982, STOCH PROC APPL, V12, P313, DOI [DOI 10.1016/0304-4149(82)90051-5, 10.1016/0304-4149(82)90051-5]
[3]  
Bojchevski A, 2018, PR MACH LEARN RES, V80
[4]  
Chen B, 2018, Arxiv, DOI arXiv:1809.00773
[5]  
Chen T., 2018, Advances in Neural Information Processing Systems
[6]  
Chen X., 2021, P MACHINE LEARNING R, V139, P1630
[7]  
CUI H., 2021, arXiv
[8]  
Dai A., 2020, ICML, P2302
[9]  
De Cao N, 2018, Arxiv, DOI [arXiv:1805.11973, DOI 10.4855/ARXIV.1805.11973, 10.48550/arXiv.1805.11973]
[10]  
De Groot S.R., 2013, NONEQUILIBRIUM THERM