本文共 3927 字，大约阅读时间需要 13 分钟。

论文地址：

这篇论文被IJCAI 2020接收，主要工作是首次将图神经网络运用在由图构成的图（Graph of Graphs）上。

Graph of Graphs(GoG)

在分化学物质间相互作用的预测任务上，每一个化学物质可以看做是一个图结构，利用图神经网络的方法能够得到化学物质的特征表示，从而预测化学物质间是否存在相互作用。这样的方法只考虑了化学物质内部的图结构信息，而忽视了大量的化学物质及其之间的相互作用本质上也是一个图结构。

如上图所示，图中的每一个节点也是一个图，所以这是一个由图构成的图（Graph of Graphs）。

Preliminaries

问题定义

本文在两个数据及上做了实验，分别是化学物质相互作用数据集(CCI)和药物相互作用数据集(DDI)。CCI数据集中的相互作用只有一种，DDI中的相互作用有多种，所以在CCI中可通过$\left(G_i,G_j\right)$来计算(i,j)之间存在相互作用的得分$p_{ij}$，在DDI中通过$(G_i,r,G_j)$三元组计算(i,j)之间存在关系r的得分$p_{ij}^r$。

Molecule Graph

在molecule graph $G_M$中，节点$a_i$代表原子i，边$e_{ij}$代表i和j之间的连接。

Interaction Graph

在interaction graph $G_I=(N,E_I)$中，$N$是节点的个数，即$G_M$的数量，$E_I$是$G_M$之间的相互作用。

在DDI数据集上，每条边还有一个属性向量$e^r$。

GoGNN

基本框架是首先在$G_M$上得到一个表示$x_{hidden}$，然后再作为$G_I$的输入。

molecule graph neural network

在$G_M$上直接套用GCN：

$$\begin{array}{rl}{M}_{(l+1)}& =GC{N}_l\left(A,M_l\right)\\ GC{N}_l\left(A,M_l\right)& =\sigma \left({\tilde{D}}^{-\frac{1}{2}}\tilde{A}{\tilde{D}}^{-\frac{1}{2}}{M}_l{W}_l\right)\end{array}$$ 为了输出图的表示，在每一层使用self-attention机制，得到$s_l\in {\mathbb{R}}^{n\times 1}$ $$s_l=\sigma \left({\tilde{D}}^{-\frac{1}{2}}\tilde{A}{\tilde{D}}^{-\frac{1}{2}}{M}_l{W}_{att}^l\right)$$

where $W_{att}^l\in {\mathbb{R}}^{d\times 1}$ is the attention weight matrix for the pooling layer to obtain the self-attention score.

挑选出值最高的$\lceil \gamma n\rceil$个维度作为第l层输出的$G_M$的表示：

$$\begin{array}{c}idx=\mathrm{top}(s,\lceil \gamma n\rceil ),s_{\mathrm{mas}k}=s_{idx}\\ M_{sel}=M\odot s_{mask}\end{array}$$最后将$[s_1,s_2,..s_L]$拼接起来即可得到整个$G_M$的表示$x_G$。

interaction graph neural network

Graph Attention Network

在$G_I$上直接套用GAT：$${\mathbf{x}}_{G_i}^{l+1}={\Vert }_{\kappa =1}^K\sigma \left(\sum _{j\in {\eta }_{G_i}}{\alpha }_{ij}^{\kappa }{\mathbf{W}}_{\kappa }^l{\mathbf{x}}_{G_j}^l\right)$$$${\alpha }_{ij}=\frac{\exp \left(\mathrm{LeakeyRelu}\left(\mathbf{a}\left[\mathbf{W}{\mathbf{x}}_{G_i}\right]\Vert \left[\mathbf{W}{\mathbf{x}}_{G_j}\right]\right)\right)}{{\sum }_{n\in {\eta }_{G_i}}\exp \left(\mathrm{Leakey}\mathrm{Relu}\left(\mathbf{a}\left[\mathbf{W}{\mathbf{x}}_{G_i}\right]\Vert \left[\mathbf{W}{\mathbf{x}}_{G_n}\right]\right)\right)}$$

在计算${\alpha }_{ij}$时论文里的公式没有给出代表是第几层的角标，可能是作者笔误，一般来说每一层${\alpha }_{ij}^l$应该都是由该层的${\mathbf{x}}_{G_j}^l$和${\mathbf{x}}_{G_i}^l$得到。

Edge Aggregation Network

由于DDI数据集上还有边的信息$e_{ij}^r$，所以在聚合时需要将边上的信息考虑进去：

$h_{{\mathbf{W}}_e}$是一个线性层，所以$h_{{\mathbf{W}}_e}(e_{ij}^r)$得到的是一个标量。这里作者可能又笔误了，在聚合了邻居节点的信息后没有体现出做线性变换的过程。

Prediction

在CCI数据集上，直接使用$p_{ij}=\sigma ({\mathbf{x}}_{G_i}^T\cdot {\mathbf{x}}_{G_j})$得到(i, j)之间的score。在DDI数据集上，需要将特定的关系r考虑进去，$p_{ij}^r=\sigma \left({\left({\mathbf{W}}_r{\mathbf{x}}_{G_i}\right)}^T\cdot \left({\mathbf{W}}_r{\mathbf{x}}_{G_j}\right)\right)$。

转载地址：http://ttdqi.baihongyu.com/