Improving Graph Property Prediction with Generalized Readout Functions

09/21/2020
by   Eric Alcaide, et al.
1

Graph property prediction is drawing increasing attention in the recent years due to the fact that graphs are one of the most general data structures since they can contain an arbitrary number of nodes and connections between them, and it is the backbone for many different tasks like classification and regression on such kind of data (networks, molecules, knowledge bases, ...). We introduce a novel generalized global pooling layer to mitigate the information loss that typically occurs at the Readout phase in Message-Passing Neural Networks. This novel layer is parametrized by two values (β and p) which can optionally be learned, and the transformation it performs can revert to several already popular readout functions (mean, max and sum) under certain settings, which can be specified. To showcase the superior expressiveness and performance of this novel technique, we test it in a popular graph property prediction task by taking the current best-performing architecture and using our readout layer as a drop-in replacement and we report new state of the art results. The code to reproduce the experiments can be accessed here: https://github.com/EricAlcaide/generalized-readout-phase

READ FULL TEXT
research
02/01/2020

Molecule Property Prediction and Classification with Graph Hypernetworks

Graph neural networks are currently leading the performance charts in le...
research
02/21/2020

Memory-Based Graph Networks

Graph neural networks (GNNs) are a class of deep models that operate on ...
research
11/24/2020

Lipophilicity Prediction with Multitask Learning and Molecular Substructures Representation

Lipophilicity is one of the factors determining the permeability of the ...
research
05/18/2023

Seq-HGNN: Learning Sequential Node Representation on Heterogeneous Graph

Recent years have witnessed the rapid development of heterogeneous graph...
research
04/17/2019

Neural Message Passing for Multi-Label Classification

Multi-label classification (MLC) is the task of assigning a set of targe...
research
02/08/2023

Geometry-Complete Diffusion for 3D Molecule Generation

Denoising diffusion probabilistic models (DDPMs) have recently taken the...

Please sign up or login with your details

Forgot password? Click here to reset