Deep Constraint-based Propagation in Graph Neural Networks

Abstract

The popularity of deep learning techniques renewed the interest in neural architectures able to process complex structures that can be represented using graphs, inspired by Graph Neural Networks (GNNs). We focus our attention on the originally proposed GNN model of Scarselli et al. 2009, which encodes the state of the nodes of the graph by means of an iterative diffusion procedure that, during the learning stage, must be computed at every epoch, until the fixed point of a learnable state transition function is reached, propagating the information among the neighbouring nodes. We propose a novel approach to learning in GNNs, based on constrained optimization in the Lagrangian framework. Learning both the transition function and the node states is the outcome of a joint process, in which the state convergence procedure is implicitly expressed by a constraint satisfaction mechanism, avoiding iterative epoch-wise procedures and the network unfolding. Our computational structure searches for saddle points of the Lagrangian in the adjoint space composed of weights, nodes state variables and Lagrange multipliers. This process is further enhanced by multiple layers of constraints that accelerate the diffusion process. An experimental analysis shows that the proposed approach compares favourably with popular models on several benchmarks.

Published in: IEEE Transactions on Pattern Analysis and Machine Intelligence. arXiv admin note: text overlap with arXiv:2002.07684

References in corpus (8)

• Deep Convolutional Networks on Graph-Structured Data
• A Gentle Introduction to Deep Learning for Graphs
• Laplacian Support Vector Machines Trained in the Primal
• Benchmarking Graph Neural Networks
• Machine Learning on Graphs: A Model and Comprehensive Taxonomy
• Distributed optimization of deeply nested systems
• Relational Pooling for Graph Representations
• A Lagrangian Approach to Information Propagation in Graph Neural Networks