arXiv · 2108.12982
Adversarial Stein Training for Graph Energy Models
Abstract
Learning distributions over graph-structured data is a challenging task with many applications in biology and chemistry. In this work we use an energy-based model (EBM) based on multi-channel graph neural networks (GNN) to learn permutation invariant unnormalized density functions on graphs. Unlike standard EBM training methods our approach is to learn the model via minimizing adversarial stein discrepancy. Samples from the model can be obtained via Langevin dynamics based MCMC. We find that this approach achieves competitive results on graph generation compared to benchmark models.
Explore related subjects
Keep this discovery
Shiv Shankar. 2021-08-30. Adversarial Stein Training for Graph Energy Models. https://arxiv.org/abs/2108.12982
Cite the original work for its findings. Save a collection to share your selection of sources.