通过删边增边增强图结构泛化能力,提升跨图节点分类效果
Graph Augmentation for Cross Graph Domain Generalization
- 低权重边删除去除噪声,保留关键结构信息
- 基于聚类的增边构建域不变结构,提升模型泛化性
- 适合跨图数据分布差异大的节点分类任务
跨图节点分类将一个图中的有标签节点用于帮助另一个图中无标签节点的分类,本质上是图神经网络的领域泛化问题,因不同图间结构差异显著。现有研究多聚焦于模型训练,而数据增强这一简单有效的领域泛化方法仍缺乏探索。本文提出一种新的图结构增强方法:对低权重边进行删除,以去除可能干扰GNN泛化能力的噪声边,促使模型捕捉不同结构下的本质不变特征;同时,基于聚类的增边策略根据同分布节点特征生成域不变结构。实验在跨分布引用网络数据集上验证,该方法在传统增强方法中达到领先性能。
原文摘要 · Abstract (English)
Cross-graph node classification, utilizing the abundant labeled nodes from one graph to help classify unlabeled nodes in another graph, can be viewed as a domain generalization problem of graph neural networks (GNNs) due to the structure shift commonly appearing among various graphs. Nevertheless, current endeavors for cross-graph node classification mainly focus on model training. Data augmentation approaches, a simple and easy-to-implement domain generalization technique, remain under-explored. In this paper, we develop a new graph structure augmentation for the crossgraph domain generalization problem. Specifically, low-weight edgedropping is applied to remove potential noise edges that may hinder the generalization ability of GNNs, stimulating the GNNs to capture the essential invariant information underlying different structures. Meanwhile, clustering-based edge-adding is proposed to generate invariant structures based on the node features from the same distribution. Consequently, with these augmentation techniques, the GNNs can maintain the domain invariant structure information that can improve the generalization ability. The experiments on out-ofdistribution citation network datasets verify our method achieves state-of-the-art performance among conventional augmentations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。