
摘要
本文旨在提供一个有效的成对学习神经链接预测(PLNLP)框架。该框架将链接预测视为成对学习排序问题,由四个主要组件构成,即邻域编码器、链接预测器、负采样器和目标函数。该框架具有灵活性,可以采用任何通用的图神经卷积或特定于链接预测的神经架构作为邻域编码器。对于链接预测器,我们设计了不同的评分函数,可以根据不同类型的图进行选择。在负采样器中,我们提供了几种针对具体问题的采样策略。至于目标函数,我们提出使用一种有效的排名损失函数,该函数近似最大化标准排名指标AUC(Area Under the Curve)。我们在Open Graph Benchmark的4个链接属性预测数据集上评估了所提出的PLNLP框架,包括ogbl-ddi、ogbl-collab、ogbl-ppa和ogbl-citation2。PLNLP仅使用基本的神经架构就在ogbl-ddi和ogbl-collab上取得了最佳性能,在ogbl-citation2上取得了第二佳性能。这些结果表明了PLNLP的有效性。
代码仓库
zhitao-wang/PLNLP
pytorch
GitHub 中提及
zhitao-wang/plnlp
官方
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| link-property-prediction-on-ogbl-citation2 | PLNLP | Ext. data: No Number of params: 146514551 Test MRR: 0.8492 ± 0.0029 Validation MRR: 0.8490 ± 0.0031 |
| link-property-prediction-on-ogbl-collab | PLNLP (val as input) | Ext. data: No Number of params: 35112192 Test Hits@50: 0.6872 ± 0.0052 Validation Hits@50: 1.0000 ± 0.0000 |
| link-property-prediction-on-ogbl-collab | PLNLP (random walk aug.) | Ext. data: No Number of params: 34980864 Test Hits@50: 0.7059 ± 0.0029 Validation Hits@50: 1.0000 ± 0.0000 |
| link-property-prediction-on-ogbl-ddi | PLNLP | Ext. data: No Number of params: 3497473 Test Hits@20: 0.9088 ± 0.0313 Validation Hits@20: 0.8242 ± 0.0253 |