MAGNN: Metapath Aggregated Graph Neural Network for Heterogeneous Graph Embedding
About
A large number of real-world graphs or networks are inherently heterogeneous, involving a diversity of node types and relation types. Heterogeneous graph embedding is to embed rich structural and semantic information of a heterogeneous graph into low-dimensional node representations. Existing models usually define multiple metapaths in a heterogeneous graph to capture the composite relations and guide neighbor selection. However, these models either omit node content features, discard intermediate nodes along the metapath, or only consider one metapath. To address these three limitations, we propose a new model named Metapath Aggregated Graph Neural Network (MAGNN) to boost the final performance. Specifically, MAGNN employs three major components, i.e., the node content transformation to encapsulate input node attributes, the intra-metapath aggregation to incorporate intermediate semantic nodes, and the inter-metapath aggregation to combine messages from multiple metapaths. Extensive experiments on three real-world heterogeneous graph datasets for node classification, node clustering, and link prediction show that MAGNN achieves more accurate prediction results than state-of-the-art baselines.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Node Classification | IMDB | Macro F1 Score0.5649 | 179 | |
| Node Classification | ACM | Macro F190.88 | 104 | |
| Node Classification | DBLP | Micro-F193.76 | 94 | |
| Node Classification | DBLP (test) | Macro-F193.28 | 70 | |
| Node Classification | IMDB (test) | Macro F1 Score56.49 | 70 | |
| Node Classification | DBLP HGB (test) | Macro F193.28 | 27 | |
| Node Classification | IMDB HGB (test) | Macro F161.36 | 27 | |
| Node Classification | ACM HGB (test) | Macro F190.88 | 27 | |
| Node Classification | DBLP (20% train) | Macro F193.13 | 20 | |
| Node Classification | DBLP (80% train) | Macro F194.1 | 20 |