[發(fā)明專利]預(yù)訓(xùn)練圖神經(jīng)網(wǎng)絡(luò)的方法以及裝置

在審

• 申請(qǐng)?zhí)枺?/td>: 202010549634.8
• 申請(qǐng)日: 2020-06-16
• 公開（公告）號(hào): CN111860769A
• 公開（公告）日: 2020-10-30
• 發(fā)明（設(shè)計(jì)）人: 黃正杰;李偉彬;馮仕堃
• 申請(qǐng)（專利權(quán)）人: 北京百度網(wǎng)訊科技有限公司
• 主分類號(hào): G06N3/04
• 分類號(hào): G06N3/04;G06N3/08
• 代理公司: 北京鴻德海業(yè)知識(shí)產(chǎn)權(quán)代理事務(wù)所(普通合伙) 11412
• 代理人: 田宏賓

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

獲取待訓(xùn)練的原樣本；

擴(kuò)增各原樣本，得到各原樣本對(duì)應(yīng)的正樣本以及負(fù)樣本；

利用各原樣本以及各原樣本對(duì)應(yīng)的正樣本、負(fù)樣本、弱樣本，構(gòu)造各原樣本對(duì)應(yīng)的樣本組；

分別將各樣本組中的原樣本以及其他樣本中的一種作為圖神經(jīng)網(wǎng)絡(luò)的輸入，對(duì)所述圖神經(jīng)網(wǎng)絡(luò)進(jìn)行預(yù)訓(xùn)練，直至所述圖神經(jīng)網(wǎng)絡(luò)收斂。

2.根據(jù)權(quán)利要求1所述的方法，其中，擴(kuò)增各原樣本得到各原樣本對(duì)應(yīng)的正樣本包括：

通過隱藏部分節(jié)點(diǎn)的屬性、隱藏部分邊的屬性、增加部分節(jié)點(diǎn)之間的邊以及刪除部分節(jié)點(diǎn)之間的邊中的至少一種，處理各原樣本；

將處理結(jié)果作為各原樣本對(duì)應(yīng)的正樣本。

3.根據(jù)權(quán)利要求1所述的方法，其中，擴(kuò)增各原樣本得到各原樣本對(duì)應(yīng)的負(fù)樣本包括：

通過隱藏全部節(jié)點(diǎn)的屬性、隱藏全部邊的屬性以及改變樣本中全部節(jié)點(diǎn)的結(jié)構(gòu)中的至少一種，處理各原樣本；

將處理結(jié)果作為各原樣本對(duì)應(yīng)的負(fù)樣本。

4.根據(jù)權(quán)利要求1所述的方法，其中，各原樣本對(duì)應(yīng)的弱樣本包含其他原樣本對(duì)應(yīng)的正樣本以及負(fù)樣本中的至少一種。

5.一種預(yù)訓(xùn)練圖神經(jīng)網(wǎng)絡(luò)的裝置，包括：

獲取單元，用于獲取待訓(xùn)練的原樣本；

處理單元，用于擴(kuò)增各原樣本，得到各原樣本對(duì)應(yīng)的正樣本以及負(fù)樣本；

構(gòu)造單元，用于利用各原樣本以及各原樣本對(duì)應(yīng)的正樣本、負(fù)樣本、弱樣本，構(gòu)造各原樣本對(duì)應(yīng)的樣本組；

訓(xùn)練單元，用于分別將各樣本組中的原樣本以及其他樣本中的一種作為圖神經(jīng)網(wǎng)絡(luò)的輸入，對(duì)所述圖神經(jīng)網(wǎng)絡(luò)進(jìn)行預(yù)訓(xùn)練，直至所述圖神經(jīng)網(wǎng)絡(luò)收斂。

6.根據(jù)權(quán)利要求5所述的裝置，其中，所述處理單元在擴(kuò)增各原樣本得到各原樣本對(duì)應(yīng)的正樣本時(shí)，具體執(zhí)行：

通過隱藏部分節(jié)點(diǎn)的屬性、隱藏部分邊的屬性、增加部分節(jié)點(diǎn)之間的邊以及刪除部分節(jié)點(diǎn)之間的邊中的至少一種，處理各原樣本；

將處理結(jié)果作為各原樣本對(duì)應(yīng)的正樣本。

7.根據(jù)權(quán)利要求5所述的裝置，其中，所述處理單元在擴(kuò)增各原樣本得到各原樣本對(duì)應(yīng)的負(fù)樣本時(shí)，具體執(zhí)行：

通過隱藏全部節(jié)點(diǎn)的屬性、隱藏全部邊的屬性以及改變樣本中全部節(jié)點(diǎn)的結(jié)構(gòu)中的至少一種，處理各原樣本；

將處理結(jié)果作為各原樣本對(duì)應(yīng)的負(fù)樣本。

8.根據(jù)權(quán)利要求5所述的裝置，其中，所述構(gòu)造單元中各原樣本對(duì)應(yīng)的弱樣本包含其他原樣本對(duì)應(yīng)的正樣本以及負(fù)樣本中的至少一種。

9.一種電子設(shè)備，其特征在于，包括：

至少一個(gè)處理器；以及

與所述至少一個(gè)處理器通信連接的存儲(chǔ)器；其中，

所述存儲(chǔ)器存儲(chǔ)有可被所述至少一個(gè)處理器執(zhí)行的指令，所述指令被所述至少一個(gè)處理器執(zhí)行，以使所述至少一個(gè)處理器能夠執(zhí)行權(quán)利要求1-4中任一項(xiàng)所述的方法。

10.一種存儲(chǔ)有計(jì)算機(jī)指令的非瞬時(shí)計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，其特征在于，所述計(jì)算機(jī)指令用于使所述計(jì)算機(jī)執(zhí)行權(quán)利要求1-4中任一項(xiàng)所述的方法。