公开(公告)号：US20240176982A1

公开(公告)日：2024-05-30

申请号：US18283131

申请日：2022-05-30

Applicant: DeepMind Technologies Limited

Inventor： Jonathan William Godwin ,  Peter William Battaglia ,  Kevin Michael Schaarschmidt ,  Alvaro Sanchez

IPC: G06N3/04 ,  G06N3/084

CPC classification number: G06N3/04 ,  G06N3/084

Abstract: Methods, systems, and apparatus, including computer programs encoded on a computer storage medium, for training a neural network that includes one or more graph neural network layers. In one aspect, a method comprises: generating data defining a graph, comprising: generating a respective final feature representation for each node, wherein, for each of one or more of the nodes, the respective final feature representation is a modified feature representation that is generated from a respective feature representation for the node using respective noise; processing the data defining the graph using one or more of the graph neural network layers of the neural network to generate a respective updated node embedding of each node; and processing, for each of one or more of the nodes having modified feature representations, the updated node embedding of the node to generate a respective de-noising prediction for the node.