公开(公告)号：EP4425353A2

公开(公告)日：2024-09-04

申请号：EP22901924.5

申请日：2022-11-04

Applicant: Lemon Inc.

Inventor： ZHU, Yibo ,  CHEN, Yangrui ,  HE, Jun ,  PENG, Yanghua ,  GUO, Chuanxiong ,  WANG, Jian

IPC: G06F18/214 ,  G06N3/08 ,  G06N3/04 ,  G06F18/25

Abstract: The disclosure relates to a method, apparatus, system, medium and electronic device for graph neural network generation. The method includes: obtaining a subgraph structure, the subgraph structure being configured to reflect a graph structure of a corresponding subgraph, and the subgraph comprising a plurality of nodes and edges between the plurality of nodes; obtaining, based on the subgraph structure and according to a predetermined priority, node features of the plurality of nodes and edge features of the edges from a plurality of memories; the predetermined priority being obtained by sorting the plurality of memories in accordance with memory size in an ascending order; fusing, based on the subgraph structure, the node features of the plurality of nodes and the edge features of the edges to obtain subgraph data; and training, based on the subgraph data, the graph neural network. The method for graph neural network generation of the present disclosure may improve the efficiency of graph neural network training.