基于tensorflow2.0实现图卷积神经网络GCN

爱被打了一巴掌 2024-03-24 19:40 38阅读 0赞

[资源下载地址][Link 1]：https://download.csdn.net/download/sheziqiong/88240220  
[资源下载地址][Link 1]：https://download.csdn.net/download/sheziqiong/88240220

## 图卷积神经网络(GCN)理解与 tensorflow2.0 代码实现 ##

图(Graph)，一般用  G = ( V , E ) G=(V,E) G=(V,E) 表示，这里的 V V V是图中节点的集合， E E E 为边的集合，节点的个数用 N N N表示。在一个图中，有三个比较重要的矩阵：

1.  特征矩阵 X X X：维度为  N × D N\\times D N×D ，表示图中有 N 个节点，每个节点的特征个数是 D。
2.  邻居矩阵 A A A：维度为  N × N N\\times N N×N ，表示图中 N 个节点之间的连接关系。
3.  度矩阵  D D D：维度为  N × N N\\times N N×N ，是一个对角矩阵，即只有对角线上不为零，其他位置元素都是 0 ，表示图中 N 个节点与其他节点相连的边的个数。对于无权图而言， D i i = ∑ j A i j D\_\{ii\}=\\sum\_j A\_\{ij\} Dii=∑jAij 。

邻接矩阵与度矩阵例子如下图所示：

![ec039831425be345680fcf893911f958.png][]

对于图像(Image)数据，我们可以用卷积核来提取特征，无论卷积核覆盖在图像的哪个部分，其内部结构都是一样的，这是因为图片结构具有平移不变性，如下图左半部分所示：

![296359ff58dcce207b374ac5be4ebb13.png][]

但是对于图(Graph)数据而言，其形状是不规则的，不具有平移不变性。于是 GCN，也就是图卷积神经网络，其目标就是设计一种特征提取器，进而完成节点分类、变预测等任务，还顺便可以得到每个节点的 embedding 表示。

上面展示了一个简单的 3x3 的卷积核，每次自左向右，自上而下扫描图片(Image)时，都是将 3x3 的像素进行加权求和，即： ∑ i = 1 9 w i x i \\sum\_\{i=1\}^9 w\_i x\_i ∑i=19wixi，然后将求和的结果作为该 3x3 区域的特征。

那么在图(graph)中要怎么提取特征？这里给出两种思路。

### 图卷积 ###

#### 思路一 ####

CNN 加权求和的思想也可以应用到图(Graph)的特征提取上，如下图所示：

![cd580be4445b45b2668d8582c004354b.png][]

对于节点  i i i，我们可以用其邻接节点加权求和的结果来表示当前节点，这个操作我们称为“聚合(aggregate)”：

a g g ( X i ) = ∑ j ∈ n e i g h b o r ( i ) A i j X j agg(X\_i) = \\sum\_\{j \\in neighbor(i)\} A\_\{ij\} X\_j agg(Xi)=j∈neighbor(i)∑AijXj

考虑到与节点  i i i 没有边连接的节点  j j j ，对应的权重  A i j A\_\{ij\} Aij 为 0 ，因此上面的公式又可以改写为：

a g g ( X i ) = ∑ j ∈ N A i j X j agg(X\_i) = \\sum\_\{j \\in N\} A\_\{ij\} X\_j agg(Xi)=j∈N∑AijXj

那么，对于所有的节点而言，其聚合的结果可以用下面的公式表示：

a g g ( X ) = A X agg(X) = AX agg(X)=AX

上面的公式只考虑了邻居加权求和的结果，很多情况下，节点自身的信息是不可忽略的，因此一般情况下会把自身的特征也加回来：

a g g ( X i ) = ∑ j ∈ N A i j X j + X i agg(X\_i) = \\sum\_\{j \\in N\} A\_\{ij\} X\_j + X\_i agg(Xi)=j∈N∑AijXj\+Xi

于是有：

a g g ( X ) = A X + X = ( A + I ) X agg(X) = AX + X = (A+I)X agg(X)=AX\+X=(A\+I)X

其中， I I I 是单位矩阵，令：

A ~ = A + I \\tilde A = A+I A~=A\+I

则有：

a g g ( X ) = A ~ X agg(X) = \\tilde AX agg(X)=A~X

也就是说把单位矩阵  I I I 加到邻接矩阵  A A A 上，即可在聚合操作中加入自身特征了。

现在有个问题，只能用自身节点以及邻居节点加权求和的结果来表示某个节点的特征吗？其实还有另一种思路。

#### 思路二 ####

