公开(公告)号：US20230281430A1

公开(公告)日：2023-09-07

申请号：US18118061

申请日：2023-03-06

Applicant: Google LLC

Inventor： Ali Kemal Sinop ,  Sreenivas Gollapudi ,  Petar Velickovic ,  Sofia Ira Ktena ,  Ameya Avinash Velingker

IPC: G06N3/047 ,  G06N3/082

CPC classification number: G06N3/047 ,  G06N3/082

Abstract: Methods and systems for conditioning graph neural networks on affinity features. One of the methods includes obtaining graph data representing an input graph that comprises a set of nodes and a set of edges that each connect a respective pair of nodes, the graph data comprising respective node features for each of the nodes, edge features for each of the edges, and a respective weight for each of the edges; generating one or more affinity features, each affinity feature representing a property of one or more random walks through the graph guided by the respective weights for the edges; and processing the graph data using a graph neural network that is conditioned on the one or more affinity features to generate a task prediction for a machine learning task for the input graph.