Task03:基于图神经网络的节点表征学习

一上来依旧是放Datawhale提供的课程链接:https://github.com/datawhalechina/team-learning-nlp/blob/master/GNN(打call打call,微信公众号是:Datawhale)

在导论部分我们就了解到,图提供了一种通用的数据表示方法,众多其他类型的数据也可以转换为图的形式。大量现实世界中的问题可以作为图上的一组小的计算任务来解决。推断节点属性、检测异常节点(如垃圾邮件发送者)、识别与疾病相关的基因、向病人推荐药物等,都可以概括为节点分类问题。推荐、药物负作用预测、药物与目标的相互作用识别和知识图谱的完成(knowledge graph completion)等,本质上是边预测问题

在图节点预测或边预测任务中,首先需要生成节点表征(Node Representation)。我们使用图神经网络来生成节点表征,并通过基于监督学习的对图神经网络的训练,使得图神经网络学会产生高质量的节点表征。高质量的节点表征能够用于衡量节点的相似性,同时高质量的节点表征也是准确分类节点的前提。

本文中,我们将学习实现多层图神经网络的方法,并以节点分类任务为例,学习训练图神经网络的一般过程

我们将以Cora数据集为例子进行说明,Cora由2708篇论文及它们之间的引用关系构成的边组成。这些论文根据主题被划分为7类,分别是神经网络、强化学习、规则学习、概率方法、遗传算法、理论研究、案例相关。每篇论文的特征是通过词袋模型得到的,维度为1433,每一维表示一个词,1表示该词在这篇文章中出现过,0表示未出现。

x:节点特征,维度为2708 * 1433

y:节点对应的标签,包括7个类别

adjacency:邻接矩阵,维度为2708 * 2708

train_mask、val_mask、test_mask:与节点数相同的掩码,用于划分训练集、验证集、测试集。

我们的任务是预测各篇论文的类别(共7类)。并将对MLP和GCN、GAT(两个知名度很高的图神经网络)三类神经网络在节点分类任务中的表现进行比较分析,以此来展现图神经网络的强大和论证图神经网络强于普通深度神经网络的原因。

本文内容如下:

1.首先,做准备工作,获取并分析数据集构建一个方法用于分析节点表征的分布

2.然后,考察MLP神经网络用于节点分类的表现,并观察基于MLP神经网络学习到的节点表征的分布

3.接着,我们逐一介绍GCN,GAT这两个图神经网络的理论对比它们在节点分类任务中的表现以及它们学习到的节点表征的质量

4.最后,我们比较了三者在节点表征学习能力上的差异

一、准备工作

1.获取并分析数据集

图1.数据集情况

我们可以看到,Cora图拥有2708个节点和10556条边,平均节点度为3.9,训练集仅使用了140个节点,占整体的5%。我们还可以看到,这个图是无向图,不存在孤立的节点。

数据转换(transform)在将数据输入到神经网络之前修改数据,这一功能可用于实现数据规范化或数据增强。在此例子中,我们使用NormalizeFeatures进行节点特征归一化,使各节点特征总和为1。其他的数据转换方法请参阅torch-geometric-transforms。

2.可视化节点表征分布的方法

图2.定义可视化代码

一个可视化高纬数据的工具:t-distributed Stochastic Neighbor Embedding

二、使用MLP神经网络进行节点分类

理论上,我们应该能够仅根据文章的内容,即它的词包特征表征(bag-of-words feature representation)来推断文章的类别,而无需考虑文章之间的任何关系信息。接下来,让我们通过构建一个简单的MLP神经网络来验证这一点。此神经网络只对输入节点的表征做变换,它在所有节点之间共享权重。

1.MLP神经网络的构造

图3.构造MLP神经网络

MLP神经网络由两个线性层、一个ReLU非线性层和一个dropout操作组成。第一个线性层将1433维的节点表征嵌入(embedding)到低维空间(hidden_channels=16),第二个线性层将节点表征嵌入到类别空间中(num_classes=7)。

2.MLP神经网络的训练

使用交叉熵损失函数Adam优化器来训练这个简单的MLP神经网络。

图4.MLP神经网络的训练

3.MLP神经网络的测试

训练完模型后,我们可以通过测试来检验这个简单的MLP神经网络在测试集上的表现。

图5.MLP神经网络的测试

