arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2308.08097cs.LGcs.AI

S-Mixup:面向图神经网络的结构化 Mixup

S-Mixup: Structural Mixup for Graph Neural Networks

KAIST(韩国科学技术院)

Junghurn Kim, Sukwon Yun, Chanyoung Park

AI总结:

本文提出 S-Mixup,利用伪标签、置信度及边梯度为节点分类构建结构化 mixup 增强,显著提升 GNN 在异配场景下的鲁棒性与泛化能力。

摘要:

现有将 mixup 技术应用于图的研究主要集中在图分类任务,而节点分类方面的研究仍不充分。本文提出一种用于节点分类的新型 mixup 增强方法,称为 Structural Mixup(S-Mixup)。其核心思想是在混合节点时考虑结构信息。具体而言,S-Mixup 通过图神经网络(GNN)分类器获取图中未标记节点的伪标签及其预测置信度。这些信息作为构建 mixup 池的标准,用于类间和类内 mixup。此外,我们利用 GNN 训练过程中得到的边梯度,提出一种基于梯度的边选择策略,用于选择要连接到 mixup 所生成节点上的边。通过在真实世界基准数据集上的大量实验,我们证明了 S-Mixup 在节点分类任务上的有效性。我们观察到,S-Mixup 提升了 GNN 的鲁棒性和泛化性能,尤其是在异配情形下。S-Mixup 的源代码可在 https://github.com/SukwonYun/S-Mixup 获取。

英文摘要:

Existing studies for applying the mixup technique on graphs mainly focus on graph classification tasks, while the research in node classification is still under-explored. In this paper, we propose a novel mixup augmentation for node classification called Structural Mixup (S-Mixup). The core idea is to take into account the structural information while mixing nodes. Specifically, S-Mixup obtains pseudo-labels for unlabeled nodes in a graph along with their prediction confidence via a Graph Neural Network (GNN) classifier. These serve as the criteria for the composition of the mixup pool for both inter and intra-class mixups. Furthermore, we utilize the edge gradient obtained from the GNN training and propose a gradient-based edge selection strategy for selecting edges to be attached to the nodes generated by the mixup. Through extensive experiments on real-world benchmark datasets, we demonstrate the effectiveness of S-Mixup evaluated on the node classification task. We observe that S-Mixup enhances the robustness and generalization performance of GNNs, especially in heterophilous situations. The source code of S-Mixup can be found at \url{https://github.com/SukwonYun/S-Mixup}

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

↑