图注意网络GAT理解及Pytorch代码实现【PyGAT代码详细注释】

GAT

题&＃xff1a;Graph Attention Networks
摘要&＃xff1a;
提出了图形注意网络(GAT) &＃xff0c;这是一种基于图结构数据的新型神经网络结构&＃xff0c;利用掩蔽的自我注意层来解决基于图卷积或其近似的先前方法的缺点。通过叠加层&＃xff0c;节点能够参与其邻域的特征&＃xff0c;我们能够(隐式地)为邻域中的不同节点指定不同的权重&＃xff0c;而不需要任何代价高昂的矩阵操作(如反演) &＃xff0c;或者依赖于预先知道图的结构。通过这种方法&＃xff0c;我们同时解决了基于谱的图形神经网络的几个关键问题&＃xff0c;并使我们的模型容易地适用于归纳和转导问题。我们的 GAT 模型已经实现或匹配了四个已建立的转导和归纳图基准的最新结果: Cora&＃xff0c;Citeseer 和 Pubmed 引用网络数据集&＃xff0c;以及protein-protein interaction dataset(其中测试图在训练期间保持不可见)。

在Paper with code 网址&＃xff0c;可找到对应论文和github源码&＃xff0c;原论文使用TensorFlow实现&＃xff0c;本篇主要对Pytorch版本的 PyGAT附详细注释帮助理解和测试。

GitHUb: keras版本实现
Pytorch版本实现 PyGAT

截图及下文代码注释参考自视频&＃xff1a;GAT详解及代码实现

视频中的eij的实现与源码不同&＃xff0c;视频中是先拼接两个W&＃xff0c;再与a乘&＃xff1b;
源码在_prepare_attentional_mechanism_input()函数中先分别与a乘&＃xff0c;再拼接。

代码实现【PyGAT】

在PyGAT &＃xff1a;

• layers.py中定义Simple GAT layer实现&＃xff08;GraphAttentionLayer&＃xff09;和Sparse version GAT layer实现&＃xff08;SpGraphAttentionLayer&＃xff09;。
• models.py 实现两个版本加入Multi-head机制
• trains.py 使用model定义的GAT构建模型进行训练&＃xff0c;使用cora数据集

GraphAttentionLayer【一个图注意力层实现】

class GraphAttentionLayer(nn.Module):
"""
Simple GAT layer, similar to https://arxiv.org/abs/1710.10903
"""
def __init__(self, in_features, out_features, dropout, alpha, concat&＃61;True):
super(GraphAttentionLayer, self).__init__()
self.dropout &＃61; dropout
self.in_features &＃61; in_features#结点向量的特征维度
self.out_features &＃61; out_features#经过GAT之后的特征维度
self.alpha &＃61; alpha#dropout参数
self.concat &＃61; concat#LeakyReLU参数
# 定义可训练参数&＃xff0c;即论文中的W和a
self.W &＃61; nn.Parameter(torch.empty(size&＃61;(in_features, out_features)))
nn.init.xavier_uniform_(self.W.data, gain&＃61;1.414)# xavier初始化
self.a &＃61; nn.Parameter(torch.empty(size&＃61;(2*out_features, 1)))
nn.init.xavier_uniform_(self.a.data, gain&＃61;1.414)# xavier初始化
# 定义leakyReLU激活函数
self.leakyrelu &＃61; nn.LeakyReLU(self.alpha)
def forward(self, h, adj):
&＃39;&＃39;&＃39;
adj图邻接矩阵&＃xff0c;维度[N,N]非零即一
h.shape: (N, in_features), self.W.shape:(in_features,out_features)
Wh.shape: (N, out_features)
&＃39;&＃39;&＃39;
Wh &＃61; torch.mm(h, self.W) # 对应eij的计算公式
e &＃61; self._prepare_attentional_mechanism_input(Wh)#对应LeakyReLU(eij)计算公式
zero_vec &＃61; -9e15*torch.ones_like(e)#将没有链接的边设置为负无穷
attention &＃61; torch.where(adj > 0, e, zero_vec)#[N,N]
# 表示如果邻接矩阵元素大于0时&＃xff0c;则两个节点有连接&＃xff0c;该位置的注意力系数保留
# 否则需要mask设置为非常小的值&＃xff0c;因为softmax的时候这个最小值会不考虑
attention &＃61; F.softmax(attention, dim&＃61;1)# softmax形状保持不变[N,N]&＃xff0c;得到归一化的注意力全忠&＃xff01;
attention &＃61; F.dropout(attention, self.dropout, training&＃61;self.training)# dropout,防止过拟合
h_prime &＃61; torch.matmul(attention, Wh)#[N,N].[N,out_features]&＃61;>[N,out_features]
# 得到由周围节点通过注意力权重进行更新后的表示
if self.concat:
return F.elu(h_prime)
else:
return h_prime
def _prepare_attentional_mechanism_input(self, Wh):
# Wh.shape (N, out_feature)
# self.a.shape (2 * out_feature, 1)
# Wh1&2.shape (N, 1)
# e.shape (N, N)
# 先分别与a相乘再进行拼接
Wh1 &＃61; torch.matmul(Wh, self.a[:self.out_features, :])
Wh2 &＃61; torch.matmul(Wh, self.a[self.out_features:, :])
# broadcast add
e &＃61; Wh1 &＃43; Wh2.T
return self.leakyrelu(e)
def __repr__(self):
return self.__class__.__name__ &＃43; &＃39; (&＃39; &＃43; str(self.in_features) &＃43; &＃39; -> &＃39; &＃43; str(self.out_features) &＃43; &＃39;)&＃39;


