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.

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

References in corpus (19)

• Inductive Representation Learning on Large Graphs
• Neural Message Passing for Quantum Chemistry
• Geometric deep learning: going beyond Euclidean data
• Deep Convolutional Networks on Graph-Structured Data
• Learning Convolutional Neural Networks for Graphs
• Pitfalls of Graph Neural Network Evaluation
• A Gentle Introduction to Deep Learning for Graphs
• Convolutional Neural Network Architectures for Signals Supported on Graphs
• Laplacian Support Vector Machines Trained in the Primal
• Residual Gated Graph ConvNets
• Benchmarking Graph Neural Networks
• Training Neural Networks Without Gradients: A Scalable ADMM Approach
• Anonymous Walk Embeddings
• Invariant and Equivariant Graph Networks
• Machine Learning on Graphs: A Model and Comprehensive Taxonomy
• Invariance-Preserving Localized Activation Functions for Graph Neural Networks
• Distributed optimization of deeply nested systems
• Relational Pooling for Graph Representations
• A Lagrangian Approach to Information Propagation in Graph Neural Networks