mlsanityData to 14 Oct 2026

Heuristic Methods are Good Teachers to Distill MLPs for Graph Link Prediction

arXiv 2504.06193 ↗First posted 8 Apr 2025cs.LG cs.AIcited by 0

On the map: Knowledge distillation → Graph neural network distillation

Abstract (arXiv)

Link prediction is a crucial graph-learning task with applications including citation prediction and product recommendation. Distilling Graph Neural Networks (GNNs) teachers into Multi-Layer Perceptrons (MLPs) students has emerged as an effective approach to achieve strong performance and reducing computational cost by removing graph dependency. However, existing distillation methods only use standard GNNs and overlook alternative teachers such as specialized model for link prediction (GNN4LP) and heuristic methods (e.g., common neighbors). This paper first explores the impact of different teachers in GNN-to-MLP distillation. Surprisingly, we find that stronger teachers do not always produce stronger students: MLPs distilled from GNN4LP can underperform those distilled from simpler GNNs, while weaker heuristic methods can teach MLPs to near-GNN performance with drastically reduced training costs. Building on these insights, we propose Ensemble Heuristic-Distilled MLPs (EHDM), which eliminates graph dependencies while effectively integrating complementary signals via a gating mechanism. Experiments on ten datasets show an average 7.93% improvement over previous GNN-to-MLP approaches with 1.95-3.32 times less training time, indicating EHDM is an efficient and effective link prediction method. Our code is available at https://github.com/ZongyueQin/EHDM

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Graph neural network distillation, one of 12 directions in Knowledge distillation. Builds on 41 of its 56 references that are in the map. First posted on arXiv 8 Apr 2025.

Builds on

41 of 56 references.

Closest ancestors · 3

Same direction, newest first

  1. 2024Teach Harder, Learn Poorer: Rethinking Hard Sample Distillation for GNN-to-MLP Knowledge Distillation20 Jul
  2. 2024AdaGMLP: AdaBoosting GNN-to-MLP Knowledge Distillation23 May
  3. 2023Quantifying the Knowledge in GNNs for Reliable Distillation into MLPs9 Jun

Other references · 38

  1. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
  2. 2017Graph Attention Networkscited by 8,923
  3. 2017Inductive Representation Learning on Large Graphscited by 6,320
  4. 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,163
  5. 2015Learning both Weights and Connections for Efficient Neural Networkscited by 2,031
  6. 2017Modeling Relational Data with Graph Convolutional Networkscited by 1,694
  7. 2016Variational Graph Auto-Encoderscited by 1,382
  8. 2020LightGCN: Simplifying and Powering Graph Convolution Network for Recommendationcited by 1,139
  9. 2018Graph Convolutional Neural Networks for Web-Scale Recommender Systemscited by 1,138
  10. 2018Pitfalls of Graph Neural Network Evaluationcited by 1,007
  11. 2016Revisiting Semi-Supervised Learning with Graph Embeddingscited by 976
  12. 2018Link Prediction Based on Graph Neural Networkscited by 832
26 more
  1. 2015Image-based Recommendations on Styles and Substitutescited by 706
  2. 2019Composition-based Multi-Relational Graph Convolutional Networkscited by 396
  3. 2015Deep Learning with Limited Numerical Precisioncited by 376
  4. 2021Neural Bellman-Ford Networks: A General Graph Neural Network Framework for Link Predictioncited by 275
  5. 2018Hyperspherical Variational Auto-Encoderscited by 189
  6. 2021Graph-less Neural Networks: Teaching Old MLPs New Tricks via Distillationcited by 152
  7. 2020Labeling Trick: A Theory of Using Graph Neural Networks for Multi-Node Representation Learningcited by 149
  8. 2019Learning Attention-based Embeddings for Relation Prediction in Knowledge Graphscited by 125
  9. 2021A Unified Lottery Ticket Hypothesis for Graph Neural Networkscited by 118
  10. 2020Degree-Quant: Quantization-Aware Training for Graph Neural Networkscited by 96
  11. 2022Graph Neural Networks for Link Prediction with Subgraph Sketchingcited by 81
  12. 2022Neo-GNNs: Neighborhood Overlap-aware Graph Neural Networks for Link Predictioncited by 78
  13. 2023Evaluating Graph Neural Networks for Link Prediction: Current Pitfalls and New Benchmarkingcited by 69
  14. 2023Neural Common Neighbor with Completion for Link Predictioncited by 64
  15. 2021Annealing Knowledge Distillationcited by 61
  16. 2021Accelerating Large Scale Real-Time GNN Inference using Channel Pruningcited by 41
  17. 2022Better Teacher Better Student: Dynamic Prior Knowledge for Knowledge Distillationcited by 41
  18. 2022Linkless Link Prediction via Relational Distillationcited by 39
  19. 2023IGB: Addressing The Gaps In Labeling, Features, Heterogeneity, and Size of Public Graph Datasets for Deep Learning Researchcited by 30
  20. 2023Revisiting Link Prediction: A Data Perspectivecited by 29
  21. 2020Learned Low Precision Graph Neural Networkscited by 25
  22. 2023Extracting Low-/High- Frequency Knowledge from Graph Neural Networks and Injecting it into MLPs: An Effective GNN-to-MLP Distillation Frameworkcited by 24
  23. 2023Pure Message Passing Can Estimate Common Neighbor for Link Predictioncited by 18
  24. 2024Mixture of Link Predictors on Graphscited by 10
  25. 2024Training MLPs on Graphs without Supervisioncited by 8
  26. 2018Structural transition in social networks: The role of homophilycited by 4

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2023Unveiling the Unseen Potential of Graph Learning through MLPs: Effective Graph Learners Using Propagation-Embracing MLPs
  2. 2023Propagate & Distill: Towards Effective Graph Learners Using Propagation-Embracing MLPs
  3. 2026InfGraND: An Influence-Guided GNN-to-MLP Knowledge Distillation
  4. 2024A Teacher-Free Graph Knowledge Distillation Framework with Dual Self-Distillation
  5. 2022SA-MLP: Distilling Graph Knowledge from GNNs into Structure-Aware MLP
  6. 2024Teaching MLP More Graph Information: A Three-stage Multitask Knowledge Distillation Framework
  7. 2023Train Your Own GNN Teacher: Graph-Aware Distillation on Textual Graphs
  8. 2021Extract the Knowledge of Graph Neural Networks and Go Beyond it: An Effective Knowledge Distillation Framework