用上面实现的单层网络测试

x &＃61; torch.randn(6,10)
adj&＃61;torch.tensor([[0,1,1,0,0,0],
[1,0,1,0,0,0],
[1,1,0,1,0,0],
[0,0,1,0,1,1],
[0,0,0,1,0,0,],
[0,0,0,1,1,0]])
my_gat &＃61; GraphAttentionLayer(10,5,0.2,0.2)
print(my_gat(x,adj))


输出&＃xff1a;
tensor([[-0.2965, 2.8110, -0.6680, -0.9643, -0.9882],
[ 0.0000, 0.0000, 0.0000, 0.0000, 0.0000],
[-0.4981, -0.7515, 1.1159, 0.3546, 1.3592],
[ 0.4679, 1.7208, 0.3084, -0.5331, -0.1291],
[-0.4375, -0.8778, 1.1767, -0.5869, 1.5154],
[-0.2164, -0.5897, 0.4988, -0.3125, 0.6423]], grad_fn&＃61;<EluBackward>)


加入Multi-head机制的GAT

用不同head捕捉不同特征&＃xff0c;使模型有更好的拟合能力。

class GAT(nn.Module):
def __init__(self, nfeat, nhid, nclass, dropout, alpha, nheads):
"""Dense version of GAT."""
super(GAT, self).__init__()
self.dropout &＃61; dropout
# 加入Multi-head机制
self.attentions &＃61; [GraphAttentionLayer(nfeat, nhid, dropout&＃61;dropout, alpha&＃61;alpha, concat&＃61;True) for _ in range(nheads)]
for i, attention in enumerate(self.attentions):
self.add_module(&＃39;attention_{}&＃39;.format(i), attention)
self.out_att &＃61; GraphAttentionLayer(nhid * nheads, nclass, dropout&＃61;dropout, alpha&＃61;alpha, concat&＃61;False)
def forward(self, x, adj):
x &＃61; F.dropout(x, self.dropout, training&＃61;self.training)
x &＃61; torch.cat([att(x, adj) for att in self.attentions], dim&＃61;1)
x &＃61; F.dropout(x, self.dropout, training&＃61;self.training)
x &＃61; F.elu(self.out_att(x, adj))
return F.log_softmax(x, dim&＃61;1)


对数据集Cora的处理

数据集中两个文件&＃xff0c;cites&＃xff1a;比如上图11行&＃xff1a;编号25和编号1114331的文章
content文件&＃xff1a;如下图&＃xff0c;每篇文章的id、features及类别

csr_matrix()处理稀疏矩阵

utils.py中对数据进行的处理

#数据是稀疏的&＃xff0c;csr_matrix操作从行开始将1的位置取出来,对数据进行压缩

features &＃61; sp.csr_matrix(idx_features_labels[:, 1:-1], dtype&＃61;np.float32)
labels &＃61; encode_onehot(idx_features_labels[:, -1])


encode_onehot()对label编号

有7个类别&＃xff0c;通过classes_dict是7*7的对角阵把每个类别映射成不同向量&＃xff0c;对所有label进行编号&＃xff0c;再将编号转换为one_hot向量

