Supervised Graph Attention Network for Semi-Supervised Node Classification

Dongkwan Kim and Alice Oh. Workshop on Graph Representation Learning at NeurIPS (NeurIPS GRL), 2019

We propose supervised graph attention network (super-GAT), a novel neural network architecture for semi-supervised node classification in graphs. While learning (unsupervised) graph attention from the original graph attention network (GAT), we jointly train the attention values by supervised learning with information whether an edge exists between a pair of nodes. By giving supervision, we can assign not only implicit weights to nodes in the neighborhood but also explicit weights to nodes in any relation. Our model is based on GAT, and it only needs a few additional parameters and computation. We show how super-GAT performs on three transductive benchmark citation datasets: Cora, CiteSeer, and PubMed, and compared to baseline models including GAT, super-GAT achieves higher prediction accuracy for the first two datasets.

Leave a Comment