4 个月前

基于神经网络的链接预测对偶学习

基于神经网络的链接预测对偶学习

摘要

本文旨在提供一个有效的成对学习神经链接预测(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-citation2PLNLP
Ext. data: No
Number of params: 146514551
Test MRR: 0.8492 ± 0.0029
Validation MRR: 0.8490 ± 0.0031
link-property-prediction-on-ogbl-collabPLNLP (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-collabPLNLP (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-ddiPLNLP
Ext. data: No
Number of params: 3497473
Test Hits@20: 0.9088 ± 0.0313
Validation Hits@20: 0.8242 ± 0.0253

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于神经网络的链接预测对偶学习 | 论文 | HyperAI超神经