homegraphsgat

View code on Github

#

Graph Attention Networks (GAT)

This is a PyTorch implementation of the paper Graph Attention Networks.

GATs work on graph data. A graph consists of nodes and edges connecting nodes. For example, in Cora dataset the nodes are research papers and the edges are citations that connect the papers.

GAT uses masked self-attention, kind of similar to transformers. GAT consists of graph attention layers stacked on top of each other. Each graph attention layer gets node embeddings as inputs and outputs transformed embeddings. The node embeddings pay attention to the embeddings of other nodes it's connected to. The details of graph attention layers are included alongside the implementation.

Here is the training code for training a two-layer GAT on Cora dataset.

labml.ai