[发明专利]结合节点和边进行网络图嵌入的方法和装置

权利要求书

1.一种获取关系网络图中的节点的点嵌入向量的方法，所述点嵌入向量为预定嵌入空间中的向量，所述关系网络图中的多个节点与电商平台中的多个用户分别对应，所述关系网络图中的边表示其连接的两个节点分别对应的用户共同购买有至少一个商品，所述方法包括：

对于所述关系网络图中的第一节点，获取所述第一节点、所述第一节点的各个相邻节点、及所述第一节点的预定数目的不相邻节点各自的当前点嵌入向量，并获取所述第一节点的相连边、及所述第一节点的预定数目的不相连边各自的当前边嵌入向量，其中，所述当前点嵌入向量和所述当前边嵌入向量都为所述嵌入空间中的向量；

获取第一目标函数，其中，所述第一目标函数中的第一项为至少一个第一减函数之和，所述至少一个第一减函数分别为关于第一节点与各个所述相邻节点之间的相似性的减函数，所述第一目标函数中的第二项为至少一个第一增函数之和，所述至少一个第一增函数分别为关于第一节点与各个所述不相邻节点之间的相似性的增函数，其中，所述关系网络图中的任一对节点之间的相似性与该对节点各自的点嵌入向量相关；

获取第二目标函数，其中，所述第二目标函数的第一项为至少一个第二减函数之和，其中，所述至少一个第二减函数分别为关于所述第一节点与各个所述相连边之间的相似性的减函数，所述第二目标函数的第二项为至少一个第二增函数之和，所述至少一个第二增函数分别为关于所述第一节点与各个所述不相连边之间的相似性的增函数，其中，所述关系网络图中的节点与边之间的相似性与该节点的点嵌入向量和该边的边嵌入向量相关；以及

基于第一目标函数与第二目标函数的加权和、各个所述当前点嵌入向量、以及各个所述当前边嵌入向量，通过随机梯度下降法更新所述第一节点的点嵌入向量，以使得所述第一目标函数与第二目标函数的加权和的值减小。

2.根据权利要求1所述的方法，其中，所述关系网络图为有向图，其中，所述第一目标函数还包括第三项和第四项，其中，所述第三项为至少一个附加减函数之和，所述至少一个附加减函数分别为关于各个以第一节点为相邻节点的节点与第一节点之间的相似性的减函数，所述第四项为至少一个附加增函数之和，所述至少一个附加增函数分别为关于预定数目的不以第一节点为相邻节点的节点与第一节点之间的相似性的增函数。

3.根据权利要求1所述的方法，其中，所述第一节点还具有上下文嵌入向量，其中，所述上下文嵌入向量为所述嵌入空间中的向量，

其中，所述第一目标函数还包括第三项和第四项，其中，所述第三项为至少一个附加减函数之和，所述至少一个附加减函数分别为关于各个以第一节点为相邻节点的节点与第一节点之间的相似性的减函数，所述第四项为至少一个附加增函数之和，所述至少一个附加增函数分别为关于预定数目的不以第一节点为相邻节点的节点与第一节点之间的相似性的增函数，其中，节点i与节点j之间的相似性与该对节点各自的点嵌入向量、及节点j的上下文嵌入向量相关，其中节点i和节点j分别为所述关系网络图中的任一节点。

4.根据权利要求1所述的方法，其中，所述一对节点之间的相似性为该对节点各自的点嵌入向量的点积的增函数。

5.根据权利要求3所述的方法，其中节点i与节点j之间的相似性为关于以下两部分的加权和的增函数：该对节点各自的点嵌入向量的点积，及节点i的点嵌入向量与节点j的上下文嵌入向量的点积。

6.根据权利要求1所述的方法，其中，所述第一目标函数的第一项在取负后为第一节点与各个所述相邻节点之间的相似性的对数的加权和，其中，用于获取所述加权和的权重为第一节点与相应相邻节点之间的边值。

7.根据权利要求1所述的方法，所述第一节点的预定数目的不相邻节点从所述第一节点的全部不相邻节点中随机获取，所述第一节点的预定数目的不相连边从所述第一节点的全部不相连边中随机获取。

8.根据权利要求1所述的方法，其中，所述方法循环进行多次，其中，在第一次循环进行之前，各个所述当前点嵌入向量和各个所述当前边嵌入向量随机获取。