Link prediction with multiple node types

[remingtonkim]

Hey there Muhan,

I was wondering if SEAL worked on graphs with multiple types of nodes (i.e. heterogeneous networks). Take the graph below, for example:

If I would like to my GNN to predict only if there is a link between between green circles and blue triangles, would I just make sure that the central nodes x and y are blue triangles and green circles, or vice versa during the subgraph extraction process?

Thanks!

[muhanzhang]

Yes! With some modifications of the GNN used SEAL is able to predict links in heterogeneous graphs. Actually matrix completion is such an example with two node types and five edge types, for which you can refer to my another paper: https://github.com/muhanzhang/IGMC

[remingtonkim]

Thank you for the info!