Echoless Label-Based Pre-computation for Memory-Efficient Heterogeneous Graph Learning
Jun Hu, Shangheng Chen, Yufei He, Yuan Li, Bryan Hooi, Bingsheng He
Abstract
Heterogeneous Graph Neural Networks (HGNNs) are widely used for deep learning on heterogeneous graphs. Typical end-to-end HGNNs require repetitive message passing during training, limiting efficiency for large-scale real-world graphs. Pre-computation-based HGNNs address this by performing message passing only once during preprocessing, collecting neighbor information into regular-shaped tensors, which enables efficient mini-batch training. Label-based pre-computation methods collect neighbors' label information but suffer from training label leakage, where a node's own label information propagates back to itself during multi-hop message passing—the echo effect. Existing mitigation strategies are memory-inefficient on large graphs or suffer from compatibility issues with advanced message passing methods. We propose Echoless Label-based Pre-computation (Echoless-LP), which eliminates training label leakage with Partition-Focused Echoless Propagation (PFEP). PFEP partitions target nodes and performs echoless propagation, where nodes in each partition collect label information only from neighbors in other partitions, avoiding echo while remaining memory-efficient and compatible with any message passing method. We also introduce an Asymmetric Partitioning Scheme (APS) and a PostAdjust mechanism to address information loss from partitioning and distributional shifts across partitions. Experiments on public datasets demonstrate that Echoless-LP achieves superior performance and maintains memory efficiency compared to baselines.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 25e9030d-76d9-4119-bc93-8118e427fdaeCited by top-tier papers1
Ask how each one uses itBuilds on8
- Open Graph Benchmark: Datasets for Machine Learning on GraphsWeihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong et al.NeurIPS 2020 · 3,935 citations
- MAGNN: Metapath Aggregated Graph Neural Network for Heterogeneous Graph EmbeddingXinyu Fu, Jiani Zhang, Ziqiao Meng, Irwin KingWWW 2020 · 1,149 citations
- GPT-GNN: Generative Pre-Training of Graph Neural NetworksZiniu Hu, Yuxiao Dong, Kuansan Wang, Kai-Wei Chang et al.KDD 2020 · 438 citations
- An Attention-Based Graph Neural Network for Heterogeneous Structural LearningHuiting Hong, Hantao Guo, Yucheng Lin, Xiaoqing Yang et al.AAAI 2020 · 278 citations
- Are we really making much progress?: Revisiting, benchmarking and refining heterogeneous graph neural networksQingsong Lv, Ming Ding, Qiang Liu, Yuxiang Chen et al.KDD 2021 · 249 citations
Related papers
- Revisiting Pre-Propagation GNNs: Robust Diffusion Operators and Hidden-State Re-PropagationZichao Yue, Zhiru ZhangICML 2026
- Pre-training on Large-Scale Heterogeneous GraphXunqiang Jiang, Tianrui Jia, Yuan Fang, Chuan Shi et al.KDD 2021 · 44 citations
- LD2: Scalable Heterophilous Graph Neural Network with Decoupled EmbeddingsNingyi Liao, Siqiang Luo, Xiang Li, Jieming ShiNeurIPS 2023 · 23 citations
- Self-supervised Heterogeneous Graph Pre-training Based on Structural ClusteringYaming Yang, Ziyu Guan, Zhe Wang, Wei Zhao et al.NeurIPS 2022 · 70 citations
- LeapGNN: Accelerating Distributed GNN Training Leveraging Feature-Centric Model MigrationWeijian Chen, Shuibing He, Haoyang Qu, Xuechen ZhangFAST 2025 · 5 citations
