Graph Convolutional Network

Author: Qi Huang, Minjie Wang, Yu Gai, Quan Gan, Zheng Zhang

This is a gentle introduction of using DGL to implement Graph Convolutional Networks (Kipf & Welling et al., Semi-Supervised Classification with Graph Convolutional Networks). We explain what is under the hood of the GraphConv module. The reader is expected to learn how to define a new GNN layer using DGL’s message passing APIs.

We build upon the earlier tutorial on DGLGraph and demonstrate how DGL combines graph with deep neural network and learn structural representations.

Model Overview

GCN from the perspective of message passing

We describe a layer of graph convolutional neural network from a message passing perspective; the math can be found here. It boils down to the following step, for each node \(u\):

1) Aggregate neighbors’ representations \(h_{v}\) to produce an intermediate representation \(\hat{h}_u\). 2) Transform the aggregated representation \(\hat{h}_{u}\) with a linear projection followed by a non-linearity: \(h_{u} = f(W_{u} \hat{h}_u)\).

We will implement step 1 with DGL message passing, and step 2 by PyTorch nn.Module.

GCN implementation with DGL

We first define the message and reduce function as usual. Since the aggregation on a node \(u\) only involves summing over the neighbors’ representations \(h_v\), we can simply use builtin functions:

import dgl
import dgl.function as fn
import torch as th
import torch.nn as nn
import torch.nn.functional as F
from dgl import DGLGraph

gcn_msg = fn.copy_src(src='h', out='m')
gcn_reduce = fn.sum(msg='m', out='h')

We then proceed to define the GCNLayer module. A GCNLayer essentially performs message passing on all the nodes then applies a fully-connected layer.

class GCNLayer(nn.Module):
    def __init__(self, in_feats, out_feats):
        super(GCNLayer, self).__init__()
        self.linear = nn.Linear(in_feats, out_feats)

    def forward(self, g, feature):
        # Creating a local scope so that all the stored ndata and edata
        # (such as the `'h'` ndata below) are automatically popped out
        # when the scope exits.
        with g.local_scope():
            g.ndata['h'] = feature
            g.update_all(gcn_msg, gcn_reduce)
            h = g.ndata['h']
            return self.linear(h)

The forward function is essentially the same as any other commonly seen NNs model in PyTorch. We can initialize GCN like any nn.Module. For example, let’s define a simple neural network consisting of two GCN layers. Suppose we are training the classifier for the cora dataset (the input feature size is 1433 and the number of classes is 7). The last GCN layer computes node embeddings, so the last layer in general does not apply activation.

class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.layer1 = GCNLayer(1433, 16)
        self.layer2 = GCNLayer(16, 7)

    def forward(self, g, features):
        x = F.relu(self.layer1(g, features))
        x = self.layer2(g, x)
        return x
net = Net()
print(net)

Out:

Net(
  (layer1): GCNLayer(
    (linear): Linear(in_features=1433, out_features=16, bias=True)
  )
  (layer2): GCNLayer(
    (linear): Linear(in_features=16, out_features=7, bias=True)
  )
)

We load the cora dataset using DGL’s built-in data module.

from dgl.data import citation_graph as citegrh
import networkx as nx
def load_cora_data():
    data = citegrh.load_cora()
    features = th.FloatTensor(data.features)
    labels = th.LongTensor(data.labels)
    train_mask = th.BoolTensor(data.train_mask)
    test_mask = th.BoolTensor(data.test_mask)
    g = DGLGraph(data.graph)
    return g, features, labels, train_mask, test_mask

When a model is trained, we can use the following method to evaluate the performance of the model on the test dataset:

def evaluate(model, g, features, labels, mask):
    model.eval()
    with th.no_grad():
        logits = model(g, features)
        logits = logits[mask]
        labels = labels[mask]
        _, indices = th.max(logits, dim=1)
        correct = th.sum(indices == labels)
        return correct.item() * 1.0 / len(labels)

