Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Space4HGNN: A Novel, Modularized and Reproducible Platform to Evaluate Heterogeneous Graph Neural Network

About

Heterogeneous Graph Neural Network (HGNN) has been successfully employed in various tasks, but we cannot accurately know the importance of different design dimensions of HGNNs due to diverse architectures and applied scenarios. Besides, in the research community of HGNNs, implementing and evaluating various tasks still need much human effort. To mitigate these issues, we first propose a unified framework covering most HGNNs, consisting of three components: heterogeneous linear transformation, heterogeneous graph transformation, and heterogeneous message passing layer. Then we build a platform Space4HGNN by defining a design space for HGNNs based on the unified framework, which offers modularized components, reproducible implementations, and standardized evaluation for HGNNs. Finally, we conduct experiments to analyze the effect of different designs. With the insights found, we distill a condensed design space and verify its effectiveness.

Tianyu Zhao, Cheng Yang, Yibo Li, Quan Gan, Zhenyi Wang, Fengqi Liang, Huan Zhao, Yingxia Shao, Xiao Wang, Chuan Shi• 2022

Related benchmarks

TaskDatasetResultRank
Node ClassificationIMDB
Macro F1 Score0.6157
179
Link PredictionPubmed
AUC81.53
123
Node ClassificationACM
Macro F192.5
104
Node ClassificationDBLP
Micro-F194.63
94
Node ClassificationFreebase
Macro F141.37
43
Link PredictionLastFM
ROC-AUC0.6689
18
Node ClassificationPubMed NC
Macro F1 Score45.53
10
Showing 7 of 7 rows

Other info

Follow for update