[发明专利]一种3DRTE时间知识图谱的三维旋转嵌入方法

权利要求书

1.一种3DRTE时间知识图谱的三维旋转嵌入方法，其特征在于：包括以下步骤：

步骤S1：定义时间知识图谱TKG，给定实体集合E，关系集合R，时间戳集合T，一个TKG表示为时间事实四元组(s,r,o,t)的集合其中s,o∈E表示实体，r∈R表示关系；令已知的时间事实集合为G，且则TKGC的目的就是在现有知识G的基础上推导出W，即在给定已知事实集合G的情况下补全出TKG的全貌W；

步骤S2：建立3DRTE模型，包括利用自注意力机制处理时间戳的各个片段带来的信息和利用四元数的旋转特性，将实体在融合了时间的关系的作用下，旋转建模在了三维空间中，从而获得实体在关系旋转下的得分函数，即完成模型建立；

步骤S3：将步骤S2中得到的模型使用损失函数进行训练，并进行数据验证，最终实现补全时间知识图的目的。

2.根据权利要求1所述的一种3DRTE时间知识图谱的三维旋转嵌入方法，其特征在于：所述步骤S2中所述利用3DRTE模型中的自注意力机制处理时间戳的各个片段带来的信息的具体内容为：

给定一个时间知识四元组(s,r,o,t)，仅考虑关系r和时间t，因此令它们的维度都是k；对于关系r，它来自关系的嵌入矩阵其中N表示整个知识图谱中关系类型的个数；对于时间t，它是一个由l个token组成的时间戳，对每个token，都从矩阵获得它们的表示，其中L表示知识图谱中时间token类型的个数；为了让关系自适应于每一个时间token带来的信息，引入自注意力机制处理关系-时间序列；自注意力通过计算每一个token对关系的影响程度，从而有选择性的融合特征；具体计算过程如下：

首先，将关系和时间序列拼接作为整个计算图的输入I，即

I＝[r；t₁；t₂......；t_l] (1)

把I分别输入进三个参数矩阵W^U、W^K、W^V，经过线性变换得到查询矩阵U、键矩阵K以及值矩阵V，即

之后，为了计算每个查询矩阵U对每个键矩阵K的权重，先将K转置得到K^T，与U作矩阵乘法之后，经过softmax的激活，得到每个时间token对关系r的权重矩阵最后，将V乘以即得到最后的输出矩阵O，公式化为：

O＝V×softmax(K^T×U) (3)

只取O的第一行作为输出，即融合了时序信息的关系表示r_t。