Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Efficient and Scalable Graph Generation through Iterative Local Expansion

About

In the realm of generative models for graphs, extensive research has been conducted. However, most existing methods struggle with large graphs due to the complexity of representing the entire joint distribution across all node pairs and capturing both global and local graph structures simultaneously. To overcome these issues, we introduce a method that generates a graph by progressively expanding a single node to a target graph. In each step, nodes and edges are added in a localized manner through denoising diffusion, building first the global structure, and then refining the local details. The local generation avoids modeling the entire joint distribution over all node pairs, achieving substantial computational savings with subquadratic runtime relative to node count while maintaining high expressivity through multiscale generation. Our experiments show that our model achieves state-of-the-art performance on well-established benchmark datasets while successfully scaling to graphs with at least 5000 nodes. Our method is also the first to successfully extrapolate to graphs outside of the training distribution, showcasing a much better generalization capability over existing methods.

Andreas Bergmeister, Karolis Martinkus, Nathana\"el Perraudin, Roger Wattenhofer• 2023

Related benchmarks

TaskDatasetResultRank
Graph generationSPECTRE SBM Graphs (test)
Degree Metric0.0119
9
Graph generationSPECTRE Planar Graphs (test)
VUN (%)9.50e+3
8
Graph generationExpanded Planar Graphs |V| = 64, N_train = 8192 (test)
VUN89.94
5
Graph generationExpanded Planar (test)
Uniqueness100
5
Graph generationExpanded SBM (test)
Uniqueness100
5
Graph generationExpanded Lobster (test)
Uniqueness99.61
5
Graph generationExpanded Lobster Graphs N_train = 8192 (test)
VUN63.96
5
Graph generationExpanded SBM Graphs N_train = 8192 (test)
VUN3.52
5
Graph generationProtein Graphs 100 <= |V| <= 500 Ntrain = 587
Degree Fidelity0.0033
5
Graph generationTree
A.Ratio2.1
3
Showing 10 of 11 rows

Other info

Follow for update