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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Graph generation | Triangle Grid | MMD RBF0.13 | 12 | |
| Graph Generative Modeling | Mutag (test) | Degree Distribution0.001 | 12 | |
| Graph Generative Modeling | PTC (test) | Degree Distribution Error0.02 | 12 | |
| Graph generation | MUTAG | Training Time (s)0.15 | 10 | |
| Graph generation | PTC | Train Time (s)0.32 | 10 | |
| Graph generation | PROTEIN | MMD RBF0.03 | 6 | |
| Graph generation | Lobster | MMD RBF0.1 | 6 | |
| Graph generation | ogbg-molbbbp | MMD RBF0.02 | 6 | |
| Graph generation | PTC | MMD RBF0.04 | 6 | |
| Graph generation | GRID (test) | Train Time (s)0.49 | 6 |