在某些情况下，我们更关注节点之间的差值，因此可以对差值进行加权求和：

a g g ( X i ) = ∑ j ∈ N A i j ( X i − X j ) = D i i X i − ∑ j ∈ N A i j X j \\begin\{aligned\} agg(X\_i) & = \\sum\_\{j \\in N\} A\_\{ij\} (X\_i - X\_j) \\\\ &= D\_\{ii\}X\_i- \\sum\_\{j \\in N\} A\_\{ij\}X\_j \\\\ \\end\{aligned\} agg(Xi)=j∈N∑Aij(Xi−Xj)=DiiXi−j∈N∑AijXj

其中，D 表示度矩阵，表示节点与其他节点相连的边的个数，对于无权图而言， D i i = ∑ j A i j D\_\{ii\}=\\sum\_j A\_\{ij\} Dii=∑jAij 。

对于整个图的节点而言，上面的公式可以转换为矩阵化的表示：

a g g ( X ) = D X − A X = ( D − A ) X \\begin\{aligned\} agg(X) &= DX - AX \\\\ &= (D-A)X \\end\{aligned\} agg(X)=DX−AX=(D−A)X

> 实际上，上面公式中的  D − A D-A D−A 是拉普拉斯矩阵（用  L L L 表示）：
> 
>  L = D − A L = D - A L=D−A
> 
> 拉普拉斯矩阵如下图所示：
> 
> ![9256f6554033bb04b17f318e1645b1f0.png][]
> 
> 如果想更多地了解拉普拉斯矩阵在 GCN 中的作用，可以参考：[如何理解 Graph Convolutional Network（GCN）？][Graph Convolutional Network_GCN]

#### 归一化 ####

无论是思路一的  A ~ \\tilde A A~ 还是思路二的  L L L，与 CNN 的卷积相似之处都是局部数据的聚合操作，只不过 CNN 中卷积的局部连接数是固定的。但是在 Graph 中每个节点的邻居个数都可能不同，进行聚合操作后，对于度较大的节点，得到的特征比较大，度较少的节点得到的特征就比较小，因此还需要进行归一化的处理。

归一化的思路有两种：

（1）算数平均

L r w = D − 1 L L^\{rw\}=D^\{-1\}L Lrw=D−1L

（2）几何平均

L s y m = D − 0.5 L D − 0.5 L^\{sym\}=D^\{-0.5\}LD^\{-0.5\} Lsym=D−0.5LD−0.5

几何平均受极端值影响较小，因此是 GCN 中比较常用的归一化方法，于是有：

a g g ( X ) = L s y m X = D − 0.5 L D − 0.5 X = D − 0.5 ( D − A ) D − 0.5 X \\begin\{aligned\} agg(X) &= L^\{sym\} X \\\\ &= D^\{-0.5\}LD^\{-0.5\}X \\\\ &= D^\{-0.5\}(D-A)D^\{-0.5\} X \\end\{aligned\} agg(X)=LsymX=D−0.5LD−0.5X=D−0.5(D−A)D−0.5X

当然也可以是：

a g g ( X ) = D − 0.5 A ~ D − 0.5 X = D − 0.5 ( A + I ) D − 0.5 X \\begin\{aligned\} agg(X) & = D^\{-0.5\}\\tilde A D^\{-0.5\} X\\\\ & = D^\{-0.5\}(A+I)D^\{-0.5\} X \\end\{aligned\} agg(X)=D−0.5A~D−0.5X=D−0.5(A\+I)D−0.5X

在实际的 GCN 代码实现中，会对聚合结果进行一些变换，第  l l l 层到第  l + 1 l+1 l\+1 层的传播方式为：

H ( l + 1 ) = σ ( D ~ − 1 2 A ~ D ~ − 1 2 H ( l ) W ( l ) ) H^\{(l+1)\}=\\sigma\\left(\\tilde\{D\}^\{-\\frac\{1\}\{2\}\} \\tilde\{A\} \\tilde\{D\}^\{-\\frac\{1\}\{2\}\} H^\{(l)\} W^\{(l)\}\\right) H(l\+1)=σ(D~−21A~D~−21H(l)W(l))

其中：

*   A ~ = A + I \\tilde A=A+I A~=A\+I ，也可以是  A ~ = D − A \\tilde A = D - A A~=D−A
 *   D ~ \\tilde D D~ 是  A ~ \\tilde A A~ 的度矩阵，每个元素为： D ~ i i = ∑ j A ~ i j \\tilde D\_\{ii\}=\\sum\_j \\tilde A\_\{ij\} D~ii=∑jA~ij
 *   H H H 是每一层的特征，对于输入层而言， H H H 就是  X X X
 *  σ 是 sigmoid 函数

