[發(fā)明專利]圖池化方法、裝置、設(shè)備以及存儲(chǔ)介質(zhì)

在審

• 申請(qǐng)?zhí)枺?/td>: 202011034251.3
• 申請(qǐng)日: 2020-09-27
• 公開(kāi)（公告）號(hào): CN112132223A
• 公開(kāi)（公告）日: 2020-12-25
• 發(fā)明（設(shè)計(jì)）人: 曹雨;方蒙
• 申請(qǐng)（專利權(quán)）人: 騰訊科技（深圳）有限公司
• 主分類號(hào): G06K9/62
• 分類號(hào): G06K9/62;G06N3/04;G06N3/08
• 代理公司: 廣州三環(huán)專利商標(biāo)代理有限公司 44202
• 代理人: 熊永強(qiáng);杜維

1.一種圖池化方法，其特征在于，所述方法包括：

獲取待池化圖特征表示，并將所述待池化圖特征表示分割為多個(gè)子圖特征表示，一個(gè)子圖特征表示中包括所述待池化圖特征表示中的一個(gè)或者多個(gè)維度的圖特征表示；

基于圖神經(jīng)網(wǎng)絡(luò)模型確定各子圖特征表示對(duì)應(yīng)的圖注意力輸出特征，一個(gè)子圖特征表示對(duì)應(yīng)一個(gè)圖注意力輸出特征；

基于所述各子圖特征表示對(duì)應(yīng)的圖注意力輸出特征聚合得到待篩選圖特征表示，所述待篩選圖特征表示中包括所述待池化圖特征表示中的各個(gè)維度對(duì)應(yīng)的特征表示；

根據(jù)所述待篩選圖特征表示中包括的多個(gè)待篩選對(duì)象的特征表示以及各待篩選對(duì)象的權(quán)重，從所述待篩選圖特征中篩選出目標(biāo)圖特征，并基于所述目標(biāo)圖特征確定池化后的目標(biāo)圖特征表示。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述基于所述各子圖特征表示對(duì)應(yīng)的圖注意力輸出特征聚合得到待篩選圖特征表示之后，所述方法還包括：

利用權(quán)重函數(shù)對(duì)所述待篩選圖特征表示中包括的各待篩選對(duì)象的特征表示進(jìn)行計(jì)算，得到所述各待篩選對(duì)象的權(quán)重。

3.根據(jù)權(quán)利要求1或2中所述的方法，其特征在于，所述根據(jù)所述待篩選圖特征表示中包括的多個(gè)待篩選對(duì)象的特征表示以及各待篩選對(duì)象的權(quán)重，從所述待篩選圖特征中篩選出目標(biāo)圖特征包括：

將所述待篩選圖特征表示中包括的多個(gè)待篩選對(duì)象按照各待篩選對(duì)象的權(quán)重進(jìn)行排序，將權(quán)重大于權(quán)重閾值的待篩選對(duì)象確定為目標(biāo)對(duì)象；

根據(jù)所述目標(biāo)對(duì)象確定目標(biāo)圖特征；

其中，所述待篩選對(duì)象包括節(jié)點(diǎn)和/或邊。

4.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述根據(jù)所述目標(biāo)對(duì)象，確定出所述目標(biāo)圖特征，包括：

根據(jù)所述目標(biāo)對(duì)象確定目標(biāo)對(duì)象索引，基于所述目標(biāo)對(duì)象索引生成目標(biāo)對(duì)象掩膜；

根據(jù)所述目標(biāo)對(duì)象掩膜以及所述待篩選圖特征表示確定目標(biāo)圖特征。

5.根據(jù)權(quán)利要求1-4任一項(xiàng)所述的方法，其特征在于，所述基于所述目標(biāo)圖特征確定池化后的目標(biāo)圖特征表示之后，所述方法還包括：

基于所述池化后的目標(biāo)圖特征表示，確定池化后的目標(biāo)圖片并輸出所述目標(biāo)圖片。

6.根據(jù)權(quán)利要求1-5任一項(xiàng)所述的方法，其特征在于，所述圖神經(jīng)網(wǎng)絡(luò)模型包括：圖卷積網(wǎng)絡(luò)模型、圖注意力網(wǎng)絡(luò)模型、圖自編碼器模型、圖生成網(wǎng)絡(luò)模型和圖時(shí)空網(wǎng)絡(luò)模型中的一種或幾種。

7.一種圖池化裝置，其特征在于，包括：

特征獲取模塊，用于獲取待池化圖特征表示；

特征分割模塊，用于將所述待池化圖特征表示分割為多個(gè)子圖特征表示，一個(gè)子圖特征表示中包括所述待池化圖特征表示中的一個(gè)或者多個(gè)維度的圖特征表示；

特征訓(xùn)練模塊，用于基于圖神經(jīng)網(wǎng)絡(luò)模型確定出各子圖特征表示對(duì)應(yīng)的圖注意力輸出特征，一個(gè)子圖特征表示對(duì)應(yīng)一個(gè)圖注意力輸出特征；

特征拼接模塊，用于基于所述各子圖特征表示對(duì)應(yīng)的圖注意力輸出特征聚合得到待篩選圖特征表示，所述待篩選圖特征表示中包括所述待池化圖特征表示中的各個(gè)維度對(duì)應(yīng)的待篩選特征表示；

特征篩選模塊，用于根據(jù)所述待篩選圖特征表示中包括的多個(gè)待篩選對(duì)象的特征表示以及各待篩選對(duì)象的權(quán)重，從所述待篩選圖特征中篩選出目標(biāo)圖特征；

特征表示模塊，用于基于所述目標(biāo)圖特征確定池化后的目標(biāo)圖特征表示。

8.根據(jù)權(quán)利要求7所述的裝置，其特征在于，所述裝置還包括：

圖片輸出模塊，用于基于所述池化后的目標(biāo)圖特征表示，確定池化后的目標(biāo)圖片并輸出所述目標(biāo)圖片。

9.一種終端設(shè)備，其特征在于，包括處理器和存儲(chǔ)器，所述處理器和存儲(chǔ)器相互連接；

所述存儲(chǔ)器用于存儲(chǔ)計(jì)算機(jī)程序，所述計(jì)算機(jī)程序包括程序指令，所述處理器被配置用于調(diào)用所述程序指令，執(zhí)行如權(quán)利要求1-6任一項(xiàng)所述的方法。

10.一種計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，其特征在于，所述計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)存儲(chǔ)有計(jì)算機(jī)程序，所述計(jì)算機(jī)程序被處理器執(zhí)行以實(shí)現(xiàn)權(quán)利要求1-6任一項(xiàng)所述的方法。