4 个月前

DEMO-Net:特定度数的图神经网络用于节点和图分类

DEMO-Net:特定度数的图神经网络用于节点和图分类

摘要

图数据在许多高影响力的应用中广泛存在。受深度学习在网格结构数据中取得成功的启发,研究人员提出了图神经网络模型,以学习强大的节点级或图级表示。然而,现有的大多数图神经网络都存在以下局限性:(1) 对图卷积性质(如种子导向、度感知和顺序无关)的分析有限;(2) 节点的特定度数图结构未在图卷积中显式表达,无法区分结构感知的节点邻域;(3) 图级池化方案的理论解释尚不清楚。为了解决这些问题,我们提出了一种通用的度数特定图神经网络——DEMO-Net,该网络受到Weisfeiler-Lehman图同构测试的启发,递归地识别1跳邻域结构。为了明确捕捉结合节点属性的图拓扑结构,我们认为图卷积应具备三个特性:种子导向、度感知和顺序无关。为此,我们提出了多任务图卷积,其中每个任务代表具有特定度数值的节点表示学习,从而保留了度数特定的图结构。具体而言,我们设计了两种多任务学习方法:度数特定权重和哈希函数用于图卷积。此外,我们还提出了一种新的图级池化/读出方案,用于学习可证明位于度数特定希尔伯特核空间中的图表示。实验结果表明,在多个节点分类和图分类基准数据集上,我们提出的DEMO-Net比最先进的图神经网络模型更有效且高效。

代码仓库

jwu4sml/DEMO-Net
官方
tf
GitHub 中提及

基准测试

基准方法指标
graph-classification-on-enzymesDEMO-Net(weight)
Accuracy: 27.2
node-classification-on-blogcatalog-1DEMO-Net(weight)
Accuracy: 84.9
node-classification-on-brazil-air-trafficDEMO-Net(weight)
Accuracy: 0.543 ± 0.034
node-classification-on-europe-air-trafficDEMO-Net(weight)
Accuracy: 45.9
node-classification-on-facebookDEMO-Net(weight)
Accuracy: 91.9
node-classification-on-flickrDEMO-Net(weight)
Accuracy: 0.656 ± 0.000
node-classification-on-usa-air-trafficDEMO-Net(weight)
Accuracy: 64.7
node-classification-on-wiki-voteDEMO-Net(weight)
Accuracy: 99.8

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
DEMO-Net:特定度数的图神经网络用于节点和图分类 | 论文 | HyperAI超神经