由于 D 是在矩阵 A 的基础上得到的，因此在给定矩阵 A 之后， D ~ − 1 2 A ~ D ~ − 1 2 \\tilde\{D\}^\{-\\frac\{1\}\{2\}\} \\tilde\{A\} \\tilde\{D\}^\{-\\frac\{1\}\{2\}\} D~−21A~D~−21 就可以事先计算好。

### 代码实现 ###

#### Cora 数据集介绍 ####

Cora 数据集由机器学习论文组成，是近年来图深度学习很喜欢使用的数据集。整个数据集有 2708 篇论文，所有样本点被分为 8 个类别，类别分别是 1）基于案例；2）遗传算法；3）神经网络；4）概率方法；5）强化学习；6）规则学习；7）理论。每篇论文都由一个 1433 维的词向量表示，所以，每个样本点具有 1433 个特征。词向量的每个元素都对应一个词，且该元素只有 0 或 1 两个取值。取 0 表示该元素对应的词不在论文中，取 1 表示在论文中。

#### 训练模型 ####

# 计算标准化的邻接矩阵：根号D * A * 根号D
    def preprocess_graph(adj):
        # _A = A + I
        _adj = adj + sp.eye(adj.shape[0])
        # _dseq：各个节点的度构成的列表
        _dseq = _adj.sum(1).A1
        # 构造开根号的度矩阵
        _D_half = sp.diags(np.power(_dseq, -0.5))
        # 计算标准化的邻接矩阵, @ 表示矩阵乘法
        adj_normalized = _D_half @ _adj @ _D_half
        return adj_normalized.tocsr()
    
    if __name__ == "__main__":
        # 读取数据
        # A_mat：邻接矩阵，以scipy的csr形式存储
        # X_mat：特征矩阵，以scipy的csr形式存储
        # z_vec：label
        # train_idx,val_idx,test_idx: 要使用的节点序号
        A_mat, X_mat, z_vec, train_idx, val_idx, test_idx = load_data_planetoid(FLAGS.dataset)
        # 邻居矩阵标准化
        An_mat = preprocess_graph(A_mat)
    
        # 节点的类别个数
        K = z_vec.max() + 1
    
        # 构造GCN模型
        gcn = GCN(An_mat, X_mat, [FLAGS.hidden1, K])
        # 训练
        gcn.train(train_idx, z_vec[train_idx], val_idx, z_vec[val_idx])
        # 测试
        test_res = gcn.evaluate(test_idx, z_vec[test_idx], training=False)
        print("Dataset {}".format(FLAGS.dataset),
              "Test loss {:.4f}".format(test_res[0]),
              "test acc {:.4f}".format(test_res[1]))

### GCN 小结 ###

GCN 的优点： 可以捕捉 graph 的全局信息，从而很好地表示 node 的特征。

GCN 的缺点：属于直推式(transductive)的学习方式，需要把所有节点都参与训练才能得到 node embedding，无法快速得到新 node 的 embedding。如果要得到新节点的表示，很可能就需要重新训练所有节点的表示。

参考文章：

[如何理解 Graph Convolutional Network（GCN）？][Graph Convolutional Network_GCN 1]

[2020 年，我终于决定入门 GCN][2020 _ GCN]

[GCN(Graph Convolutional Network)的理解][GCN_Graph Convolutional Network]

[资源下载地址][Link 1]：https://download.csdn.net/download/sheziqiong/88240220  
[资源下载地址][Link 1]：https://download.csdn.net/download/sheziqiong/88240220

[Link 1]: https://download.csdn.net/download/sheziqiong/88240220
[ec039831425be345680fcf893911f958.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/13/fa93ec61752940beb6ae7d7c621044e5.png
[296359ff58dcce207b374ac5be4ebb13.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/13/96feddf05d354da186625cc7a1a430a4.png
[cd580be4445b45b2668d8582c004354b.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/13/3ca47909a8ab401d9a44c8e683367d0f.png
[9256f6554033bb04b17f318e1645b1f0.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/13/077a85d0988b43828692e48f08f6cbc6.png
[Graph Convolutional Network_GCN]: https://www.zhihu.com/question/54504471/answer/630639025
[Graph Convolutional Network_GCN 1]: https://www.zhihu.com/question/54504471/answer/611222866
[2020 _ GCN]: https://zhuanlan.zhihu.com/p/112277874
[GCN_Graph Convolutional Network]: https://purelyvivid.github.io/2019/07/07/GCN_1/