4 个月前

图卷积神经网络中的动态边条件滤波器

图卷积神经网络中的动态边条件滤波器

摘要

许多问题可以表述为在图结构数据上的预测。在这项工作中,我们从规则网格中推广卷积算子到任意图上,同时避免了频谱域的使用,这使得我们可以处理不同大小和连通性的图。为了超越简单的扩散过程,滤波器权重被设定为依赖于顶点邻域中的特定边标签。结合适当的图粗化选择,我们探讨了构建用于图分类的深度神经网络。特别是,我们在点云分类中展示了我们方法的普适性,并达到了新的最先进水平;在图分类数据集上,我们的方法也优于其他深度学习方法。源代码可在 https://github.com/mys007/ecc 获取。

代码仓库

mys007/ecc
官方
pytorch
GitHub 中提及
rusty1s/pytorch_cluster
pytorch
GitHub 中提及

基准测试

基准方法指标
3d-object-classification-on-modelnet10ECC (12 votes)
Accuracy: 90
3d-object-classification-on-modelnet40ECC (12 votes)
Classification Accuracy: 83.2
3d-point-cloud-classification-on-modelnet40ECC
Mean Accuracy: 83.2
Overall Accuracy: 87.4
3d-point-cloud-classification-on-sydney-urbanECC
F1: 78.4
graph-classification-on-ddECC (5 scores)
Accuracy: 74.1%
graph-classification-on-enzymesECC (5 scores)
Accuracy: 52.67%
graph-classification-on-mutagECC (5 scores)
Accuracy: 88.33%
graph-classification-on-nci1ECC (5 scores)
Accuracy: 83.8%
graph-classification-on-nci109ECC (5 scores)
Accuracy: 82.14

用 AI 构建 AI

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

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

Hyper Newsletters

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