公开(公告)号：US20210383228A1

公开(公告)日：2021-12-09

申请号：US17338974

申请日：2021-06-04

Applicant: DeepMind Technologies Limited

Inventor： Petar Velickovic ,  Charles Blundell ,  Oriol Vinyals ,  Razvan Pascanu ,  Lars Buesing ,  Matthew Overlan

IPC: G06N3/08 ,  G06N3/04 ,  G06F16/23 ,  G06F16/901

Abstract: Methods, systems, and apparatus, including computer programs encoded on a computer storage medium, for generating prediction outputs characterizing a set of entities. In one aspect, a method comprises: obtaining data defining a graph, comprising: (i) a set of nodes, wherein each node represents a respective entity from the set of entities, (ii) a current set of edges, wherein each edge connects a pair of nodes, and (iii) a respective current embedding of each node; at each of a plurality of time steps: updating the respective current embedding of each node, comprising processing data defining the graph using a graph neural network; and updating the current set of edges based at least in part on the updated embeddings of the nodes; and at one or more of the plurality of time steps: generating a prediction output characterizing the set of entities based on the current embeddings of the nodes.