arXiv:2412.08144cs.LGcs.AI2024-12中稿 · AAAI被引 8

自适应图混叠提升节点分类性能,更好保留图结构。

AGMixup: Adaptive Graph Mixup for Semi-supervised Node Classification

  • 以子图为单位进行混合,更贴近图像域处理方式。
  • 根据上下文相似性和不确定性动态调整混合比例λ。
  • 在7个数据集上优于现有方法,适合图神经网络研究者。

Mixup是一种通过在图像域中插值数据点来增强模型泛化能力的数据增强技术,其混合比例为$λ$。近年来,该思想被扩展到图域,采用基于节点的插值方法。然而,这些方法常无法应对复杂互连关系,可能破坏图的自然拓扑并削弱节点间交互。此外,现有图混叠方法对所有混合对采用统一随机采样的$λ$,忽略了不同配对的差异化需求。本文提出自适应图混叠(AGMixup)框架,用于半监督节点分类。AGMixup引入子图中心策略,将每个子图视为类似欧几里得域中的图像,从而更自然地融入图学习。我们还提出一种自适应机制,依据子图的上下文相似性和不确定性动态调节混合比例$λ$。在七个半监督节点分类基准数据集上的大量实验表明,AGMixup显著优于当前最先进的图混叠方法。源代码已公开于 exttt{https://github.com/WeigangLu/AGMixup}。

原文摘要 · Abstract (English)

Mixup is a data augmentation technique that enhances model generalization by interpolating between data points using a mixing ratio $λ$ in the image domain. Recently, the concept of mixup has been adapted to the graph domain through node-centric interpolations. However, these approaches often fail to address the complexity of interconnected relationships, potentially damaging the graph's natural topology and undermining node interactions. Furthermore, current graph mixup methods employ a one-size-fits-all strategy with a randomly sampled $λ$ for all mixup pairs, ignoring the diverse needs of different pairs. This paper proposes an Adaptive Graph Mixup (AGMixup) framework for semi-supervised node classification. AGMixup introduces a subgraph-centric approach, which treats each subgraph similarly to how images are handled in Euclidean domains, thus facilitating a more natural integration of mixup into graph-based learning. We also propose an adaptive mechanism to tune the mixing ratio $λ$ for diverse mixup pairs, guided by the contextual similarity and uncertainty of the involved subgraphs. Extensive experiments across seven datasets on semi-supervised node classification benchmarks demonstrate AGMixup's superiority over state-of-the-art graph mixup methods. Source codes are available at \url{https://github.com/WeigangLu/AGMixup}.

图神经网络数据增强半监督学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。