Gated Graph Sequence Neural Networks
Yujia LiDaniel TarlowMarc BrockschmidtRichard Zemel
Introduces Gated Graph Sequence Neural Networks, modernizing graph neural architectures with gated recurrent units to learn representations and generate sequential outputs across graph-structured tasks such as program verification.
The paper introduces Gated Graph Sequence Neural Networks, a neural network architecture designed to learn features from graph-structured inputs and produce sequential outputs. Graph data arises in many domains, and tasks such as reasoning over relations or describing program memory states require models that can both process graphs and emit ordered sequences rather than single classifications. Prior Graph Neural Networks handled only fixed outputs and imposed convergence constraints that limited their practicality.
The work adapts those networks by replacing the propagation rule with Gated Recurrent Units, removing the contraction-map requirement, incorporating node annotations, and extending the model to a sequence of gated steps that can update both outputs and internal node states. Experiments evaluate the resulting models on transformed bAbI reasoning tasks, two new graph-algorithm problems, and a program-verification benchmark that maps heap snapshots to separation-logic formulas describing data structures.
On four single-step bAbI tasks the gated models reach 100 percent accuracy with only 50 training examples, while recurrent and LSTM baselines either require several times more data or fail to reach the same threshold. On path-finding and two sequence-output algorithm tasks the gated sequence networks attain 99–100 percent accuracy with 50–250 examples; the same baselines remain near chance. In the verification setting the model predicts correct logical invariants at 89.96 percent accuracy across held-out formulas, matching or slightly exceeding a heavily engineered feature-based system while using no manual feature design. The predicted invariants suffice to verify correctness of several list-manipulating programs inside an existing theorem prover.
These results indicate that the architecture supplies useful inductive bias for graph problems that involve ordering or enumeration, reduces reliance on domain-specific engineering, and can be trained end-to-end from modest amounts of synthetic data. The approach therefore offers a practical route to automated invariant inference and to other sequence-generation tasks on graphs.
Further work is needed to incorporate temporal order, handle higher-arity relations, accept less-structured inputs, and condition generation dynamically on a query. Additional data and evaluation on larger programs would strengthen before deployment in production verification tools. The reported experiments rest on synthetic or transformed data whose distribution may differ from real-world instances, so performance on naturally occurring graphs remains to be quantified.
- Paper: The Graph Neural Network Model, Franco Scarselli et al. (2009). Reading the foundational Graph Neural Network paper first is essential because the source paper directly modifies and extends its architecture with modern gated units.
- Paper: Semi-Supervised Classification with Graph Convolutional Networks, Thomas N. Kipf et al. (2017). This paper builds directly on the source's graph neural network foundation to introduce efficient spectral graph convolutions for semi-supervised node classification.
- Paper: How Powerful are Graph Neural Networks?, Keyulu Xu et al. (2019). This study extends the representational analysis established in earlier graph neural networks like the source paper to evaluate and improve their expressive power.
- Paper: Graph Attention Networks, Petar Veličković et al. (2018). Following the source paper, this work introduces attention mechanisms to graph neural networks, replacing fixed message-passing weights with dynamically learned neighborhood importance.
- Paper: Inductive Representation Learning on Large Graphs, William L. Hamilton et al. (2017). This work extends the message-passing paradigm introduced in the source paper to an inductive framework capable of generating embeddings for entirely unseen nodes.
- Paper: Representation Learning on Graphs with Jumping Knowledge Networks, Keyulu Xu et al. (2018). This paper builds on the neighborhood aggregation techniques of the source model by introducing jumping knowledge connections to adaptively leverage multiple depths.
- Paper: GNNExplainer: Generating Explanations for Graph Neural Networks, Rex Ying et al. (2019). This work directly extends the predictive capabilities of graph neural networks developed in the source paper by providing an optimization framework to explain their outputs.
- Paper: Simplifying Graph Convolutional Networks, Felix Wu et al. (2019). This paper takes the graph convolutional concepts initiated in the source literature and strips away their nonlinear layers to demonstrate a simpler, faster linear model.
- Paper: Predict then Propagate: Graph Neural Networks meet Personalized PageRank, Johannes Gasteiger et al. (2019). Following the graph neural network foundations of the source paper, this work integrates personalized PageRank propagation to prevent oversmoothing in deep architectures.
