2019 WWW | HAN:Heterogeneous Graph Attention Network
Paper: https://arxiv.org/pdf/1903.07293.pdf
Code: https://github.com/Jhy1993/HAN
2019 WWW | HAN:异构图注意网络
这篇文章在图神经网络中将注意力机制从同质图扩展到节点和边有不同类型的异质图。文章提出了一种包含节点级注意力和语义级注意力的层次注意力异质图神经网络。
定义
异构图是指含有不同的节点和不同类型的边的图网络,比如下面的图包含电影、演员、导演三种类型的节点,两个电影节点之间可以通过元路径“电影-演员-电影”关联,该元路径表示共同演员的关系,也可以通过元路径“电影-导演-电影”关联,该原路径表示的是共同导演的关系,可以看出在异构图中两个相同的节点对应的不同的元路径具有不同的语义信息。
元路径(Meta-path):指从一个节点集合 [ A 1 , A 2 , . . . , A l + 1 ] [A_1, A_2, ..., A_{l+1}] [A1,A2,...,Al+1]通过关系 [ R 1 , R 2 , . . . , R l ] [R_1, R_2, ... , R_l] [R1,R2,...,Rl]连接成的一条路径 ϕ \phi ϕ。
基于元路径的邻居(Meta-path based Neighbors):在异构图中,节点 i i i的基于元路径 ϕ \phi ϕ的邻居 N i ϕ N_i^{\phi} Niϕ是指通过元路径(meta-path)和 i i i相连的节点的集合,包含自己本身。
其他
模型
节点级的注意力
对于
ϕ
i
\phi_i
ϕi类型的节点,作者设计了转换矩阵
M
ϕ
i
M_{\phi_i}
Mϕi,将节点表示由
h
i
h_i
hi转换成
h
i
′
h_i'
hi′
通过一层注意力网络
a
t
t
e
n
n
o
d
e
atten_{node}
attennode学习得到节点j对节点i的基于meth-path
Φ
\Phi
Φ 的权重
e
i
j
Φ
e^{\Phi}_{ij}
eijΦ。
之后通过softmax得到归一化权重
α
i
j
Φ
\alpha^{\Phi}_{ij}
αijΦ
聚合邻居节点的特征得到节点
i
i
i基于
Φ
\Phi
Φ的语义信息的节点表示向量
z
i
Φ
z_i^{\Phi}
ziΦ
与GAT相同,为了让节点注意力机制更加鲁棒,作者设计了多头注意力机制,然后将每次学习到的结果进行concat,得到多头注意力的节点表示:
语义级的注意力
为了学习不同meta-path对应的节点表示的注意力权重,作者首先对不同meta-path对应的节点表示
z
i
Φ
p
z_i^{\Phi_p}
ziΦp进行非线性变换,然后定义语义注意力向量q,将q和非线性变换后的节点向量内积的平均值作为meta-path
Φ
p
\Phi_p
Φp的权重
w
Φ
p
w_{\Phi_p}
wΦp:
softmax得到语义注意力权重
β
Φ
p
\beta_{\Phi_p}
βΦp
最后基于语义注意力权重,得到最终的节点表示
Z
Z
Z:
伪代码流程如下: