arXiv:2603.08270cs.LGcs.AI2026-03被引 2

提出新框架SCL-GNN,提升图神经网络在分布外数据上的泛化能力。

SCL-GNN: Towards Generalizable Graph Neural Networks via Spurious Correlation Learning

  • 利用HSIC度量节点表示与类别得分的相关性,识别虚假关联
  • 在真实与合成数据上均显著优于现有方法,尤其在分布外场景
  • 适合需要强泛化能力的图学习任务,如跨域推荐、异常检测

图神经网络(GNN)在多种任务中表现优异,但其泛化能力常受节点特征与标签间虚假相关性的制约。分析表明,GNN会利用训练数据中难以察觉的统计相关性,即使这些相关性对预测不可靠。为此,我们提出新型框架SCL-GNN,旨在提升在独立同分布(IID)和分布外(OOD)图上的泛化性能。SCL-GNN引入基于希尔伯特-施密特独立性准则(HSIC)的伪相关学习机制,量化节点表示与类别得分之间的相关性,从而有效识别并削弱无关但有影响力的虚假关联。同时,设计一种高效的双层优化策略,联合优化模块与GNN参数,防止过拟合。在真实世界与合成数据集上的大量实验表明,SCL-GNN在各种分布偏移下持续优于当前最优基线,展现出卓越的鲁棒性与泛化能力。

原文摘要 · Abstract (English)

Graph Neural Networks (GNNs) have demonstrated remarkable success across diverse tasks. However, their generalization capability is often hindered by spurious correlations between node features and labels in the graph. Our analysis reveals that GNNs tend to exploit imperceptible statistical correlations in training data, even when such correlations are unreliable for prediction. To address this challenge, we propose the Spurious Correlation Learning Graph Neural Network (SCL-GNN), a novel framework designed to enhance generalization on both Independent and Identically Distributed (IID) and Out-of-Distribution (OOD) graphs. SCL-GNN incorporates a principled spurious correlation learning mechanism, leveraging the Hilbert-Schmidt Independence Criterion (HSIC) to quantify correlations between node representations and class scores. This enables the model to identify and mitigate irrelevant but influential spurious correlations effectively. Additionally, we introduce an efficient bi-level optimization strategy to jointly optimize modules and GNN parameters, preventing overfitting. Extensive experiments on real-world and synthetic datasets demonstrate that SCL-GNN consistently outperforms state-of-the-art baselines under various distribution shifts, highlighting its robustness and generalization capabilities.

图神经网络泛化能力虚假相关分布外

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