知识图谱——TransE模型原理

知识图谱——TransE模型原理

1 TransE模型介绍

1.1 TransE模型引入

在我们之前的文章中，提到了知识图谱补全任务的前提任务是知识表示学习，在知识表示学习中，最为经典的模型就是TransE模型，TransE模型的核心作用就是将知识图谱中的三元组翻译成embedding向量。

1.2 TransE模型思想

为了后面便于表示，我们先设定一些符号

1. h 表示知识图谱中的头实体的向量。
2. t 表示知识图谱中的尾实体的向量。
3. r 表示知识图谱中的关系的向量。

在TransE模型中，有这样一个假设：
$$t=h+r$$
也就是说，正常情况下的尾实体向量=头实体向量+关系向量。用图的方式描述如下：

如果一个三元组不满足上述的的关系，我们就可以认为这是一个错误的三元组。

1.3 TransE模型的目标函数

首先，我们先来介绍两个数学概念：

L1范数

也称为曼哈顿距离，对于一个向量X而言，其L1范数的计算公式为：
$$||X|{|}_{L1}=\sum _{i=1}^n|x_i|$$
其中，$x_i$表示向量X的第i个属性值，这里我们取的是绝对值。并且，使用L1范数可以衡量两个向量之间的差异性，也就是两个向量的距离。
$$Distanc{e}_{L1}(X_1,X_2)=\sum _{i=1}^n|X_{1i}-X_{2i}|$$

L2范数

也称为欧式距离，对于一个向量X而言，其L2范数的计算公式为：
$$||X|{|}_{L2}=\sqrt{\sum _{i=1}^n{X}_i^2}$$
同样，L2范数也可以用来衡量两个向量之间的差距:
$$Distanc{e}_{L2}(X_1,X_2)=\sum _{i=1}^n(X_{1i}-X_{2i}{)}^2$$

根据我们上面介绍的Trans中的假设，我们可以知道，对于一个三元组而言，头实体向量和关系向量之和与尾实体向量越接近，那么说明该三元组越接近一个正确的三元组，差距越大，那么说明这个三元组越不正常。那么我们可以选择L1或者L2范数来衡量三个向量的差距。而我们的目标就是使得争取的三元组的距离越小越好，错误的三元组距离越大越好，也就是其相反数越小越好。数学化的表示就是：
$$min\sum _{(h,r,t)\in G}\sum _{(h^{\prime },r^{\prime },t^{\prime })\in G^{\prime }}[\gamma +distance(h+r,t)-distance(h^{\prime }+r^{\prime },t^{\prime }){]}_{+}$$

其中：

$(h,r,t)$：表示正确的三元组
$(h^{\prime },r^{\prime },t^{\prime })$： 表示错误的三三元组
$\gamma$ ：表示正样本和负样本之间的间距，一个常数
$[x{]}_{+}$：表示max(0,x)

我们来简单的解释以下目标函数，我们的目标是让正例的距离最小，也就是$min(distance(h+r,t))$，让负例的相反数最小也就是$(min(-distance(h^{\prime }+r^{\prime },t^{\prime })))$，对于每一个正样本和负样本求和，再增加一个常数的间距，就是整体距离的最小值。也就是我们的目标函数。

1.4 目标函数的数学推导

这里，我们采用欧氏距离作为distance函数，则目标函数可以改写为：
$$min\sum _{(h,r,t)\in G}\sum _{(h^{\prime },r^{\prime },t^{\prime })\in G^{\prime }}[\gamma +(h+r-t{)}^2-(h^{\prime }+r^{\prime }-t^{\prime }{)}^2{]}_{+}$$
则对于损失函数loss就有：
$$Loss=\sum _{(h,r,t)\in G}\sum _{(h^{\prime },r^{\prime },t^{\prime })\in G^{\prime }}[\gamma +(h+r-t{)}^2-(h^{\prime }+r-t^{\prime }{)}^2{]}_{+}$$
在损失函数中，我们知道所有的参数包括{$h,r,t,h^{\prime },r,t^{\prime }$}。下面，我们来逐个进行梯度推导：

