arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2502.09263cs.LG
收藏

经典GNN能成为图级任务的强基线吗?简单架构展现卓越性能

Can Classic GNNs Be Strong Baselines for Graph-level Tasks? Simple Architectures Meet Excellence

Beihang University(北洋大学);The Hong Kong Polytechnic University(香港理工大学)

Yuankai Luo, Lei Shi, Xiao-Ming Wu

AI总结:

本研究提出GNN+框架,通过集成六种常用技术增强经典GNN,在14个图级数据集上系统评估表明,经典GNN性能匹配或超越GTs且效率更高,挑战了复杂机制对图级性能必不可少的观念。

摘要:

消息传递图神经网络(GNNs)常因表达能力有限、过平滑和过挤压等问题,以及捕获长程依赖的挑战而受到批评。相反,图Transformer(GTs)因其采用全局注意力机制,有望缓解这些挑战,被认为是更优的选择。文献常指出GTs在图级任务中优于GNNs,尤其是在小分子图的分类和回归上。本研究通过增强框架GNN+探索了GNNs的未开发潜力,该框架集成了六种广泛使用的技术:边特征整合、归一化、dropout、残差连接、前馈网络和位置编码,以有效处理图级任务。我们在14个知名图级数据集上,对由GNN+框架增强的三种经典GNN(GCN、GIN和GatedGCN)进行了系统性的重新评估。结果表明,与普遍看法相反,这些经典GNN始终匹配或超越GTs的性能,在所有数据集中均获前三名,并在八个数据集中夺冠。此外,它们效率更高,在许多数据集上运行速度比GTs快数倍。这凸显了简单GNN架构的潜力,挑战了GTs中复杂机制对于卓越图级性能必不可少的观念。源代码可在https://github.com/LUOyk1999/GNNPlus获取。

英文摘要:

Message-passing Graph Neural Networks (GNNs) are often criticized for their limited expressiveness, issues like over-smoothing and over-squashing, and challenges in capturing long-range dependencies. Conversely, Graph Transformers (GTs) are regarded as superior due to their employment of global attention mechanisms, which potentially mitigate these challenges. Literature frequently suggests that GTs outperform GNNs in graph-level tasks, especially for graph classification and regression on small molecular graphs. In this study, we explore the untapped potential of GNNs through an enhanced framework, GNN+, which integrates six widely used techniques: edge feature integration, normalization, dropout, residual connections, feed-forward networks, and positional encoding, to effectively tackle graph-level tasks. We conduct a systematic re-evaluation of three classic GNNs (GCN, GIN, and GatedGCN) enhanced by the GNN+ framework across 14 well-known graph-level datasets. Our results reveal that, contrary to prevailing beliefs, these classic GNNs consistently match or surpass the performance of GTs, securing top-three rankings across all datasets and achieving first place in eight. Furthermore, they demonstrate greater efficiency, running several times faster than GTs on many datasets. This highlights the potential of simple GNN architectures, challenging the notion that complex mechanisms in GTs are essential for superior graph-level performance. Our source code is available at https://github.com/LUOyk1999/GNNPlus.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