def encode_onehot(labels):
# The classes must be sorted before encoding to enable static class encoding.
# In other words, make sure the first class always maps to index 0.
classes &＃61; sorted(list(set(labels)))
classes_dict &＃61; {c: np.identity(len(classes))[i, :] for i, c in enumerate(classes)}
labels_onehot &＃61; np.array(list(map(classes_dict.get, labels)), dtype&＃61;np.int32)
return labels_onehot


build graph

见注释&＃xff1a;

# build graph
idx &＃61; np.array(idx_features_labels[:, 0], dtype&＃61;np.int32)#获取所有文章id
idx_map &＃61; {j: i for i, j in enumerate(idx)}#按文章数目&＃xff0c;对id重新映射
# 读取数据集中文章和文章直接的引用关系
edges_unordered &＃61; np.genfromtxt("{}{}.cites".format(path, dataset), dtype&＃61;np.int32)
# 根据idx_map,将文章引用关系也重新映射
edges &＃61; np.array(list(map(idx_map.get, edges_unordered.flatten())), dtype&＃61;np.int32).reshape(edges_unordered.shape)
adj &＃61; sp.coo_matrix((np.ones(edges.shape[0]), (edges[:, 0], edges[:, 1])), shape&＃61;(labels.shape[0], labels.shape[0]), dtype&＃61;np.float32)
# build symmetric adjacency matrix 生成邻接矩阵
adj &＃61; adj &＃43; adj.T.multiply(adj.T > adj) - adj.multiply(adj.T > adj)
features &＃61; normalize_features(features)
adj &＃61; normalize_adj(adj &＃43; sp.eye(adj.shape[0]))


邻接矩阵构造

csr_matrix()只记录了&＃xff08;0&＃xff0c;1&＃xff09;1,忽略了&＃xff08;1&＃xff0c;0&＃xff09;1。所以需要coo_matrix()操作&＃xff01;才能还原出无向图的邻接矩阵&＃xff01;

本文的一些代码注释及截图还可见视频

一个拓展&＃xff1a;

GAT的推广

GAT的推广
GAT仅仅是应用在了单层图结构网络上&＃xff0c;我们是否可以将它推广到多层网络结构呢&＃xff1f;

这里我们假设一个有N层网络的结构&＃xff0c;每层网络都定义了相同的节点&＃xff0c;但是节点之间的关系有所差异。举一个简单的例子&＃xff0c;假设有一个用户关系型网络&＃xff0c;每层网络的节点都被定义成了网络中的用户&＃xff0c;网络的第一层视图的关系可以定义为&＃xff0c;两个用户之间是否具有好友关系&＃xff1b;网络的第二层视图可以定义为&＃xff0c;你评论过我的动态&＃xff1b;网络的第三层视图可以定义为你转发过我的动态&＃xff1b;第四层关系可以定义为&＃xff0c;你at过我等等。

通过这样的定义我们就完成了一个多层网络的构建&＃xff0c;他们共享相同的节点&＃xff0c;但又分别具有不同的邻边&＃xff0c;如果我们分别处理每一层视图视图&＃xff0c;然后将他们得出的节点表示单纯相加的话&＃xff0c;就可能会失去不同视图之间的协作关系&＃xff0c;降低分类&＃xff08;预测&＃xff09;的精度。

基于以上观点&＃xff0c;我们提出了一种新的方法&＃xff1a;首先在每一层单视图中应用GAT进行学习&＃xff0c;并计算出每层视图的节点表示。之后再不同视图之间引入attention机制来让网络自行学习不同视图的权重。之后根据学习的权重&＃xff0c;将各个视图加权相加得到全局节点表示并进行后续的诸如节点表示&＃xff0c;链接预测等任务。

同时&＃xff0c;因为不同视图共享同样的节点&＃xff0c;即使每一层视图都表示了不同的节点关系&＃xff0c;最终得到的每一层的节点嵌入表示应具有一定的相关性。基于以上理论&＃xff0c;我们在每层GAT的网络参数间引入正则化项来约束参数&＃xff0c;使其向互相相近的方向学习。大致的网络流程图如下&＃xff1a;

这部分来源于 链接&＃xff1a;https://www.jianshu.com/p/d5d366ba1a57 来源&＃xff1a;简书