摘要
arXiv:2504.06193v1 交叉类型:cross
摘要:链接预测是图学习任务中的关键任务,其应用包括引用预测和产品推荐。将图神经网络(GNNs)教师精炼为多层感知器(MLPs)学生,已成为一种有效的减少计算成本并实现高性能的方法,通过去除图依赖性。然而,现有的精炼方法仅使用标准GNNs,忽略了链接预测专用模型(GNN4LP)和启发式方法(例如,共同邻居)等替代教师。本文首先探索GNN-to-MLP精炼中不同教师的影响。令人惊讶的是,我们发现更强的教师并不总是产生更强的学生:从GNN4LP精炼出的MLPs有时会不如从更简单的GNNs精炼出的MLPS,而较弱的启发式方法可以通过大幅减少训练成本来教MLPs达到接近GNNs的性能。基于这些见解,我们提出了一种集成启发式精炼的MLP(EHDM),它通过门控机制有效地整合互补信号,同时消除图依赖性。在十个数据集上的实验表明,与之前的GNN-to-MLP方法相比,EHDM平均提高了7.93%的性能,且训练时间减少了1.95-3.32倍,表明EHDM是一种高效且有效的链接预测方法。