Neural Message Passing on High Order Paths

Daniel Flam-Shepherd,Tony Wu,Pascal Friederich,Alan Aspuru-Guzik
DOI: https://doi.org/10.48550/arXiv.2002.10413
2020-02-25
Abstract:Graph neural network have achieved impressive results in predicting molecular properties, but they do not directly account for local and hidden structures in the graph such as functional groups and molecular geometry. At each propagation step, GNNs aggregate only over first order neighbours, ignoring important information contained in subsequent neighbours as well as the relationships between those higher order connections. In this work, we generalize graph neural nets to pass messages and aggregate across higher order paths. This allows for information to propagate over various levels and substructures of the graph. We demonstrate our model on a few tasks in molecular property prediction.
Machine Learning
What problem does this paper attempt to address?