[发明专利]一种分布式知识图谱嵌入方法

摘要

本发明公开了一种分布式知识图谱嵌入方法。工业界常用的普通服务器计算能力和内存容量有限，在处理大规模知识图谱数据时，单台服务器往往不能完成处理工作。为了解决这个问题，本发明提出了一种分布式知识图谱嵌入方法，该方法采用一种分布式架构，将多台服务器统一在一个计算系统中，通过网络进行通信，共享CPU和内存等计算资源，从而完成大规模知识图谱嵌入的计算。该方法中，参数按照维度划分到各台服务器上，从而减少计算中产生的通信开销，加快运行速度。

主权项

1.一种分布式知识图谱嵌入方法，其特征在于，包括Worker执行的步骤和Server执行的步骤：所述方法中Worker执行的具体步骤为：步骤100，初始化算法参数，包括嵌入向量的维度d，总共执行的轮数T，每轮采样的三元组数B，学习率μ，正则项参数λ，间隔γ，Worker数量Nw，Server的数量Ns；步骤101，读取知识图谱的统计信息，包括实体的数量ne，关系的数量nr；步骤102，读取当前Worker所负责训练的三元组数据，整个知识图谱中的三元组集合被等量划分为Nw份，每个Worker读取一份；步骤103，令计数器i＝0；步骤104，从当前Worker负责的三元组中采样B/Nw个，作为轮训练的正例三元组集合；步骤105，对于采样出的每一个正例三元组随机替换其头部或尾部，作为负例三元组，与原正例三元组组成一个三元组对；步骤106，发送各个三元组对到各个Server；步骤107，从各个Server处接受各个三元组对的损失；步骤108，统计各个三元组对的总损失，对于总损失大于间隔γ的三元组对设置更新标志为1，对于总损失小于间隔γ的三元组对设置更新标志为0；步骤109，将更新标志发送给各个Server；步骤110，令计数器i＝i+1；步骤111，判断计数器i的值是否小于执行轮数T；如果是，回到步骤104，进行新的一轮计算；如果不是，已满足停止条件，进入结束阶段；步骤112，结束计算。