ICLR 2020 开源论文 | 多关系图神经网络CompGCN

©PaperWeekly 原创 · 作者｜纪厚业

学校｜北京邮电大学博士生

研究方向｜图神经网络、推荐系统

图神经网络已经成为图数据分析领域的经典算法了，各大顶会上相关的论文也层出不穷。但是，这些工作主要关注点都在如何在简单无向图上设计一个更新颖更复杂的模型，而没有从图数据的复杂程度来考虑。

实际生活中的图往往更加复杂。比如，知识图谱上的有向多关系图。下面是一个知识图谱的例子，这里有多种节点，如 London 和 United Kingdom；也有多种有向关系，如 Born-in 和 Citizen-of。注意，这里的关系是有方向的，Born-in 和 Bord-in_inv 是同一关系的两个方向。

可以看出，多关系图的建模是更符合实际也具有更高的应用价值。本文作者从实际的图数据特点出发，设计了一种针对多关系有向图的图神经网络 CompGCN 来同时学习节点和关系的表示。同时，为了降低大量不同类型关系带来的参数，这里作者设计了一种分解操作 composition operation，将所有关系都用一组基的加权组合来表示。这样用于描述关系的参数只和基的个数有关。 

论文链接：https://arxiv.org/abs/1911.03082

源码链接：https://github.com/malllabiisc/CompGCN

下图对比了几种图神经网络的特点，可以看出：学习关系的表示是 CompGCN 独有的特点。 

总的来说，本文的贡献有 3 点： 

• 设计了 CompGCN 模型，一种可以考虑多种关系信息的图神经网络框架，来同时学习节点和关系的表示。

• 证明了 CompGCN 与之前的多关系 GNN 之间的联系。

• 实验验证了 CompGCN 的有效性。

相关工作回顾

首先，作者回顾了多关系图及多关系图神经网络的做法。

多关系图中的边可以表示为 (u,v,r)，代表存在一条从节点 u 指向节点 v 的类型为 r 的边，其中 r∈R。同时，也存在一个相应的反向边 。 

多关系图神经网络，也是分别聚合特定关系 r 下的邻居：

其中， 是针对关系 r 的参数矩阵。而经典的单关系 GCN 聚合公式如下：

可以看出，两者主要差异就在