LLM2D
在图OOD泛化的高标准:超越显式环境建模的不变性学习
Raising the Bar in Graph OOD Generalization: Invariant Learning Beyond Explicit Environment Modeling
作者: Xu Shen, Yixin Liu, Yili Wang, Rui Miao, Yiwei Dai, Shirui Pan, Xin Wang
发布日期: 2/18/2025
arXiv ID: oai:arXiv.org:2502.10706v1

摘要

arXiv:2502.10706v1 类型: cross 摘要:异常分布(OOD)泛化在图学习中已成为一个重要挑战,因为现实世界的图数据常常表现出传统模型无法泛化的多样化且不断变化的环境。解决这一问题的一个有前途的解决方案是图不变量学习(GIL),其旨在通过拆分标签相关的不变子图和环境特定的子图来学习不变表示。然而,现有的GIL方法面临两个主要挑战:(1)捕获和建模图数据中多样化的环境的难度,以及(2)语义悬崖,不同类别中的不变子图难以区分,导致类别可分性差和增加误分类。为应对这些挑战,我们提出了一种名为多原型超球面不变学习(MPHIL)的新方法,该方法引入了两个关键创新:(1)超球面不变表示提取,能够实现鲁棒且高度区分的超球面不变特征提取,以及(2)多原型超球面分类,采用类原型作为中间变量消除GIL中显式环境建模的需要并缓解语义悬崖问题。基于GIL的理论框架,我们引入了两个新的目标函数:不变原型匹配损失,确保样本与正确的类原型匹配,以及原型分离损失,在超球面上增加不同类原型之间的区分度。在来自不同领域和分布转移的各种11个OOD泛化基准数据集上的广泛实验表明,MPHIL实现了最先进的性能,显著优于现有方法,无论是在各种图数据上还是在不同分布转移中。