提出可变尺度图神经网络,提升节点分类在异质图上的性能。
ScaleNet: Scale Invariance Learning in Directed Graphs
- 用有向多尺度边替代传统无向单边,构建可变尺度的中心图
- 在7个数据集上5个达最优,异质图表现显著优于现有方法
- 为异质图节点分类提供统一框架,适合复杂关系建模任务
图神经网络在关系数据分析中取得进展,但缺乏图像分类中常见的尺度不变性学习。在节点分类任务中,实际分类的是中心节点的自图。本文借鉴图像处理中的尺度不变思想,提出“缩放自图”概念:将传统无向单边替换为有序的多有向边构成的“缩放边”,实现对多尺度结构的建模。我们在七个基准数据集(包含同质与异质结构)上评估该方法,结果表明基于尺度不变性的图学习在准确率、速度和简洁性上均优于基于随机游走的Inception模型。该方法解释了Inception模型在同质图上成功但在异质图上受限的原因。为进一步提升在异质图上的适用性,我们提出ScaleNet架构,利用多尺度特征进行学习。ScaleNet在七数据集中有五个达到当前最优,另两个持平,展现出卓越泛化能力。该研究为图学习提供了统一框架,显著提升了跨类型图的节点分类性能。代码已开源。
原文摘要 · Abstract (English)
Graph Neural Networks (GNNs) have advanced relational data analysis but lack invariance learning techniques common in image classification. In node classification with GNNs, it is actually the ego-graph of the center node that is classified. This research extends the scale invariance concept to node classification by drawing an analogy to image processing: just as scale invariance being used in image classification to capture multi-scale features, we propose the concept of ``scaled ego-graphs''. Scaled ego-graphs generalize traditional ego-graphs by replacing undirected single-edges with ``scaled-edges'', which are ordered sequences of multiple directed edges. We empirically assess the performance of the proposed scale invariance in graphs on seven benchmark datasets, across both homophilic and heterophilic structures. Our scale-invariance-based graph learning outperforms inception models derived from random walks by being simpler, faster, and more accurate. The scale invariance explains inception models' success on homophilic graphs and limitations on heterophilic graphs. To ensure applicability of inception model to heterophilic graphs as well, we further present ScaleNet, an architecture that leverages multi-scaled features. ScaleNet achieves state-of-the-art results on five out of seven datasets (four homophilic and one heterophilic) and matches top performance on the remaining two, demonstrating its excellent applicability. This represents a significant advance in graph learning, offering a unified framework that enhances node classification across various graph types. Our code is available at https://github.com/Qin87/ScaleNet/tree/July25.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。