[發明專利]圖池化方法、裝置、設備以及存儲介質

在審

• 申請號: 202011034251.3
• 申請日: 2020-09-27
• 公開（公告）號: CN112132223A
• 公開（公告）日: 2020-12-25
• 發明（設計）人: 曹雨;方蒙
• 申請（專利權）人: 騰訊科技（深圳）有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06N3/04;G06N3/08
• 代理公司: 廣州三環專利商標代理有限公司 44202
• 代理人: 熊永強;杜維

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

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

基于圖神經網絡模型確定各子圖特征表示對應的圖注意力輸出特征，一個子圖特征表示對應一個圖注意力輸出特征；

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

根據所述待篩選圖特征表示中包括的多個待篩選對象的特征表示以及各待篩選對象的權重，從所述待篩選圖特征中篩選出目標圖特征，并基于所述目標圖特征確定池化后的目標圖特征表示。

2.根據權利要求1所述的方法，其特征在于，所述基于所述各子圖特征表示對應的圖注意力輸出特征聚合得到待篩選圖特征表示之后，所述方法還包括：

利用權重函數對所述待篩選圖特征表示中包括的各待篩選對象的特征表示進行計算，得到所述各待篩選對象的權重。

3.根據權利要求1或2中所述的方法，其特征在于，所述根據所述待篩選圖特征表示中包括的多個待篩選對象的特征表示以及各待篩選對象的權重，從所述待篩選圖特征中篩選出目標圖特征包括：

將所述待篩選圖特征表示中包括的多個待篩選對象按照各待篩選對象的權重進行排序，將權重大于權重閾值的待篩選對象確定為目標對象；

根據所述目標對象確定目標圖特征；

其中，所述待篩選對象包括節點和/或邊。

4.根據權利要求3所述的方法，其特征在于，所述根據所述目標對象，確定出所述目標圖特征，包括：

根據所述目標對象確定目標對象索引，基于所述目標對象索引生成目標對象掩膜；

根據所述目標對象掩膜以及所述待篩選圖特征表示確定目標圖特征。

5.根據權利要求1-4任一項所述的方法，其特征在于，所述基于所述目標圖特征確定池化后的目標圖特征表示之后，所述方法還包括：

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

6.根據權利要求1-5任一項所述的方法，其特征在于，所述圖神經網絡模型包括：圖卷積網絡模型、圖注意力網絡模型、圖自編碼器模型、圖生成網絡模型和圖時空網絡模型中的一種或幾種。

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

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

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

特征訓練模塊，用于基于圖神經網絡模型確定出各子圖特征表示對應的圖注意力輸出特征，一個子圖特征表示對應一個圖注意力輸出特征；

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

特征篩選模塊，用于根據所述待篩選圖特征表示中包括的多個待篩選對象的特征表示以及各待篩選對象的權重，從所述待篩選圖特征中篩選出目標圖特征；

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

8.根據權利要求7所述的裝置，其特征在于，所述裝置還包括：

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

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

所述存儲器用于存儲計算機程序，所述計算機程序包括程序指令，所述處理器被配置用于調用所述程序指令，執行如權利要求1-6任一項所述的方法。

10.一種計算機可讀存儲介質，其特征在于，所述計算機可讀存儲介質存儲有計算機程序，所述計算機程序被處理器執行以實現權利要求1-6任一項所述的方法。