4 个月前

基于快速局部谱滤波的图卷积神经网络

基于快速局部谱滤波的图卷积神经网络

摘要

在这项工作中,我们致力于将卷积神经网络(CNNs)从低维规则网格(图像、视频和语音在此类网格上表示)推广到高维不规则域,例如社交网络、脑连接组或词嵌入(这些域由图表示)。我们提出了在谱图理论框架下对CNNs进行公式化的方案,该理论提供了必要的数学背景和高效的数值方法来设计图上的快速局部卷积滤波器。重要的是,所提出的技术具有与经典CNN相同的线性计算复杂度和常数学习复杂度,同时适用于任何图结构。通过在MNIST和20NEWS数据集上的实验,证明了这一新型深度学习系统能够学习图上的局部、平稳和组合特征。

代码仓库

mdeff/cnn_graph
官方
tf
GitHub 中提及
hazdzz/ChebyNet
pytorch
GitHub 中提及
ajbisberg/gcn
tf
GitHub 中提及
selmiss/gp-tlstgcn
pytorch
GitHub 中提及

基准测试

基准方法指标
graph-property-prediction-on-ogbg-molpcbaChebNet
Ext. data: No
Number of params: 1475003
Test AP: 0.2306 ± 0.0016
Validation AP: 0.2372 ± 0.0018
node-classification-on-citeseerChebNet
Accuracy: 69.8%
node-classification-on-citeseer-05ChebyNet
Accuracy: 45.3%
node-classification-on-citeseer-1ChebyNet
Accuracy: 59.4%
node-classification-on-citeseer-with-publicChebyNet
Accuracy: 70.1%
node-classification-on-coraChebNet
Accuracy: 81.2%
node-classification-on-cora-05ChebyNet
Accuracy: 33.9%
node-classification-on-cora-1ChebyNet
Accuracy: 44.2%
node-classification-on-cora-3ChebyNet
Accuracy: 62.1%
node-classification-on-cora-with-public-splitChebyNet
Accuracy: 78.0%
node-classification-on-pubmedChebNet
Accuracy: 74.4%
node-classification-on-pubmed-003ChebyNet
Accuracy: 45.3%
node-classification-on-pubmed-005ChebyNet
Accuracy: 48.2%
node-classification-on-pubmed-01ChebyNet
Accuracy: 55.2%
node-classification-on-pubmed-with-publicChebyNet
Accuracy: 69.8%
skeleton-based-action-recognition-on-sbuChebyNet
Accuracy: 96.00%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于快速局部谱滤波的图卷积神经网络 | 论文 | HyperAI超神经