[發(fā)明專(zhuān)利]圖神經(jīng)網(wǎng)絡(luò)的訓(xùn)練方法和系統(tǒng)

在審

• 申請(qǐng)?zhí)枺?/td>: 202111131282.5
• 申請(qǐng)日: 2021-09-26
• 公開(kāi)（公告）號(hào): CN113837382A
• 公開(kāi)（公告）日: 2021-12-24
• 發(fā)明（設(shè)計(jì)）人: 段石石;劉長(zhǎng)偉;王軍正;譚鈞心;程純;陸超紅;汪磊;朱一飛;蘇杭;李寬;蔣文偉;趙榮生
• 申請(qǐng)（專(zhuān)利權(quán)）人: 杭州網(wǎng)易云音樂(lè)科技有限公司
• 主分類(lèi)號(hào): G06N3/08
• 分類(lèi)號(hào): G06N3/08;G06N3/04
• 代理公司: 北京同立鈞成知識(shí)產(chǎn)權(quán)代理有限公司 11205
• 代理人: 楊澤;劉芳

1.一種圖神經(jīng)網(wǎng)絡(luò)的訓(xùn)練方法，包括：

獲取原始圖數(shù)據(jù)結(jié)構(gòu)，所述原始圖數(shù)據(jù)結(jié)構(gòu)包括：多個(gè)原始結(jié)點(diǎn)和原始邊，所述原始邊用于表示相連的原始結(jié)點(diǎn)之間的關(guān)聯(lián)關(guān)系；

基于圖神經(jīng)網(wǎng)絡(luò)的第一數(shù)量，將所述原始圖數(shù)據(jù)結(jié)構(gòu)分為所述第一數(shù)量的第一圖數(shù)據(jù)結(jié)構(gòu)，所述第一圖數(shù)據(jù)結(jié)構(gòu)包括至少一個(gè)第一結(jié)點(diǎn)和至少一個(gè)第二結(jié)點(diǎn)，所述第二結(jié)點(diǎn)與至少一個(gè)所述第一結(jié)點(diǎn)通過(guò)所述原始邊連接，其中，不同所述第一圖數(shù)據(jù)結(jié)構(gòu)的第一結(jié)點(diǎn)不同，所述第一數(shù)量是預(yù)先配置的；

基于所述第一圖數(shù)據(jù)結(jié)構(gòu)，訓(xùn)練所述圖神經(jīng)網(wǎng)絡(luò)。

2.根據(jù)權(quán)利要求1所述的圖神經(jīng)網(wǎng)絡(luò)的訓(xùn)練方法，所述基于所述第一圖數(shù)據(jù)結(jié)構(gòu)，訓(xùn)練所述圖神經(jīng)網(wǎng)絡(luò)，包括：

基于所述第一圖數(shù)據(jù)結(jié)構(gòu)，確定多個(gè)目標(biāo)線數(shù)據(jù)結(jié)構(gòu)，所述目標(biāo)線數(shù)據(jù)結(jié)構(gòu)包括主結(jié)點(diǎn)和至少一個(gè)輔結(jié)點(diǎn)，其中所述主結(jié)點(diǎn)和其中一個(gè)輔結(jié)點(diǎn)通過(guò)所述原始邊連接，所述輔結(jié)點(diǎn)之間通過(guò)所述原始邊連接，所述主結(jié)點(diǎn)和所述輔結(jié)點(diǎn)的數(shù)量之和為預(yù)先配置的采樣長(zhǎng)度，所述采樣長(zhǎng)度為大于1的正整數(shù)，所述主結(jié)點(diǎn)為所述第一圖數(shù)據(jù)結(jié)構(gòu)中的第一結(jié)點(diǎn)，任一所述輔結(jié)點(diǎn)為所述第一圖數(shù)據(jù)結(jié)構(gòu)中第一結(jié)點(diǎn)或第二結(jié)點(diǎn)；

根據(jù)多個(gè)所述目標(biāo)線數(shù)據(jù)結(jié)構(gòu)，訓(xùn)練所述圖神經(jīng)網(wǎng)絡(luò)。

3.根據(jù)權(quán)利要求2所述的圖神經(jīng)網(wǎng)絡(luò)的訓(xùn)練方法，所述基于所述第一圖數(shù)據(jù)結(jié)構(gòu)，確定多個(gè)目標(biāo)線數(shù)據(jù)結(jié)構(gòu)，包括：

提取預(yù)設(shè)個(gè)數(shù)的所述第一圖數(shù)據(jù)結(jié)構(gòu)，組合成第二圖數(shù)據(jù)結(jié)構(gòu)；

確定所述第二圖數(shù)據(jù)結(jié)構(gòu)內(nèi)的任一所述第一結(jié)點(diǎn)為鍵結(jié)點(diǎn)，與所述鍵結(jié)點(diǎn)具有關(guān)聯(lián)關(guān)系的結(jié)點(diǎn)為值結(jié)點(diǎn)；

根據(jù)所述鍵結(jié)點(diǎn)和所述值結(jié)點(diǎn)，確定所述第二圖數(shù)據(jù)結(jié)構(gòu)對(duì)應(yīng)的多個(gè)結(jié)點(diǎn)序列；

基于所述結(jié)點(diǎn)序列，確定多個(gè)所述目標(biāo)線數(shù)據(jù)結(jié)構(gòu)。

