[發明專利]分布式圖嵌入方法、裝置、設備及系統

有效

• 申請號: 201810975295.2
• 申請日: 2018-08-24
• 公開（公告）號: CN109344295B
• 公開（公告）日: 2020-05-05
• 發明（設計）人: 石磊磊
• 申請（專利權）人: 阿里巴巴集團控股有限公司
• 主分類號: G06F16/901
• 分類號: G06F16/901
• 代理公司: 北京博思佳知識產權代理有限公司 11415
• 代理人: 林祥

1.一種分布式圖嵌入方法，圖數據結構至少分布于兩個計算節點，所述方法包括：

在利用預設采樣算法對圖數據結構進行路徑采樣時，將當前采樣頂點的標識和當前嵌入向量加入預定義的消息數據結構，所述消息數據結構中記錄有采樣過程中采樣頂點的標識和當前嵌入向量、且所述消息數據結構由上一采樣頂點所在計算節點傳遞而獲得；

在消息數據結構的數據長度達到詞向量算法所定義的窗口長度時，利用所述詞向量算法計算更新后的消息數據結構中各采樣頂點的嵌入向量的更新梯度；

基于更新后的消息數據結構中的標識，將各更新梯度分別發送至相應頂點，并利用所述更新梯度對歸屬當前計算節點的采樣頂點的當前嵌入向量進行更新；

達到預設收斂條件后，將圖數據結構中各頂點的最終嵌入向量，構成圖嵌入矩陣。

2.根據權利要求1所述的方法，圖數據結構中頂點的當前嵌入向量存儲于頂點所在計算節點的圖頂點數據中。

3.根據權利要求1所述的方法，所述方法還包括：

在消息數據結構的數據長度未達到詞向量算法所定義的窗口長度時，利用預設采樣算法對與所述當前采樣頂點相鄰且連接的頂點進行采樣，將更新后的消息數據結構發送至所確定的采樣頂點所在計算節點。

4.根據權利要求1所述的方法，所述消息數據結構的數據長度上限值為窗口長度，所述方法還包括：

在消息數據結構的數據長度達到或大于詞向量算法所定義的窗口長度時，刪除消息數據結構中加入時間最早的采樣頂點的標識和當前嵌入向量。

5.根據權利要求1至4任一項所述的方法，所述方法還包括：

基于預設任務利用圖嵌入矩陣構建訓練樣本，并利用所構建的訓練樣本對預設模型進行訓練，獲得用于處理預設任務的模型。

6.一種分布式圖嵌入裝置，圖數據結構至少分布于兩個計算節點，所述裝置設于其中一個計算節點，包括：

消息更新模塊，用于在利用預設采樣算法對圖數據結構進行路徑采樣時，將當前采樣頂點的標識和當前嵌入向量加入預定義的消息數據結構，所述消息數據結構中記錄有采樣過程中采樣頂點的標識和當前嵌入向量、且所述消息數據結構由上一采樣頂點所在計算節點傳遞而獲得；

梯度計算模塊，用于在消息數據結構的數據長度達到詞向量算法所定義的窗口長度時，利用所述詞向量算法計算更新后的消息數據結構中各采樣頂點的嵌入向量的更新梯度；

向量更新模塊，用于基于更新后的消息數據結構中的標識，將各更新梯度分別發送至相應頂點，并利用所述更新梯度對歸屬當前計算節點的采樣頂點的當前嵌入向量進行更新；

矩陣生成模塊，用于達到預設收斂條件后，將圖數據結構中各頂點的最終嵌入向量，構成圖嵌入矩陣。

7.根據權利要求6所述的裝置，圖數據結構中頂點的當前嵌入向量存儲于頂點所在計算節點的圖頂點數據中。

8.根據權利要求6所述的裝置，所述裝置還包括：

頂點采樣模塊，用于在消息數據結構的數據長度未達到詞向量算法所定義的窗口長度時，利用預設采樣算法對與所述當前采樣頂點相鄰且連接的頂點進行采樣；

信息傳輸模塊，用于將更新后的消息數據結構發送至所確定的采樣頂點所在計算節點。

9.根據權利要求6所述的裝置，所述消息數據結構的數據長度上限值為窗口長度，所述消息更新模塊還用于：

在消息數據結構的數據長度達到或大于詞向量算法所定義的窗口長度時，刪除消息數據結構中加入時間最早的采樣頂點的標識和當前嵌入向量。