中文

SCL-GNN:通过虚假关联学习实现通用图神经网络

机器学习 2026-03-10 v1 人工智能

摘要

图神经网络(GNN)在众多任务中展现出显著的成功,但其泛化能力常受节点特征与图中标签之间虚假关联的制约。我们的分析表明,GNN倾向于利用训练数据中甚至不可靠的统计关联,即便这些关联在预测中并不可靠。为此,我们提出了一种名为Spurious Correlation Learning Graph Neural Network(虚假关联学习图神经网络,SCL-GNN)的新型框架,以提升其在独立同分布(IID)和超出分布(OOD)图上的泛化能力。SCL-GNN融合了原则化的虚假关联学习机制,利用希尔伯特-史迪姆独立性准则(Hilbert-Schmidt Independence Criterion, HSIC)量化节点表示与类别得分之间的关联,从而有效识别并消除不相关但具有影响力的虚假关联。此外,我们引入了高效的双层优化策略,以联合优化模块和GNN参数,防止过拟合。在真实数据集和合成数据集上的大量实验表明,SCL-GNN在各种分布迁移情景下均优于当前最先进的基线方法,凸显其鲁健性和泛化能力。

关键词

引用

@article{arxiv.2603.08270,
  title  = {SCL-GNN: Towards Generalizable Graph Neural Networks via Spurious Correlation Learning},
  author = {Yuxiang Zhang and Enyan Dai},
  journal= {arXiv preprint arXiv:2603.08270},
  year   = {2026}
}