arrow
返回

Efficient Heterogeneous Graph Learning via Random Projection

delete2024-12-01
delete1
delete
OA
AI
J
Jun Hu
B
Bryan Hooi *
Bingsheng He 封面图
Bingsheng He (Bingsheng He)
DOI:10.1109/TKDE.2024.3434956delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Heterogeneous Graph Neural Networks (HGNNs) are powerful tools for deep learning on heterogeneous graphs. Typical HGNNs require repetitive message passing during training, limiting efficiency for large-scale real-world graphs. Recent pre-computation-based HGNNs use one-time message passing to transform a heterogeneous graph into regular-shaped tensors, enabling efficient mini-batch training. Existing pre-computation-based HGNNs can be mainly categorized into two styles, which differ in how much information loss is allowed and efficiency. We propose a hybrid pre-computation-based HGNN, named Random Projection Heterogeneous Graph Neural Network (RpHGNN), which combines the benefits of one style's efficiency with the low information loss of the other style. To achieve efficiency, the main framework of RpHGNN consists of propagate-then-update iterations, where we introduce a Random Projection Squashing step to ensure that complexity increases only linearly. To achieve low information loss, we introduce a Relation-wise Neighbor Collection component with an Even-odd Propagation Scheme, which aims to collect information from neighbors in a finer-grained way. Experimental results indicate that our approach achieves state-of-the-art results on seven small and large benchmark datasets while also being 230% faster compared to the most effective baseline. Surprisingly, our approach not only surpasses pre-processing-based baselines but also outperforms end-to-end methods.
Keyword:
Message passing
Training
Tensors
Graph neural networks
Vectors
Memory management
Graphics processing units
Heterogeneous graph neural networks
graph deep learning

期刊

IEEE Transactions on Knowledge and Data Engineering 封面图
IEEE Transactions on Knowledge and Data Engineering
IF:
10.4
论文数:
6.9K
被引数:
3.2W

机构

N
National University of Singapore
学者数:
7.6W
论文数: 6.5W
被引数: 11.4W
引用论文

引用论文

err分享
err收藏
Rotational spectra of indole in the lowest vibrational states
err2019-09-01
err0
PREAI
errK. Vávra; K. Luková; P. Kania; J. Koucký; Š. Urban
err分享
err收藏
Interpretation of aeromagnetic data in the Franklin area, northern New Zealand
err2017-01-09
err0
PREAI
errD. J. Robertson; P. N. P. Vidanovich; S. K. Zoellner; J. B. Meyers
err分享
err收藏
The heritability of insomnia: A meta‐analysis of twin studies
err2020-12-03
err0
errOAAI
errNicola L. Barclay; Desi Kocevska; Wichor M. Bramer; Eus J. W. Van Someren; Philip Gehrman
err分享
err收藏
Psychometric Network Analysis of the Hungarian WAIS
err2019-09-09
err0
errOAAI
errChristopher J. Schmank; Sara Anne Goring; Kristof Kovacs; Andrew R. A. Conway
err分享
err收藏
err
IF0
err
err0
PREAI
err
err分享
err收藏
学者 查看更多内容