4 个月前

强大的图神经网络是否必要?对图分类的剖析

强大的图神经网络是否必要?对图分类的剖析

摘要

图神经网络(GNNs)近年来受到了越来越多的关注,部分原因是它们在许多节点和图分类任务中表现出色。然而,对于这些网络所学习的内容以及所学图函数的复杂程度,目前还缺乏深入的理解。在这项工作中,我们提出将图分类中的GNNs分解为两个部分:1)图滤波,其中进行基于图的邻居聚合;2)集合函数,其中将一组隐藏节点特征组合用于预测。为了研究这两个部分的重要性,我们分别对它们进行了线性化处理。首先,我们将图滤波函数线性化,得到了图特征网络(GFN),这是一种定义在经过图增强特征集上的简单轻量级神经网络。进一步线性化GFN的集合函数,则得到了图线性网络(GLN),即一个线性函数。我们在常见的图分类基准数据集上进行了实证评估。令人惊讶的是,尽管进行了简化处理,GFN仍能匹配或超过最近提出的GNNs的最佳准确率(计算成本仅为后者的一小部分),而GLN的表现则显著逊色。我们的结果表明了非线性集合函数的重要性,并建议使用线性图滤波与非线性集合函数相结合的方法来建模现有的图分类基准数据集是一种高效且强大的方案。

代码仓库

chentingpc/gfn
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
graph-classification-on-collabGFN-light
Accuracy: 81.34%
graph-classification-on-collabGFN
Accuracy: 81.50%
graph-classification-on-ddGFN-light
Accuracy: 78.62%
graph-classification-on-ddGFN
Accuracy: 78.78%
graph-classification-on-enzymesGFN-light
Accuracy: 69.50%
graph-classification-on-enzymesGFN
Accuracy: 70.17%
graph-classification-on-imdb-bGFN
Accuracy: 73.00%
graph-classification-on-imdb-bGFN-light
Accuracy: 73.00%
graph-classification-on-imdb-mGFN-light
Accuracy: 51.20%
graph-classification-on-imdb-mGFN
Accuracy: 51.80%
graph-classification-on-mutagGFN
Accuracy: 90.84%
graph-classification-on-mutagGFN-light
Accuracy: 89.89%
graph-classification-on-nci1GFN
Accuracy: 83.65%
graph-classification-on-nci1GFN-light
Accuracy: 81.43%
graph-classification-on-proteinsGFN
Accuracy: 76.46%
graph-classification-on-proteinsGFN-light
Accuracy: 77.44%
graph-classification-on-re-m12kGFN-light
Accuracy: 49.75%
graph-classification-on-re-m12kGFN
Accuracy: 49.43%
graph-classification-on-re-m5kGFN-light
Accuracy: 49.75%
graph-classification-on-re-m5kGFN
Accuracy: 49.43%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
强大的图神经网络是否必要?对图分类的剖析 | 论文 | HyperAI超神经