We then train the network as follows:

import time
import numpy as np
g, features, labels, train_mask, test_mask = load_cora_data()
optimizer = th.optim.Adam(net.parameters(), lr=1e-2)
dur = []
for epoch in range(50):
    if epoch >=3:
        t0 = time.time()

    net.train()
    logits = net(g, features)
    logp = F.log_softmax(logits, 1)
    loss = F.nll_loss(logp[train_mask], labels[train_mask])

    optimizer.zero_grad()
    loss.backward()
    optimizer.step()

    if epoch >=3:
        dur.append(time.time() - t0)

    acc = evaluate(net, g, features, labels, test_mask)
    print("Epoch {:05d} | Loss {:.4f} | Test Acc {:.4f} | Time(s) {:.4f}".format(
            epoch, loss.item(), acc, np.mean(dur)))

Out:

/home/ubuntu/.pyenv/versions/miniconda3-latest/lib/python3.7/site-packages/numpy/core/fromnumeric.py:3257: RuntimeWarning: Mean of empty slice.
  out=out, **kwargs)
/home/ubuntu/.pyenv/versions/miniconda3-latest/lib/python3.7/site-packages/numpy/core/_methods.py:161: RuntimeWarning: invalid value encountered in double_scalars
  ret = ret.dtype.type(ret / rcount)
