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

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

About

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

Zongyue Qin, Shichang Zhang, Mingxuan Ju, Tong Zhao, Neil Shah, Yizhou Sun• 2025

Related benchmarks

TaskDatasetResultRank
Link PredictionCora (test)--
69
Link PredictionPubMed (test)--
65
Link PredictionCiteseer (test)--
31
Link Predictionogbl-citation2--
23
Link PredictionCS (test)--
11
Link PredictionPubmed--
10
Link PredictionCOLLAB (test)
Hits@1049.27
5
Link PredictionCitation2 (test)
Hits@100.4658
4
Link PredictionCora
ΔGNN8.83
1
Link PredictionCiteseer
Delta GNN (ΔGNN)-7.08
1
Showing 10 of 16 rows

Other info

Follow for update