arrow
Return

Multiresolution equivariant graph variational autoencoder

delete2023-03-21
delete5
delete
OA
AI
T
Truong Son Hy *
R
Risi Kondor
DOI:10.1088/2632-2153/acc0d8delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
In this paper, we propose Multiresolution Equivariant Graph Variational Autoencoders (MGVAE), the first hierarchical generative model to learn and generate graphs in a multiresolution and equivariant manner. At each resolution level, MGVAE employs higher order message passing to encode the graph while learning to partition it into mutually exclusive clusters and coarsening into a lower resolution that eventually creates a hierarchy of latent distributions. MGVAE then constructs a hierarchical generative model to variationally decode into a hierarchy of coarsened graphs. Importantly, our proposed framework is end-to-end permutation equivariant with respect to node ordering. MGVAE achieves competitive results with several generative tasks including general graph generation, molecular generation, unsupervised molecular representation learning to predict molecular properties, link prediction on citation graphs, and graph-based image generation.
Keywords:
graph neural networks
graph variational autoencoders
hierarchical generative models
molecule generation
supervised and unsupervised molecular representation learning

Journal

M
Machine Learning-Science and Technology
IF:
4.6
Papers:
1.1K
Citations:
3.4K

Organization

U
university of chicago
Scholars:
4.4W
Papers: 3.7W
Citations: 80