Epoch 00000 | Loss 1.9584 | Test Acc 0.2100 | Time(s) nan
Epoch 00001 | Loss 1.7631 | Test Acc 0.3460 | Time(s) nan
Epoch 00002 | Loss 1.5782 | Test Acc 0.3750 | Time(s) nan
Epoch 00003 | Loss 1.4405 | Test Acc 0.4420 | Time(s) 0.0742
Epoch 00004 | Loss 1.3171 | Test Acc 0.4480 | Time(s) 0.0740
Epoch 00005 | Loss 1.2189 | Test Acc 0.4720 | Time(s) 0.0740
Epoch 00006 | Loss 1.1374 | Test Acc 0.5230 | Time(s) 0.0738
Epoch 00007 | Loss 1.0684 | Test Acc 0.5890 | Time(s) 0.0737
Epoch 00008 | Loss 1.0067 | Test Acc 0.6350 | Time(s) 0.0739
Epoch 00009 | Loss 0.9503 | Test Acc 0.6680 | Time(s) 0.0740
Epoch 00010 | Loss 0.8978 | Test Acc 0.6840 | Time(s) 0.0740
Epoch 00011 | Loss 0.8476 | Test Acc 0.6950 | Time(s) 0.0740
Epoch 00012 | Loss 0.7988 | Test Acc 0.6970 | Time(s) 0.0740
Epoch 00013 | Loss 0.7513 | Test Acc 0.7090 | Time(s) 0.0740
Epoch 00014 | Loss 0.7056 | Test Acc 0.7180 | Time(s) 0.0740
Epoch 00015 | Loss 0.6626 | Test Acc 0.7230 | Time(s) 0.0742
Epoch 00016 | Loss 0.6224 | Test Acc 0.7330 | Time(s) 0.0744
Epoch 00017 | Loss 0.5854 | Test Acc 0.7420 | Time(s) 0.0743
Epoch 00018 | Loss 0.5518 | Test Acc 0.7500 | Time(s) 0.0743
Epoch 00019 | Loss 0.5213 | Test Acc 0.7610 | Time(s) 0.0743
Epoch 00020 | Loss 0.4932 | Test Acc 0.7650 | Time(s) 0.0743
Epoch 00021 | Loss 0.4670 | Test Acc 0.7600 | Time(s) 0.0742
Epoch 00022 | Loss 0.4424 | Test Acc 0.7630 | Time(s) 0.0742
Epoch 00023 | Loss 0.4194 | Test Acc 0.7710 | Time(s) 0.0742
Epoch 00024 | Loss 0.3982 | Test Acc 0.7740 | Time(s) 0.0742
Epoch 00025 | Loss 0.3785 | Test Acc 0.7750 | Time(s) 0.0741
Epoch 00026 | Loss 0.3599 | Test Acc 0.7790 | Time(s) 0.0741
Epoch 00027 | Loss 0.3422 | Test Acc 0.7800 | Time(s) 0.0741
Epoch 00028 | Loss 0.3255 | Test Acc 0.7870 | Time(s) 0.0741
Epoch 00029 | Loss 0.3101 | Test Acc 0.7880 | Time(s) 0.0742
Epoch 00030 | Loss 0.2957 | Test Acc 0.7870 | Time(s) 0.0742
Epoch 00031 | Loss 0.2821 | Test Acc 0.7890 | Time(s) 0.0742
Epoch 00032 | Loss 0.2691 | Test Acc 0.7880 | Time(s) 0.0742
Epoch 00033 | Loss 0.2566 | Test Acc 0.7900 | Time(s) 0.0742
Epoch 00034 | Loss 0.2448 | Test Acc 0.7930 | Time(s) 0.0742
Epoch 00035 | Loss 0.2338 | Test Acc 0.7950 | Time(s) 0.0741
Epoch 00036 | Loss 0.2234 | Test Acc 0.7960 | Time(s) 0.0741
Epoch 00037 | Loss 0.2134 | Test Acc 0.7960 | Time(s) 0.0741
Epoch 00038 | Loss 0.2041 | Test Acc 0.7960 | Time(s) 0.0741
Epoch 00039 | Loss 0.1954 | Test Acc 0.7960 | Time(s) 0.0741
Epoch 00040 | Loss 0.1871 | Test Acc 0.7960 | Time(s) 0.0741
Epoch 00041 | Loss 0.1791 | Test Acc 0.7970 | Time(s) 0.0741
Epoch 00042 | Loss 0.1716 | Test Acc 0.7970 | Time(s) 0.0741
Epoch 00043 | Loss 0.1646 | Test Acc 0.7980 | Time(s) 0.0741
Epoch 00044 | Loss 0.1578 | Test Acc 0.7960 | Time(s) 0.0740
Epoch 00045 | Loss 0.1513 | Test Acc 0.7950 | Time(s) 0.0740
Epoch 00046 | Loss 0.1453 | Test Acc 0.7960 | Time(s) 0.0740
Epoch 00047 | Loss 0.1395 | Test Acc 0.7960 | Time(s) 0.0741
Epoch 00048 | Loss 0.1340 | Test Acc 0.7980 | Time(s) 0.0741
Epoch 00049 | Loss 0.1289 | Test Acc 0.7970 | Time(s) 0.0741

GCN in one formula

Mathematically, the GCN model follows this formula:

\(H^{(l+1)} = \sigma(\tilde{D}^{-\frac{1}{2}}\tilde{A}\tilde{D}^{-\frac{1}{2}}H^{(l)}W^{(l)})\)

Here, \(H^{(l)}\) denotes the \(l^{th}\) layer in the network, \(\sigma\) is the non-linearity, and \(W\) is the weight matrix for this layer. \(D\) and \(A\), as commonly seen, represent degree matrix and adjacency matrix, respectively. The ~ is a renormalization trick in which we add a self-connection to each node of the graph, and build the corresponding degree and adjacency matrix. The shape of the input \(H^{(0)}\) is \(N \times D\), where \(N\) is the number of nodes and \(D\) is the number of input features. We can chain up multiple layers as such to produce a node-level representation output with shape :math`N times F`, where \(F\) is the dimension of the output node feature vector.

The equation can be efficiently implemented using sparse matrix multiplication kernels (such as Kipf’s pygcn code). The above DGL implementation in fact has already used this trick due to the use of builtin functions. To understand what is under the hood, please read our tutorial on PageRank.

Total running time of the script: ( 0 minutes 13.375 seconds)

Gallery generated by Sphinx-Gallery