在本节中,我们将介绍用于构建任意图注意力网络的构建块层(通过堆叠该层),并直接概述其与神经图处理领域的先前工作相比的理论和实践优势和局限性。
我们将从描述单个图形注意层开始,作为我们实验中使用的所有 GAT 架构中使用的唯一层。 我们使用的特定注意设置密切遵循 Bahdanau 等人的工作。 (2015)——但该框架与注意力机制的特定选择无关。
我们层的输入是一系列的节点特征,h={h1,h2,…,hN},hi∈RF,其中 N 是节点数,F 是每个节点中的特征数。该层产生一组新的节点特征(可能具有不同的基数F′),h′={h1′,h2′,…,hN′},hi′∈RF′,作为它的输出。
为了获得足够的表达能力将输入特征转化为更高层次的特征,至少需要一个可学习的线性变换。为此,作为初始步骤,将由_weight matrix_W∈RF′×F参数化的共享线性变换应用于每个节点。然后我们在节点上执行_self-attention_–一种共享的注意机制a:RF′×RF′→R计算_attention coefficients_:
eij=a(Whi,Whj)
这表明节点j的特征对节点i的_importance_。在其最一般的公式中,该模型允许每个节点参与每一个其他节点,dropping all structural information。我们通过执行_masked attention_将图结构注入到机制中–我们只为节点计算了eij,这里的j∈Ni,Ni是途中节点i的某个_neighborhood。_在我们的实验中,这都将确切地是i(包括i)的一阶邻居。为了使系数在不同节点之间易于比较,我们使用 softmax 函数在 j的所有选择中对它们进行归一化:
αij=softmaxj(eij)=∑k∈Niexp(eik)exp(eij)
在我们的实验中,注意力机制 a 是一个单层前馈神经网络,由权重向量a∈R2F′参数化,并应用 LeakyReLU 非线性(负输入斜率α=0.2)。完全展开后,注意力机制计算的系数(如图 1(左)所示)可以表示为:
αij=∑k∈Niexp(LeakyReLU(aT[Whi∥Whk]))exp(LeakyReLU(aT[Whi∥Whj]))
其中·T 表示转置,‖是拼接操作。
一旦获得,归一化的注意系数被用来计算对应于它们的特征的线性组合,以用作每个节点的最终输出特征(在潜在的应用可一个非线性激活函数σ之后):
hi′=σ(∑j∈NiαijWhj)
图1:左图:我们的模型使用的注意机制a(Whi,Whj),通过权重向量a∈R2F′初始化,应用LeakyReLU激活)。右图:节点1在其邻域上的多头注意(K=3个头)的插图。不同的箭头样式和颜色表示独立的注意力计算。来自每个头部的聚集特征被拼接或平均以获得h1′。
为了稳定自我注意的学习过程,我们发现扩展我们的机制以使用_multi-head attention_是有益的,类似于Vaswani等人。(2017年)。具体地说,K个独立的注意机制执行等式4(上面式子)的变换,然后将它们的特征拼接起来,产生以下输出特征表示:
hi′=∥k=1Kσ(∑j∈NiαijkWkhj)
其中,||表示拼接,αijK是由第k个注意机制(αk)计算的归一化注意系数,Wk是对应的输入线性变换的权重矩阵。请注意,在此设置中,最终返回的输出h′将由每个结点的KF′特征(而不是F′)组成。
特别是,如果我们在网络的最后(预测)层上执行多头注意,拼接不再合理-相反,我们使用_averaging_,并推迟应用最终的非线性激活函数(对于分类问题通常是Softmax或Logistic Sigmoid),直到那时:
hi′=σ(K1∑k=1K∑j∈NiαijkWkhj)
多头图注意层的聚集过程如图1(右)所示。