Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Bilinear Graph Neural Network with Neighbor Interactions
(Submitted on 10 Feb 2020 (v1), last revised 30 May 2020 (this version, v5))
Abstract: Graph Neural Network (GNN) is a powerful model to learn representations and make predictions on graph data. Existing efforts on GNN have largely defined the graph convolution as a weighted sum of the features of the connected nodes to form the representation of the target node. Nevertheless, the operation of weighted sum assumes the neighbor nodes are independent of each other, and ignores the possible interactions between them. When such interactions exist, such as the co-occurrence of two neighbor nodes is a strong signal of the target node's characteristics, existing GNN models may fail to capture the signal. In this work, we argue the importance of modeling the interactions between neighbor nodes in GNN. We propose a new graph convolution operator, which augments the weighted sum with pairwise interactions of the representations of neighbor nodes. We term this framework as Bilinear Graph Neural Network (BGNN), which improves GNN representation ability with bilinear interactions between neighbor nodes. In particular, we specify two BGNN models named BGCN and BGAT, based on the well-known GCN and GAT, respectively. Empirical results on three public benchmarks of semi-supervised node classification verify the effectiveness of BGNN -- BGCN (BGAT) outperforms GCN (GAT) by 1.6% (1.5%) in classification accuracy.Codes are available at: this https URL
Submission history
From: Hongmin Zhu [view email][v1] Mon, 10 Feb 2020 06:43:38 GMT (3942kb,D)
[v2] Thu, 20 Feb 2020 08:20:11 GMT (3942kb,D)
[v3] Wed, 13 May 2020 01:50:32 GMT (1971kb,D)
[v4] Sat, 16 May 2020 03:51:46 GMT (4064kb,D)
[v5] Sat, 30 May 2020 02:41:36 GMT (1409kb,D)
Link back to: arXiv, form interface, contact.