Edge-Featured Graph Attention Network

Abstract

Lots of neural network architectures have been proposed to deal with learning tasks on graph-structured data. However, most of these models concentrate on only node features during the learning process. The edge features, which usually play a similarly important role as the nodes, are often ignored or simplified by these models. In this paper, we present edge-featured graph attention networks, namely EGATs, to extend the use of graph neural networks to those tasks learning on graphs with both node and edge features. These models can be regarded as extensions of graph attention networks (GATs). By reforming the model structure and the learning process, the new models can accept node and edge features as inputs, incorporate the edge information into feature representations, and iterate both node and edge features in a parallel but mutual way. The results demonstrate that our work is highly competitive against other node classification approaches, and can be well applied in edge-featured graph learning tasks.

References in corpus (10)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Semi-Supervised Classification with Graph Convolutional Networks
• Inductive Representation Learning on Large Graphs
• Spectral Networks and Locally Connected Networks on Graphs
• Graph Neural Networks: A Review of Methods and Applications
• Deep Convolutional Networks on Graph-Structured Data
• Representation Learning on Graphs with Jumping Knowledge Networks
• Gated Graph Sequence Neural Networks
• Revisiting Semi-Supervised Learning with Graph Embeddings
• Deep Learning on Graphs: A Survey

Cited by in corpus (5)

• Wind Park Power Prediction: Attention-Based Graph Networks and Deep Learning to Capture Wake Losses
• A Network Science perspective of Graph Convolutional Networks: A survey
• Heterogeneous Edge-Enhanced Graph Attention Network For Multi-Agent Trajectory Prediction
• Spatial Graph Attention and Curiosity-driven Policy for Antiviral Drug Discovery
• Edge-featured Graph Neural Architecture Search