公开(公告)号：US20240394507A1

公开(公告)日：2024-11-28

申请号：US18714656

申请日：2022-11-04

Applicant: Lemon Inc.

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

IPC: G06N3/042 ,  G06N5/01

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.