Variational Recurrent Neural Networks for Graph Classification
arXiv:1902.02721
Abstract
We address the problem of graph classification based only on structural information. Inspired by natural language processing techniques (NLP), our model sequentially embeds information to estimate class membership probabilities. Besides, we experiment with NLP-like variational regularization techniques, making the model predict the next node in the sequence as it reads it. We experimentally show that our model achieves state-of-the-art classification results on several standard molecular datasets. Finally, we perform a qualitative analysis and give some insights on whether the node prediction helps the model better classify graphs.
Representation Learning on Graphs and Manifolds workshop, ICLR 2019
References in corpus (5)
- Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modeling
- Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translation
- graph2vec: Learning Distributed Representations of Graphs
- Feature-Based Classification of Networks
- Dynamics Based Features For Graph Classification