1. 首先是对h的梯度，对于某一个$h_i$而言
   $$\frac{\partial Loss}{\partial h_i}=\sum _{(h_{,}r,t)\in G}\sum _{(h^{\prime },r,t^{\prime })\in G^{\prime }}\frac{\partial [\gamma +(h+r-t{)}^2-(h^{\prime }+r-t^{\prime }{)}^2{]}_{+}}{\partial h_i}$$
   在整个求和的过程中，只针对包含$h_i$的项求导：
   $$\frac{\partial [\gamma +(h_i+r-t{)}^2-(h^{\prime }+r-t^{\prime }{)}^2{]}_{+}}{\partial h_i}$$
   有：
   $$\frac{\partial [\gamma +(h_i+r-t{)}^2-(h^{\prime }+r-t^{\prime }{)}^2{]}_{+}}{\partial h_i}=\{\begin{array}{llc}2(h_i+r-t)& & \gamma +(h+r-t{)}^2-(h^{\prime }+r-t^{\prime }{)}^2>0\\ 0& & \gamma +(h+r-t{)}^2-(h^{\prime }+r-t^{\prime }{)}^2<=0\end{array}$$
   则原式变为：
   $$\frac{\partial Loss}{\partial h_i}=\sum _{(h_i,r,t)\in G}\sum _{(h^{\prime },r,t^{\prime })\in G^{\prime }}\{\begin{array}{llc}2(h_i+r-t)& & \gamma +(h_i+r-t{)}^2-(h^{\prime }+r-t^{\prime }{)}^2>0\\ 0& & \gamma +(h_i+r-t{)}^2-(h^{\prime }+r-t^{\prime }{)}^2<=0\end{array}$$
   同理对于$t_i,h_i^{\prime },t_i^{\prime }$有：
   $$\frac{\partial Loss}{\partial t_i}=\sum _{(h,r,t_i)\in G}\sum _{(h^{\prime },r,t^{\prime })\in G^{\prime }}\{\begin{array}{llc}-2(h+r-t_i)& & \gamma +(h+r-t_i{)}^2-(h^{\prime }+r-t^{\prime }{)}^2>0\\ 0& & \gamma +(h+r-t_i{)}^2-(h^{\prime }+r-t^{\prime }{)}^2<=0\end{array}$$
   $$\frac{\partial Loss}{\partial h_i^{\prime }}=\sum _{(h,r,t)\in G}\sum _{(h_i^{\prime },r,t^{\prime })\in G^{\prime }}\{\begin{array}{llc}-2(h^{\prime }+r-t^{\prime })& & \gamma +(h+r-t{)}^2-(h_i^{\prime }+r-t^{\prime }{)}^2>0\\ 0& & \gamma +(h+r-t{)}^2-(h_i^{\prime }+r-t^{\prime }{)}^2<=0\end{array}$$
   $$\frac{\partial Loss}{\partial t_i^{\prime }}=\sum _{(h,r,t)\in G}\sum _{(h^{\prime },r,t_i^{\prime })\in G^{\prime }}\{\begin{array}{llc}2(h^{\prime }+r-t^{\prime })& & \gamma +(h+r-t{)}^2-(h^{\prime }+r-t_i^{\prime }{)}^2>0\\ 0& & \gamma +(h+r-t{)}^2-(h^{\prime }+r-t_i^{\prime }{)}^2<=0\end{array}$$
   最后对于$r_i$，γ有：
   $$\frac{\partial Loss}{\partial r_i}=\sum _{(h,r_i,t)\in G}\sum _{(h^{\prime },r_i,t^{\prime })\in G^{\prime }}\{\begin{array}{llc}2(h+r_i-t)-2(h^{\prime }+r_i-t^{\prime })& & \gamma +(h+r_i-t{)}^2-(h^{\prime }+r_i-t^{\prime }{)}^2>0\\ 0& & \gamma +(h+r_i-t{)}^2-(h^{\prime }+r_i-t^{\prime }{)}^2<=0\end{array}$$
   $$\frac{\partial Loss}{\partial \gamma }=\sum _{(h,r,t)\in G}\sum _{(h^{\prime },r^{\prime },t^{\prime })\in G^{\prime }}\{\begin{array}{llc}1& & \gamma +(h+r-t{)}^2-(h^{\prime }+r-t^{\prime }{)}^2>0\\ 0& & \gamma +(h+r-t{)}^2-(h^{\prime }+r-t^{\prime }{)}^2<=0\end{array}$$

1.5 如何产生负样本

在我们之前算法描述中，我们提到了负样本的问题，对于一个知识图谱而言，其中保存的全部都是正样本时肯定的了。那么，我们应该如何获取负样本呢？

具体的可以通过随机替换头实体的方式来实现一个错误的三元组，或者采用随机替换一个错误的尾实体的方式来形成一个错误的三元组。

同时，为了避免，我们在替换形成的三元组也存在于知识图谱中，我们需要在替换之后进行过滤。

2 参考文章

1. TransE如何进行向量更新？
2. TransE算法详解