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

Micro and Macro Level Graph Modeling for Graph Variational Auto-Encoders

About

Generative models for graph data are an important research topic in machine learning. Graph data comprise two levels that are typically analyzed separately: node-level properties such as the existence of a link between a pair of nodes, and global aggregate graph-level statistics, such as motif counts. This paper proposes a new multi-level framework that jointly models node-level properties and graph-level statistics, as mutually reinforcing sources of information. We introduce a new micro-macro training objective for graph generation that combines node-level and graph-level losses. We utilize the micro-macro objective to improve graph generation with a GraphVAE, a well-established model based on graph-level latent variables, that provides fast training and generation time for medium-sized graphs. Our experiments show that adding micro-macro modeling to the GraphVAE model improves graph quality scores up to 2 orders of magnitude on five benchmark datasets, while maintaining the GraphVAE generation speed advantage.

Kiarash Zahirnia, Oliver Schulte, Parmis Naddaf, Ke Li• 2022

Related benchmarks

TaskDatasetResultRank
Graph generationTriangle Grid
MMD RBF0.13
12
Graph Generative ModelingMutag (test)
Degree Distribution0.001
12
Graph Generative ModelingPTC (test)
Degree Distribution Error0.02
12
Graph generationMUTAG
Training Time (s)0.15
10
Graph generationPTC
Train Time (s)0.32
10
Graph generationPROTEIN
MMD RBF0.03
6
Graph generationLobster
MMD RBF0.1
6
Graph generationogbg-molbbbp
MMD RBF0.02
6
Graph generationPTC
MMD RBF0.04
6
Graph generationGRID (test)
Train Time (s)0.49
6
Showing 10 of 14 rows

Other info

Code

Follow for update