3 个月前

GraphAIR:基于邻域聚合与交互的图表示学习

GraphAIR:基于邻域聚合与交互的图表示学习

摘要

图表示学习在各类图分析任务中具有至关重要的作用,涵盖节点分类、社区检测等多个方面。近年来,图卷积网络(Graph Convolutional Networks, GCNs)已被成功应用于图表示学习。这类模型通过聚合邻域特征来生成节点表示,遵循“邻域聚合”机制。尽管在多种任务上取得了令人瞩目的性能,现有基于GCN的模型在捕捉图数据复杂的非线性特性方面仍存在明显局限。本文首次从理论上证明,当前模型中邻域交互项的系数相对较小,这解释了为何GCN的性能往往仅略优于线性模型。为更好地建模图数据的复杂非线性关系,本文提出一种新颖的GraphAIR框架,该框架在传统邻域聚合的基础上,进一步显式建模邻域之间的交互关系。在多个公开数据集上,针对节点分类与链接预测等基准任务所开展的全面实验表明,所提出方法具有显著的有效性。

代码仓库

CRIPAC-DIG/GraphAIR
官方
tf
GitHub 中提及

基准测试

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
GraphAIR:基于邻域聚合与交互的图表示学习 | 论文 | HyperAI超神经