We propose PiNet, a generalised differentiable attention-based pooling mechanism for utilising graph convolution operations for graph level classification. We demonstrate high sample efficiency and superior performance over other graph neural networks in distinguishing isomorphic graph classes, as well as competitive results with state of the art methods on standard chemo-informatics datasets.
@article{arxiv.2008.04575,
title = {PiNet: Attention Pooling for Graph Classification},
author = {Peter Meltzer and Marcelo Daniel Gutierrez Mallea and Peter J. Bentley},
journal= {arXiv preprint arXiv:2008.04575},
year = {2020}
}