个人博客网站链接
1. 总览
要解决什么问题?(what?)
链接预测和实体分类
为什么要解决这个问题?(why?)
因为虽然知识图谱用途很多,而现有的知识图谱都存在不完整的问题。
用什么方法解决?(how?)
- 用图卷积网络和因式分解相结合来解决链接预测问题;
- 用图卷积网络单独解决实体分类问题。
文章有什么创新?
- 首次把GCN引入关系数据建模;
- 提出了一种参数共享和增强稀疏限制的方法——权重矩阵的基本分解和块分解
- 用GCN与因式分解组成auto-encoder的方法,可以提高因式分解模型在链接预测上的效果。
效果如何?
在FB15k-237上高出baseline 29.8%
还存在什么问题?
2 模型R-GCN
2.1 Relational graph convolutional networks
作者基于消息传递框架:
hi(l+1)=σ(m∈Mi∑gm(hi(l),hj(l)))
- 其中hi(l)∈Rd(l)是节点vi在第l层神经网络的隐藏状态;d(l)是当前层表示的维度;gm是传入消息的累积形式。
设计了一个在关系多图中的传播策略:
hi(l+1)=σ⎝⎛r∈R∑j∈Nir∑ci,r1Wr(l)hj(l)+W0(l)hi(l)⎠⎞
- 其中Nir代表在关系r∈R下节点i的邻居索引集合。其中ci,r是一个特定于问题的归一化常数,可以预先学习或选择ci,r=∣Nir∣。W0(l)hi(l)是作者添加的对于每个节点的一个特定关系的自连接。
- 此处采取简单的线性消息转换,其实可以选择更灵活的函数,如多层神经网络(以牺牲计算效率为代价)。
2.2 Regularization | 正则化
**核心问题:**在处理多元关系数据时,图中参数数量和关系数量快速增长可能会导致对稀有关系的过拟合和模型规模过大。
为了解决这个问题作者提出了两种调整R-GCN层的权重的方式:
basis-decomposition:
Wr(l)=b=1∑Barb(l)Vb(l)
- 其中Vb(l)∈Rd(l+1)×d(l)是一个作为基础变换的线性组合,arb(l)是一个只依赖于r的系数。
- 基函数分解可以看作是不同关系类型之间有效权重共享的一种形式
block-diagonal-decomposition:
Wr(l)=b=1⨁BQbr(l)
- 其中Qb,r(l)∈RBd(l+1)×Bd(l),Wr(l)是对角块矩阵:diag(Q1r(l),…,QBr(l))。
- 而块分解可以看做是每种关系类型对权重矩阵的系数约束。
- 块分解结构编码一种直觉,即可以将潜在特征分组为变量集,这些变量集在组内比组间更加紧密地联系。
同时,作者期望基本参数化可以减轻稀疏关系的过度拟合,因为稀疏关系和更频繁关系之间共享参数更新。
对所有的R-GCN模型都采取以下形式:
- 按照作者提出的传播模型进行堆叠L层
- 如果不存在其他特征,则可以将第一层的输入选择为图中每个节点的唯一one-hot向量;
- 对于块表示,作者将one-hot向量通过一个单一的线性转换为一个dense表示;
- 作者只考虑用一个featureless的方法,不同于GCN模型。
3 实体分类
作者通过堆叠R-GCN的传播函数,最后一层通过softmax函数输入,在所有节点上采用交叉熵损失(忽略无标签节点):
L=−i∈Y∑k=1∑Ktiklnhik(L)
- 其中 Y代表有标签的节点索引集,hik(L)代表第i个标记节点的网络输出的第k个条目,tik表示真实标签。
4 链接预测
作者引入一个auto-encoder模型:
- encoder:将每个实体vi∈V 映射到一个是指向量 ei∈Rd ;
- decoder:根据顶点表示重建图的边:通过一个scorce函数 s:Rd×R×Rd→R 来将三元组(subject, relation, object)映射为一个实数分数。
在本文中,作者采用的是DistMult函数作为解码器,每一个关系r被映射为一个对角矩阵 Rr∈Rd×d,一个三元组(s,r,o)的分数为:
f(s,r,o)=esTRreo
最后作者得出自己的损失函数:
L=−(1+w)∣E^∣1(s,r,o,y)∈T∑ylogl(f(s,r,o))+(1−y)log(1−l(f(s,r,o)))
- 其中 T 是真实三元组和破坏得到的负样例三元组的总样本。
5 实验
5.1 实体分类实验
数据集:AIFB、MUTAG、BGS、AM
baseline:Feat、WL、RDF2Vec
评价准则:准确率
结果:
5.2 链接预测
关系抽取实验:
数据集:WordNet(WN18),Freebase(FB15K)
baseline:LinkFeat,DistMult,CP,TransE,HolE,ComplEx
评价准则:MRR(mean reciprocal rank)(Raw,Filtered),Hits @(1,3,10)
实验结果:
- 其中 f(s,r,t)R−GCN+=αf(s,r,t)R−GCN+(1−α)f(s,r,t)DistMult ,R-GCN 和 DistMult都是各自训练好的。
参考链接