3 个月前

GIDN:一种面向高效链接预测的轻量级图卷积扩散网络

GIDN:一种面向高效链接预测的轻量级图卷积扩散网络

摘要

本文提出了一种图Inception扩散网络(Graph Inception Diffusion Networks, GIDN)模型。该模型在不同特征空间中对图扩散过程进行了泛化,并引入Inception模块以避免复杂网络结构带来的大量计算开销。我们在开放图基准测试集(Open Graph Benchmark, OGB)上对GIDN模型进行了评估,在ogbl-collab数据集上的性能较AGDN模型提升了11%。

基准测试

基准方法指标
link-property-prediction-on-ogbl-collabGIDN@YITU
Ext. data: No
Number of params: 60449025
Test Hits@50: 0.7096 ± 0.0055
Validation Hits@50: 0.9620 ± 0.0040
link-property-prediction-on-ogbl-ddiGIDN@YITU
Ext. data: No
Number of params: 3506691
Test Hits@20: 0.9542 ± 0.0000
Validation Hits@20: 0.8258 ± 0.0000

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
GIDN:一种面向高效链接预测的轻量级图卷积扩散网络 | 论文 | HyperAI超神经