Generating Node Embeddings for Multiple Roles

A system comprising:

• a memory component;
• and
• a processing device coupled to the memory component, the processing device to perform operations comprising:
• clustering nodes of a graph into clusters;
• computing an initial role membership vector for each of the nodes based on the clusters;
• determining an updated role membership vector for each of the nodes by aggregating information from nodes connected to each of the nodes in the graph;
• generating a first set of role embeddings for a particular node of the nodes by using the updated role membership vector for the particular node to condition role embeddings of nodes connected to the particular node in the graph;
• and
• determining an indication of a graph prediction based on the first set of role embeddings and a second set of role embeddings for an additional node of the nodes.

A non-transitory computer-readable storage medium storing executable instructions, which when executed by a processing device, cause the processing device to perform operations comprising:

• clustering nodes of a graph into clusters;
• computing an initial role membership vector for each of the nodes based on the clusters;
• generating a first set of role embeddings for a particular node of the nodes based on the initial role membership vector for the particular node and nodes connected to the particular node in the graph;
• and
• determining an indication of at least one of a node classification or a link prediction for the graph based on the first set of role embeddings and a second set of role embeddings for an additional node of the nodes.