4 个月前

Edge2Node:将边预测转化为节点分类

Edge2Node:将边预测转化为节点分类

摘要

尽管图神经网络模型在节点分类任务中取得了成功,但边预测(即预测图中节点之间缺失或潜在的链接)仍然是这些模型面临的一个挑战性问题。一种常见的边预测方法是首先获取两个节点的嵌入表示,然后使用预定义的评分函数来预测这两个节点之间是否存在边。本文介绍了一种初步的概念——Edge2Node,该概念建议直接为每条边生成嵌入表示,而无需使用评分函数。Edge2Node 的目标是基于给定的用于边预测任务的图 G 创建一个新的图 H,并将图 G 上的边预测任务转化为图 H 上的节点分类任务。我们预计这一初步方法能够激发对边预测任务进一步的研究。

基准测试

基准方法指标
link-prediction-on-ogbl-collabEdge2Node
Test Hits@50: 0.9515
link-property-prediction-on-ogbl-collabE2N
Ext. data: No
Number of params: 526851
Test Hits@50: 0.9515 ± 0.1410
Validation Hits@50: 0.9546 ± 0.1270
link-property-prediction-on-ogbl-ppa** E2N**
Ext. data: No
Number of params: 526851
Test Hits@100: 0.8911 ± 0.1266
Validation Hits@100: 0.8857 ± 0.1331

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
Edge2Node:将边预测转化为节点分类 | 论文 | HyperAI超神经