提升图数据分布与标签一致性,增强模型跨域泛化能力
Enhancing Distribution and Label Consistency for Graph Out-of-Distribution Generalization
- 统一生成增强图与不变子图,避免传统两步法的不一致问题
- 在保持标签-图关系完整性的前提下扩充训练数据
- 适用于需要稳定跨域性能的图学习任务,如社交网络分析
针对图数据中的分布偏移问题,现有图外分布(OOD)泛化方法多采用两步策略:先构建增强环境,再提取不变子图以提升泛化能力。然而该方法在一致性方面存在不足:一方面,通过修改图结构保留标签的增强过程可能产生不真实或与原始分布无关的图,导致分布不一致;另一方面,直接修改图得到的子图未必保持与标签的一致预测关系,影响标签一致性。为此,本文提出一种新方法,统一生成增强图与不变图,既丰富训练数据又不破坏标签-图关系的完整性。框架进一步强化不变图中的标签一致性,保留监督信息。在多个真实世界数据集上的大量实验表明,本方法显著优于当前主流基线。
原文摘要 · Abstract (English)
To deal with distribution shifts in graph data, various graph out-of-distribution (OOD) generalization techniques have been recently proposed. These methods often employ a two-step strategy that first creates augmented environments and subsequently identifies invariant subgraphs to improve generalizability. Nevertheless, this approach could be suboptimal from the perspective of consistency. First, the process of augmenting environments by altering the graphs while preserving labels may lead to graphs that are not realistic or meaningfully related to the origin distribution, thus lacking distribution consistency. Second, the extracted subgraphs are obtained from directly modifying graphs, and may not necessarily maintain a consistent predictive relationship with their labels, thereby impacting label consistency. In response to these challenges, we introduce an innovative approach that aims to enhance these two types of consistency for graph OOD generalization. We propose a modifier to obtain both augmented and invariant graphs in a unified manner. With the augmented graphs, we enrich the training data without compromising the integrity of label-graph relationships. The label consistency enhancement in our framework further preserves the supervision information in the invariant graph. We conduct extensive experiments on real-world datasets to demonstrate the superiority of our framework over other state-of-the-art baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。