PiNet: Attention Pooling for Graph Classification

Peter Meltzer, Marcelo Daniel Gutierrez Mallea, Peter J. Bentley

We propose PiNet, a generalised differentiable attention-based pooling mechanism for utilising graph convolution operations for graph level classification. We demonstrate high samp…