4.根據(jù)權(quán)利要求3所述的圖神經(jīng)網(wǎng)絡(luò)的訓(xùn)練方法，所述基于所述結(jié)點(diǎn)序列，確定多個(gè)所述目標(biāo)線數(shù)據(jù)結(jié)構(gòu)，包括：

在第一圖數(shù)據(jù)結(jié)構(gòu)中獲取多個(gè)所述第一結(jié)點(diǎn)；

對(duì)于任一第一結(jié)點(diǎn)，在所述結(jié)點(diǎn)序列中查詢(xún)與所述第一結(jié)點(diǎn)相同的第一鍵結(jié)點(diǎn)；

確定與所述第一鍵結(jié)點(diǎn)屬于同一結(jié)點(diǎn)序列的一個(gè)值結(jié)點(diǎn)；

執(zhí)行結(jié)點(diǎn)序列查詢(xún)步驟：根據(jù)當(dāng)前采樣長(zhǎng)度，在所述結(jié)點(diǎn)序列中查詢(xún)當(dāng)前值結(jié)點(diǎn)作為新的鍵結(jié)點(diǎn)所在的結(jié)點(diǎn)序列，在所述結(jié)點(diǎn)序列中確定所述新的鍵結(jié)點(diǎn)所對(duì)應(yīng)的一個(gè)值結(jié)點(diǎn)作為輔結(jié)點(diǎn)或停止采樣，其中，所述新的鍵結(jié)點(diǎn)為除所述第一鍵結(jié)點(diǎn)之外的其他鍵結(jié)點(diǎn)；

在所述第一鍵結(jié)點(diǎn)和對(duì)應(yīng)的所有輔結(jié)點(diǎn)數(shù)量未達(dá)到采樣長(zhǎng)度的情況下，繼續(xù)執(zhí)行所述執(zhí)行結(jié)點(diǎn)序列查詢(xún)步驟；

在所述第一鍵結(jié)點(diǎn)和對(duì)應(yīng)的所有輔結(jié)點(diǎn)數(shù)量達(dá)到采樣長(zhǎng)度的情況下，則將所述第一鍵結(jié)點(diǎn)作為所述目標(biāo)線數(shù)據(jù)結(jié)構(gòu)的主結(jié)點(diǎn)，根據(jù)所述所有輔結(jié)點(diǎn)之間的采樣順序，構(gòu)成所述目標(biāo)線數(shù)據(jù)結(jié)構(gòu)。

5.根據(jù)權(quán)利要求3或4所述的圖神經(jīng)網(wǎng)絡(luò)的訓(xùn)練方法，還包括：

接收第一配置參數(shù)，所述第一配置參數(shù)包括：第二數(shù)量的查詢(xún)服務(wù)器；

調(diào)度所述第二數(shù)量的查詢(xún)服務(wù)器，并將屬于同一所述第二圖數(shù)據(jù)結(jié)構(gòu)的結(jié)點(diǎn)序列對(duì)應(yīng)存儲(chǔ)在各所述查詢(xún)服務(wù)器中。

6.根據(jù)權(quán)利要求1至4任一項(xiàng)所述的圖神經(jīng)網(wǎng)絡(luò)的訓(xùn)練方法，所述基于圖神經(jīng)網(wǎng)絡(luò)的第一數(shù)量，將所述原始圖數(shù)據(jù)結(jié)構(gòu)分為所述第一數(shù)量的第一圖數(shù)據(jù)結(jié)構(gòu)，包括：

基于所述圖神經(jīng)網(wǎng)絡(luò)的第一數(shù)量和所述原始圖數(shù)據(jù)結(jié)構(gòu)中所述原始結(jié)點(diǎn)的序號(hào)，確定所述原始結(jié)點(diǎn)對(duì)應(yīng)的標(biāo)識(shí)信息；

將所述標(biāo)識(shí)信息相同的原始結(jié)點(diǎn)作為第一結(jié)點(diǎn)歸類(lèi)；

遍歷與所述第一結(jié)點(diǎn)直接相連的第二結(jié)點(diǎn)；

根據(jù)所述第一結(jié)點(diǎn)和所述第二結(jié)點(diǎn)構(gòu)建所述第一圖數(shù)據(jù)結(jié)構(gòu)。

7.根據(jù)權(quán)利要求3或4所述的圖神經(jīng)網(wǎng)絡(luò)的訓(xùn)練方法，所述提取預(yù)設(shè)個(gè)數(shù)的所述第一圖數(shù)據(jù)結(jié)構(gòu)，組合成第二圖數(shù)據(jù)結(jié)構(gòu)，包括：

獲取所述第一圖數(shù)據(jù)結(jié)構(gòu)的標(biāo)識(shí)信息，所述標(biāo)識(shí)信息為大于或等于1的正整數(shù)，各所述第一數(shù)據(jù)結(jié)構(gòu)的標(biāo)識(shí)信息按照從小到大的順序排序；

基于所述標(biāo)識(shí)信息與預(yù)設(shè)所述第二圖數(shù)據(jù)結(jié)構(gòu)的數(shù)量，確定對(duì)應(yīng)的哈希值；

將所述哈希值相同的所述第一圖數(shù)據(jù)結(jié)構(gòu)進(jìn)行提取，并組合成第二圖數(shù)據(jù)結(jié)構(gòu)。