正如我们所看到的,我们的MLP表现得不太好艾,只有大概59%的测试准确性。为什么MLP没有表现得更好呢?其中一个重要的原因就是用于训练此神经网络的有标签节点数量过少,此神经网络被过拟合,它对未见过的节点泛化能力很差。

三、使用GCN卷积图神经网络进行节点分类

图6.GCN原理
图7.GCN模块说明

提出GCN的论文:Semi-supervised Classification with Graph Convolutional Network

GCNConv官方文档:torch_geometric.nn.conv.GCNConv

1.GCN图神经网络的构造

将MLP中的torch.nn.Linear替换为torch_geometric.nn.GCNConv,我们就可以得到一个GCN图神经网络。

图8.构造GCN卷积图神经网络

2.可视化由未经训练的GCN图神经网络生成的节点表征

图9.未经训练的GCN图神经网络生成的节点表征的可视化

经过visualize函数的处理,7维度特征的节点被映射到2维的平面上。我们可以发现“同类节点群聚”的现象。

3.GCN图神经网络的训练

图10.GCN的训练

4.GCN图神经网络的测试

图11.GCN的测试

通过简单地将torch.nn.Linear替换成torch_geometric.nn.GCNConv,我们可以取得80%左右的测试准确率。与前面的仅获得59%的测试准确率的MLP图神经网络相比,GCN图神经网络准确性要高得多。这表明节点的邻接信息在取得更好的准确率方面起着关键的作用

5.可视化由训练后的GCN图神经网络生成的节点表征

图12.训练后的GCN图神经网络生成的节点表征的可视化

训练后的GCN图神经网络生成的节点表征的可视化中,“同类节点群聚”的现象更加明显了。这意味着在训练后,GCN图神经网络生成的节点表征质量更高了。

四、使用GAT图注意力神经网络进行节点分类

图13.GAT定义
图14.GATConv模块说明

提出GAT的论文: Graph Attention Networks

1.GAT图神经网络的构造

图15.GAT图神经网络

2.可视化由未经训练的GAT图神经网络生成的节点表征

图16.未经训练的GAT图神经网络生成的节点表征的可视化

3.GAT图神经网络的训练

图17.GAT图神经网络的训练

4.GAT图神经网络的测试

图18. GAT图神经网络的测试

5.可视化由训练后的GCN图神经网络生成的节点表征

图19.训练后的GCN图神经网络生成的节点表征的可视化

五、总结

在节点表征的学习中,MLP神经网络只考虑了节点自身属性,忽略了节点之间的连接关系,它的结果是最差的;而GCN图神经网络与GAT图神经网络,同时考虑了节点自身信息与周围邻接节点的信息,因此它们的结果都优于MLP神经网络。也就是说,对周围邻接节点的信息的考虑,是图神经网络优于普通深度神经网络的原因

GCN图神经网络与GAT图神经网络的相同点为:

1.它们都遵循消息传递范式

2.在邻接节点信息变换阶段,它们都对邻接节点做归一化和线性变换;

3.在邻接节点信息聚合阶段,它们都将变换后的邻接节点信息做求和聚合;

4.在中心节点信息变换阶段,它们都只是简单返回邻接节点信息聚合阶段的聚合结果。

GCN图神经网络与GAT图神经网络的区别在于邻接节点信息聚合过程中的归一化方法不同:

1.前者根据中心节点与邻接节点的度计算归一化系数,后者根据中心节点与邻接节点的相似度计算归一化系数。

2.前者的归一化方式依赖于图的拓扑结构:不同的节点会有不同的度,同时不同节点的邻接节点的度也不同,于是在一些应用中GCN图神经网络会表现出较差的泛化能力。

3.后者的归一化方式依赖于中心节点与邻接节点的相似度,相似度是训练得到的,因此不受图的拓扑结构的影响,在不同的任务中都会有较好的泛化表现。

六、作业

图20.作业

这里我用了另一种PyG中的图卷积模块——SAGEConv,即在导入库那里改成from torch_geometric.nn import SAGEConv

图21.SAGEConv构造

模型训练和测试的代码与GCN和GAT的一样,这里就不写出来拉!最后得到的测试数据集准确率和用训练后的GAT图神经网络生成的节点表征的可视化结果如下图所示:

图22. SAGEConv测试结果和节点表征可视化情况
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,293评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,604评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,958评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,729评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,719评论 5 366
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,630评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,000评论 3 397
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,665评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,909评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,646评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,726评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,400评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,986评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,959评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,197评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,996评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,481评论 2 342

推荐阅读更多精彩内容