课程网址
斯坦福CS224W图机器学习、图神经网络、知识图谱【同济子豪兄】
斯坦福大学CS224W图机器学习公开课-同济子豪兄中文精讲
cs224w(图机器学习)2021冬季课程学习笔记集合
到图神经网络GCN为止的内容参考了斯坦福CS224W图机器学习、图神经网络、知识图谱【同济子豪兄】,后面的内容根据课程网址的ppt课件学习整理。
图是描述各种关联现象的通用语言。与传统数据分析中的样本服从独立同分布假设不一样,图数据自带关联结构,数据和数据,样本和样本之间有联系。
图神经网络是端到端的表示学习,无需人工特征工程,可以自动学习特征(类似CNN)。
图神经网络的目标是实现图嵌入,即将一个节点映射成d维向量,同时保证网络中相似的节点有相近的向量表示。这个d维向量应该包含节点在原图中的结构信息,语意信息,以方便后续的数据挖掘。
节点、连接、子图、全图都可以带有特征
在节点、连接、子图、全图层面,都可以进行图数据挖掘
图(network/graph)由节点(nodes/vertices)和连接(links/edges)组成。
节点的集合用
N
表示,连接的集合用E
表示,整个图用G(N,E)
表示。
图的设计牵涉到一个概念,本体图(Ontology)。本体图应该显示节点可能的类型,以及各类型节点(包括节点类型到其自身)之间可能存在的关系。如下图,是一个医疗领域的知识图谱的本体图。
如何设计本体图,取决于要解决的问题。如下图,例如要解决的问题是,什么疾病可以吃什么,那么疾病和食物就需要设计成节点。可以吃,不可以吃, 不推荐吃就应该设计成节点之间的关系
有时本体图是唯一、无歧义的,如社交网络
有时本体图不唯一,取决于你要研究的问题,如考虑红楼梦的家族,地点,事件等
图可以分为无向图、有向图、异质图(heterogeneous graph)、二分图。
二分图可以展开,如下图,在节点集U中,如果两个节点都连接到V中的同一个节点,则在图
Projection U
中添加一条连接。
这样就可以将二分图转化为两张各自只有一类节点的图。
对于无向图
i
个节点的度:记为 k i k_i ki,表示与第i
个节点邻接的边的数量对于有向图
平均入度=平均出度,因为一个出度对应的边必然对应一个入度对应的边
入度为0的节点称为Source,出度为0的节点称为Sink
节点的度可以一定程度上反应节点的重要程度
i
个节点和第j
个节点之间存在边,则邻接矩阵A
的 A i j A_{ij} Aij和 A j i A_{ji} Aji对应的值为1无向图对应的邻接矩阵是对称阵
如果没有自身到自身的环,则主对角线全为0
i
个节点存在指向第j
个节点之间的边,则邻接矩阵A
的 A i j A_{ij} Aij对应的值为1有向图对应的邻接矩阵是非对称阵
如上面所示,用邻接矩阵来表示图,存在稀疏性问题,造成存储空间的浪费。
上述图是无权图,在邻接矩阵中的值非1即0,如果是带权图,则可以在邻接矩阵中将1改成权重。
无向图中
连通图(Connected graph):任意两个点都有一条路径可达,则称为连通图。
不连通图虽然本身不连通,但是可以划分得到多个连通分支(connected components)。
最大的连通分支被称为Giant Component
孤立的节点称为Isolated node。
不连通矩阵的邻接矩阵呈现出分块对角的形式
如果存在一个节点将不连通图的两个连通分支连接上了,那么它会打破分块对角形式
有向图中
强连通的有向图:如果任意两个节点存在有向路径可达,则称为强连通的有向图。
弱连通的有向图:如果忽略边的方向,即将它看成无向图,此时如果图是连通的,那么这个有向图称为弱连通的。
对于有向图,可能整体不是强连通的,但其中的某个子图是强连通的,称为强连通分支(Strongly connected components(SCC))
E和G指向SCC,称为In-componet
、D和F是由SCC出发的,称为Out-component
。
总的思路
本节用传统机器学习方法,做特征工程,人工设计一些特征,把节点,边,全图特征编码成d
维向量,再用该向量进行后续机器学习预测。
1.特征工程
抽取d
个特征,编码为d
维向量。本节只考虑连接特征,不考虑属性特征。
节点自己的特征,称为属性特征(Attributes)
节点和图中其他节点的连接关系,称为连接特征。
2.训练一个机器学习模型
利用RF、SVM、NN等进行训练。
3.应用模型
给定一个新的节点/链接/图,获得图她的特征并做预测。
本节主要聚焦无向图,并针对节点、边、图层面做特征工程。
目标:区分节点在图中的结构和位置,可考虑的(连接)特征有:节点的度(Node degree)、节点的重要度(Node centrality)、聚类系数(Clustering coefficient)、子图模式(Graphlets)
聚合系数是指,与当前节点邻接的节点是否有联系。
Node degree只考虑了邻接节点的数量,不能反应节点的质量
Node centrality考虑了节点在图中的重要度。有不同的方式来对此进行建模:
特征向量重要度(Eigenvector centrality)
思想:如果一个节点的邻接节点很重要,那么这个节点也很重要
建模:
c v = 1 λ ∑ u ∈ N ( v ) c u c_v=\frac{1}{\lambda}\sum_{u \in N(v)} c_u cv=λ1u∈N(v)∑cu
λ \lambda λ是归一化系数,往往是A的最大特征值
实现:这是一个递归问题,如何解决?
上面公式等价于求解 λ c = A c \lambda \boldsymbol{c}=\boldsymbol{A c} λc=Ac
可以发现, c \boldsymbol{c} c向量就是 A \boldsymbol{A} A的最大特征值对应的特征向量
根据perron-frobenius定理,最大特征值 λ m a x \lambda_{max} λmax一定为正且唯一
Betweenness centrality
思想:如果在任意两个节点间的最短路径中,有一个节点频繁出现,那么这个节点可以被认为是重要的
建模:
c v = ∑ s ≠ v ≠ t # ( shortest paths betwen s and t that contain v ) # ( shortest paths between s and t ) c_v=\sum_{s \neq v \neq t} \frac{\#(\text { shortest paths betwen } s \text { and } t \text { that contain } v)}{\#(\text { shortest paths between } s \text { and } t)} cv=s=v=t∑#( shortest paths between s and t)#( shortest paths betwen s and t that contain v)
Closeness centrality
思想:如果一个节点到其他所有节点的路径都很短,那么这个节点可以被认为是重要的
建模:
c v = 1 ∑ u ≠ v shortest path length between u and v c_v=\frac{1}{\sum_{u \neq v} \text { shortest path length between } u \text { and } v} cv=∑u=v shortest path length between u and v1
聚类系数(Clustering Coefficient),衡量一个节点的邻接节点的连接有多紧密。
建模:
e v = # ( edges among neighboring nodes ) ( k v 2 ) ∈ [ 0 , 1 ] e_v=\frac{\#(\text{edges among neighboring nodes})}{\binom{k_v}{2}}\in [0, 1] ev=(2kv)#(edges among neighboring nodes)∈[0,1]
一个节点的自我网络(ego-network)是指以一个节点为中心,只包含他和他邻接节点,以及这些节点之间的边的图。
可以发现,节点v的聚类系数本质上就是计数了节点v的自我网络中以v为顶点的三角形的个数。
这个三角形可以理解为我们预先定义的一类子图。
那么如果修改这个预定义的子图类型,就可以得到新的计数特征,这个预定义的子图类型,就是我们下面要提到的graphlets。
可以看到,从原图中取一些节点,并取这些节点所有出现的边可以构成导出子图。
下面给出Graphlets的精确定义,即有根连通导出异构子图(Rooted connected induced non-isomorphic subgraphs)
上图分别展示了2节点、3节点、4节点和5节点的graphlets,共有73种。
2个节点构成的子图中,可以定义1种类型的graphlet
3个节点构成的子图中,可以定义3种类型的graphlets
…
在右上角的例子中,节点u对应的graphlets类型有0、1、2、3、5、10、11、…
聚类系数中的三角形其实就是G2对应的graphlet。
下面引入Graphlets相关的特征向量:Graphlets Degree Vector(GDV),它一个基于给定节点,以它为根的各类graphlets的实例个数组成向量,如下面的例子。
注意,原图中没有以c为根结点的导出子图。
GDV描述了节点局部领域的拓扑结构信息,用一些已经定义好的子图模式去匹配,并计数每种模式下的数量。
比较两个节点的GDV向量,可以计算距离和相似度。
在NetworkX中,子图模式Graphlets被称为Atlas
介绍的结构特征可以分为:
基于重要度的特征(描述节点中心度/重要度):
可用于预测有影响力的节点
基于结构的特征(描述节点的邻域拓扑连接结构):
可用于预测节点在图中的功能,桥接、枢纽、中心
连接层面的预测任务:基于已知连接去预测(补全)未知连接。
在模型训练阶段,节点对被排序,top K节点对被预测。
关键是如何设计节点对的特征。
思路1:直接提取link的特征,变成d维向量。
思路2:把link两端的d维向量拼在一起,但是这会丢失link本身连接结构信息。
link prediction task有两种情况:
随机丢失连接:
对于客观静态图,如蛋白质,分子,我们可以通过随机移除一些连接,并尝试预测它们
随时间变化的连接:
对于如论文引用、社交网络、微信好友、学术合作等图,给定一段时间 [ t 0 , t 0 ′ ] [t_0, t_0^{'}] [t0,t0′]的图,预测下一个时间段 [ t 1 , t 1 ′ ] [t_1, t_1^{'}] [t1,t1′]的一个关于边的ranked list L
。
评估的方式:先计算得到 [ t 1 , t 1 ′ ] [t_1, t_1^{'}] [t1,t1′]内真实出现的边的数量,记为 n = ∣ E n e w ∣ n=|E_{new}| n=∣Enew∣,然后从上面预测的列表中选出top n条边,然后计算预测的n个连接的准确率。
准确率 = 预测的 t o p n 个连接中正确的数量 n 准确率 = \frac{预测的top\ n个连接中正确的数量}{n} 准确率=n预测的top n个连接中正确的数量
连接的特征可以分为三类:基于距离的特征、基于两节点局部邻域信息的特征、基于两节点全局领域信息的特征
但仅考虑最短路径长度,会忽略连接的质量。如同样最短路径长度是2,A和B只有一条通路,而B和H有两条。
考虑两个节点v1和v2的邻接节点。
Common neighbors:
思路:记录共同好友个数
公式:
∣ N ( v 1 ) ∩ N ( v 2 ) ∣ \left|N\left(v_1\right) \cap N\left(v_2\right)\right| ∣N(v1)∩N(v2)∣
Jaccard’s coefficient:
思路:共同好友个数/两节点邻接节点的并集
公式:
∣ N ( v 1 ) ∩ N ( v 2 ) ∣ ∣ N ( v 1 ) ∪ N ( v 2 ) ∣ \frac{\left|N\left(v_1\right) \cap N\left(v_2\right)\right|}{\left|N\left(v_1\right) \cup N\left(v_2\right)\right|} ∣N(v1)∪N(v2)∣∣N(v1)∩N(v2)∣
Adamic-Adar index:
思路:共同好友是不是社牛,如果v1和v2的共同好友是社牛,那么v1和v2的联系就很廉价。
公式:
∑ u ∈ N ( v 1 ) ∩ N ( v 2 ) 1 log ( k u ) \sum_{u \in N\left(v_1\right) \cap N\left(v_2\right)} \frac{1}{\log \left(k_u\right)} u∈N(v1)∩N(v2)∑log(ku)1
基于两节点的局部邻域信息的特征的缺点是:对于没有共同好友的两节点,他们的上述度量都是0。
但事实上,他们在未来可能会有连接。
而全局领域的信息度量可以解决这个缺陷。
Katz index:
思路:计数节点u和v之间所有长度路径的加权和
可以使用图邻接矩阵的幂可以结算长度为k的路径个数
结合下图,利用数学归纳法,可以推导出 A u v l A_{u v}^l Auvl表示节点u和v之间长度为l的路径个数。
公式:
S v 1 v 2 = ∑ l = 1 ∞ β l A v 1 v 2 l S_{v_1 v_2}=\sum_{l=1}^{\infty} \beta^{l} A_{v_1 v_2}^l Sv1v2=l=1∑∞βlAv1v2l
其中, 0 < β < 1 0<\beta<1 0<β<1表示折减系数
它的等价矩阵形式是(类比等比数列求和,并求无穷级数可得):
( I − β A ) − 1 − I (\boldsymbol{I}-\beta \boldsymbol{A})^{-1}-\boldsymbol{I} (I−βA)−1−I
一般可以将最大特征值的倒数作为折减系数 β \beta β
目标:将全图 G G G的结构特点表示为一个d维特征向量 ϕ ( G ) \phi(G) ϕ(G)。
思路:类比NLP中的Bag-of-Words,
注意,这里是从全图的视角去分析,所以这里的graphlets和前面在节点特征工程中提到的graphlets有两点不同:
可以存在孤立节点的graphlets
graphlets不区分根,如下图,g2对应一个graphlets,而不是两个(如果考虑根,是两个)
Graphlet Count Vector:给定一个图G,和graphlets列表 G k = ( g 1 , g 2 , . . . g n k ) G_k=(g_1,g_2, ... g_{n_k}) Gk=(g1,g2,...gnk),Graphlet Count Vector可以定义为(向量的第i个分量可以定义为第i个graphlet在全图中的个数):
( f G ) i = # ( g i ⊆ G ) for i = 1 , 2 , … , n k \left(f_G\right)_i=\#\left(g_i \subseteq G\right) \text { for } i=1,2, \ldots, n_k (fG)i=#(gi⊆G) for i=1,2,…,nk
例子:
给定两个图, G G G和 G ′ G' G′,且有了它们对应的GCV,进一步,可以计算Graphlet Kernel:
K ( G , G ′ ) = f G T f G ′ K\left(G, G^{\prime}\right)=\boldsymbol{f}_G^{\mathrm{T}} \boldsymbol{f}_{G^{\prime}} K(G,G′)=fGTfG′
它可以反应这两张图的关系。
如果两个GCV的数量级悬殊,那么则需要先对这两个特征向量作归一化: h G = f G Sum ( f G ) \boldsymbol{h}_G=\frac{\boldsymbol{f}_G}{\operatorname{Sum}\left(\boldsymbol{f}_G\right)} hG=Sum(fG)fG,再计算Graphlet Kernel: K ( G , G ′ ) = h G T h G ′ K\left(G, G^{\prime}\right)=\boldsymbol{h}_G{ }^{\mathrm{T}} \boldsymbol{h}_{G^{\prime}} K(G,G′)=hGThG′
获取GCV在算力上是很昂贵的,在大小为n的图上对大小为k的graphlet作子图匹配,需要的时间复杂度是多项式复杂度: O ( n k ) O(n^k) O(nk)。
即使图节点的度被限制为 d d d,复杂度也仍有 O ( n d k − 1 ) O(nd^{k-1}) O(ndk−1)
由于Graphlets Kernel不够高效,下面引入更高效的Weisfeiler-Lehman Kernel。
目标:设计一个更高效的特征编码。
思路:使用邻域结构迭代式地丰富节点词库
算法实现:颜色微调
主要的步骤是:
哈希表由两张图共同贡献
c ( K ) ( v ) c^{(K)}(v) c(K)(v)中包含了K跳邻域的信息。
总体而言,WL Kernel的时间复杂度是O(#(edges))。
核方法是传统技巧学习在图层面的预测的常用方法。它的核心是如何设计Kernel而非特征向量。
Kernel K ( G , G ′ ) K(G, G') K(G,G′)是标量,描述了数据间的相似度
核矩阵 K = ( K ( G , G ′ ) ) G , G ′ \boldsymbol{K}=\left(K\left(G, G^{\prime}\right)\right)_{G, G^{\prime}} K=(K(G,G′))G,G′永远半正定,即有正的特征值。
存在特征表示 ϕ ( ⋅ ) \phi(\cdot) ϕ(⋅)使得 K ( G , G ′ ) = ϕ ( G ) T ϕ ( G ′ ) K\left(G, G^{\prime}\right)=\phi(G)^{\mathrm{T}} \phi\left(G^{\prime}\right) K(G,G′)=ϕ(G)Tϕ(G′)
一定kernel确定了,现成的机器学习模型,如Kernel SVM就可以用来预测。
图表示学习减轻了做特征工程的工作。
映射得到的向量具有低维(向量维度远小于节点数)、连续(每个元素都是实数)、稠密(每个元素都不为0)的特点。
假设:G是图,V是节点集,A是无权图,本节仍仅考虑连接信息,不考虑节点信息。
目标是:节点编码后,两节点在嵌入空间中的向量的(余弦)相似度可以反应(近似)两节点在图中的相似度。即
s i m i l a r i t y ( u , v ) ≈ z v T z u similarity(u,v)\approx \mathbf{z}_v^{\mathrm{T}} \mathbf{z}_u similarity(u,v)≈zvTzu
关键:如何定义节点的相似度。
步骤:
node embeddings方法是无监督/自监督的,且与下游任务无关。
最简单的编码器-查表
只需要直接优化Z矩阵
对应方法有:DeepWalk,node2vec
包括deepwalk和node2vec
transductive直推式学习:整个过程没有新的节点进入,可以利用所有已有节点的信息,并且预测标签未知节点的标签
inductive归纳式学习:要对新的节点进行预测
本节考虑属性特征向量 f v f_v fv
LP(Label Propagation),ICA(Iterative classification)往往是用于被图神经网络做对比的模型。
上面的模型主要基于图的基本假设:在网络中,homophily是大量存在的,即相近的节点相似。
关于为什么网络中的节点行为是correlated的两种解释:
如何利用这种correlation进行分类?
节点的类别依赖于(1)节点的特征(2)节点邻域的标签(3)节点邻域的特征
挑战:不保证收敛;仅用到网络连接信息,没有用到节点的属性特征
LP/Relational classification没有用到节点的属性特征。
下面提到的ICA则既会用到属性特征 f v f_v fv,又会用到连接信息 z v z_v zv。
训练两个分类器
z v z_v zv的计算方法有:考虑领域节点标签数量的直方图;考虑领域节点最多的标签类、考虑领域节点不同标签的数量等
1.在已经标注好的训练集上,训练两个分类器。
2.迭代直到递归:
这个算法不能保证收敛
ICA算法可以被抽象为Collective Classification。它基于马尔科夫假设:一个节点是什么类别,取决于它的邻居节点的类别。
Collective Classification包含三步:
可以被视为一种后处理的技巧。在很多榜单中有很好的表现。
Normailized diffusion matrix A ~ ≡ D − 1 / 2 A D − 1 / 2 \widetilde{A} \equiv D^{-1 / 2} A D^{-1 / 2} A ≡D−1/2AD−1/2具有很好的性质:所有特征值在[-1,1];特征值为1的特征向量为 D 1 / 2 1 D^{1/2}\bold{1} D1/21 ( 1 \bold{1} 1是全1的向量); A ~ K \widetilde{\boldsymbol{A}}^K A K的特征值也都是[-1,1],且最大特诊值一定是1
s是超参数,最后再执行归一化
类似Correct step,不过矩阵换成了Z矩阵,它是表示标签的矩阵。
注意有标注的训练集的标签向量要修改成真实标签向量
C&S可以和GNN结合
label-label potential matrix ψ \psi ψ:反映了节点和它的邻居之间的依赖关系。矩阵的元素 ψ ( Y i , Y j ) \boldsymbol{\psi}\left(Y_i, Y_j\right) ψ(Yi,Yj)表示当邻居节点 i i i为类别 Y i Y_i Yi时,节点 j j j为类别 Y j Y_j Yj的概率。
Prior belief ϕ \phi ϕ: ϕ ( Y i ) \phi\left(Y_i\right) ϕ(Yi)是 i i i为类别 Y i Y_i Yi的概率。
m i → j ( Y j ) m_{i \rightarrow j}\left(Y_j\right) mi→j(Yj):表示节点 i i i认为节点 j j j是类别 Y j Y_j Yj的概率。
L \mathcal{L} L:所有类/标签的集合
图和节点的嵌入表示应该与节点的编号顺序无关。(置换不变性)
编号不一样,但本质上是同一个节点的两个节点,得到的图嵌入向量应该是一样的。
通过消息传递的框架设计具有置换不变性的图神经网络。
每个节点分别构建自己的计算图
每个计算图就是一个单独的样本
所有黑色盒子公用一套权重,所有白色盒子也公用一套权重
图神经网络的层数是指计算图的层数,而不是神经网络的层数,如上图是两层的图神经网络
k层图神经网络其实就是k hops的节点,如2层,就是看邻居的邻居。
图神经网络不能太深,如果太深,所有节点都会很类似,会产生过拟合。所以一般2-3层就够了
第0层的嵌入就是节点的属性特征
row normalized matrix的最大特征值为1
如果节点 i i i和节点 j j j是相连的,则权重为 1 d i d j \frac{1}{\sqrt{d_i} \sqrt{d_j}} didj1
这个矩阵也被称为Normalized diffusion matrix(Normalized Adjacency Matrix/Symmetric Normalized Matrix): A ~ ≡ D − 1 / 2 A D − 1 / 2 \widetilde{\boldsymbol{A}} \equiv D^{-1 / 2} \boldsymbol{A} D^{-1 / 2} A ≡D−1/2AD−1/2
在图中为节点添加到节点自身的环,并改进计算图。
还可以继续改进,对来自领域节点用一套权重,对来自环的自己的节点用另一套权重
GCN属于一种直推式(transductive)的学习,要求在一个确定的图中去学习顶点的embedding,无法直接泛化到在训练过程没有出现过的顶点。
GCN由于卷积的训练过程涉及到邻接矩阵、度矩阵(可理解为拉普拉斯矩阵),节点一旦变化,拉普拉斯矩阵随之变化,也就是你说的需要“重新计算前面的归一化矩阵”,然后重新训练模型,不能“活学活用”,所以是Transductive的。
通过矩阵分解或者随机游走获得节点嵌入的方法的缺点是:
CNN的公式可以改写成类似GNN的形式
CNN:
- CNN的领域是固定的
- CNN的卷积核权重需要学习获得
- CNN不满足置换不变性
GCN:- GCN的领域可以是任意的
- GCN的卷积核权重由 A ~ \tilde{A} A~预定义
- GCN满足置换不变性
transformer的核心概念是自注意力,使得每两个单词/节点间都可以互相影响,都可以有注意力权重。
transformer可以看成是一个全连接词图上的GNN
节点的邻域决定了计算图。
在计算图上传递和传播信息。
GNN层=消息+聚合
目的是为了以一些向量为输入,得到一个向量
每个节点产生一个消息,用于后续传递到其他节点。
消息函数: m u ( l ) = M S G ( l ) ( h u ( l − 1 ) ) \mathbf{m}_u^{(l)}=\mathrm{MSG}^{(l)}\left(\mathbf{h}_u^{(l-1)}\right) mu(l)=MSG(l)(hu(l−1))
举例:线性消息函数: m u ( l ) = W ( l ) h u ( l − 1 ) \mathbf{m}_u^{(l)}=\mathbf{W}^{(l)} \mathbf{h}_u^{(l-1)} mu(l)=W(l)hu(l−1)
每个节点会聚合来自其他节点的消息
聚合函数: h v ( l ) = A G G ( l ) ( { m u ( l ) , u ∈ N ( v ) } ) \mathbf{h}_v^{(l)}=\mathrm{AGG}^{(l)}\left(\left\{\mathbf{m}_u^{(l)}, u \in N(v)\right\}\right) hv(l)=AGG(l)({mu(l),u∈N(v)})
举例:AGG可以是 Sum ( ⋅ ) , Mean ( ⋅ ) \operatorname{Sum}(\cdot), \operatorname{Mean}(\cdot) Sum(⋅),Mean(⋅) or Max ( ⋅ ) \operatorname{Max}(\cdot) Max(⋅)聚合器中的任意一种
聚合的问题是忽略了 v v v节点自身的信息。解决办法是计算 h v ( l ) \mathbf{h}_v^{(l)} hv(l)时,考虑 h v ( l − 1 ) \mathbf{h}_v^{(l-1)} hv(l−1)。此时消息和聚合应改为:
消息: m u ( l ) = W ( l ) h u ( l − 1 ) m v ( l ) = B ( l ) h v ( l − 1 ) \mathbf{m}_u^{(l)}=\mathbf{W}^{(l)} \mathbf{h}_u^{(l-1)} \quad \mathbf{m}_v^{(l)}=\mathbf{B}^{(l)} \mathbf{h}_v^{(l-1)} mu(l)=W(l)hu(l−1)mv(l)=B(l)hv(l−1)
聚合:先利用AGG函数聚合领域的消息,再拼接聚合 v v v节点的消息
h v ( l ) = CONCAT ( AGG ( { m u ( l ) , u ∈ N ( v ) } ) , m v ( l ) ) \mathbf{h}_v^{(l)}=\operatorname{CONCAT}\left(\operatorname{AGG}\left(\left\{\mathbf{m}_u^{(l)}, u \in N(v)\right\}\right), \mathbf{m}_v^{(l)}\right) hv(l)=CONCAT(AGG({mu(l),u∈N(v)}),mv(l))
GCN论文中的经过改进后的消息函数其实是:
m u ( l ) = 1 ∣ N ( v ) ∣ ∣ N ( u ) ∣ W ( l ) h u ( l − 1 ) \mathbf{m}_u^{(l)}=\frac{1}{\sqrt{|N(v)|}\sqrt{|N(u)|}} \mathbf{W}^{(l)} \mathbf{h}_u^{(l-1)} mu(l)=∣N(v)∣∣N(u)∣1W(l)hu(l−1)
GraphSAGE的公式: h v ( l ) = σ ( W ( l ) ⋅ CONCAT ( h v ( l − 1 ) , AGG ( { h u ( l − 1 ) , ∀ u ∈ N ( v ) } ) ) ) \mathbf{h}_v^{(l)}=\sigma\left(\mathbf{W}^{(l)} \cdot \operatorname{CONCAT}\left(\mathbf{h}_v^{(l-1)}, \operatorname{AGG}\left(\left\{\mathbf{h}_u^{(l-1)}, \forall u \in N(v)\right\}\right)\right)\right) hv(l)=σ(W(l)⋅CONCAT(hv(l−1),AGG({hu(l−1),∀u∈N(v)})))
GAT的公式: h v ( l ) = σ ( ∑ u ∈ N ( v ) α v u W ( l ) h u ( l − 1 ) ) \mathbf{h}_v^{(l)}=\sigma\left(\sum_{u \in N(v)} \alpha_{v u} \mathbf{W}^{(l)} \mathbf{h}_u^{(l-1)}\right) hv(l)=σ u∈N(v)∑αvuW(l)hu(l−1)
其中 α v u \alpha_{v u} αvu是attention weights
为了使不同的邻居节点有不同的重要度,引入attention机制。
注意力权重的计算公式(基于节点的消息):
e v u = a ( W ( l ) h u ( l − 1 ) , W ( l ) h v ( l − 1 ) ) α v u = exp ( e v u ) ∑ k ∈ N ( v ) exp ( e v k ) e_{v u}=a\left(\mathbf{W}^{(l)} \mathbf{h}_u^{(l-1)}, \mathbf{W}^{(l)} \boldsymbol{h}_v^{(l-1)}\right) \\ \alpha_{v u}=\frac{\exp \left(e_{v u}\right)}{\sum_{k \in N(v)} \exp \left(e_{v k}\right)} evu=a(W(l)hu(l−1),W(l)hv(l−1))αvu=∑k∈N(v)exp(evk)exp(evu)
函数 a a a可以是单层神经网络,并和GAT的其他网络共同训练。
e A B = a ( W ( l ) h A ( l − 1 ) , W ( l ) h B ( l − 1 ) ) = Linear ( Concat ( W ( l ) h A ( l − 1 ) , W ( l ) h B ( l − 1 ) ) ) \begin{aligned} & e_{A B}=a\left(\mathbf{W}^{(l)} \mathbf{h}_A^{(l-1)}, \mathbf{W}^{(l)} \mathbf{h}_B^{(l-1)}\right) \\ & =\operatorname{Linear}\left(\operatorname{Concat}\left(\mathbf{W}^{(l)} \mathbf{h}_A^{(l-1)}, \mathbf{W}^{(l)} \mathbf{h}_B^{(l-1)}\right)\right)\end{aligned} eAB=a(W(l)hA(l−1),W(l)hB(l−1))=Linear(Concat(W(l)hA(l−1),W(l)hB(l−1)))
很多现代深度学习模块可以被引入GNN层,如:
参考资料:https://zhuanlan.zhihu.com/p/522525435
在GNN中,Dropout被应用在消息函数的线性层上
过多地堆叠GNN层会导致Over-smoothing问题,这是因为
如何克服Over-smoothing问题?
之前的假设都是输入的图可以直接转化为计算图。
但是如果存在下面的情况,就需要先对图进行处理。
对应的处理的方式有:
为什么需要特征增强。
还有其他指标也可以用于增强特征,如clustering coefficient、pagerank、centrality。
对于稀疏的图,需要增加虚拟节点或边
想法:消息传递时,随机采样领域节点用于消息传递
好处:大大降低计算复杂度,同时获得的节点嵌入的效果和将所有领域节点都用上的效果差不多。