摘要
arXiv:2502.10706v2 通知类型: replace-cross
摘要:离分布(out-of-distribution, OOD)泛化已成为图学习中的一个关键挑战,因为现实世界中的图数据往往表现出传统模型难以跨环境泛化的多样化和变化的环境。解决这一问题的一个有前景的解决方案是图不变学习(graph invariant learning, GIL),其目标是通过分离标签相关不变子图和环境特定子图来学习不变表示。然而,现有的GIL方法面临两大主要挑战:(1) 图数据中多样化和变化的环境捕捉和建模的难度,以及(2) 语义悬崖,不同类别的不变子图难以区分,导致类别区分能力差且增加错误分类。为应对这些挑战,我们提出了一种名为多原型超球面不变学习(multi-prototype hyperspherical invariant learning, MPHIL)的新方法,该方法引入了两个关键创新点:(1) 超球面不变表示提取,使稳健且高度区分性的超球面不变特征提取成为可能,以及(2) 多原型超球面分类,它使用类别原型作为中间变量,以消除GIL中的显式环境建模需求并缓解语义悬崖问题。基于GIL的理论框架,我们介绍了两个新的目标函数:不变原型匹配损失,以确保样本被匹配到正确的类别原型,以及原型分离损失,以增加不同类别的原型在超球面空间的区分程度。对11个OOD泛化基准数据集的广泛实验表明,MPHIL达到了最先进的性能,显著优于现有方法,涵盖了来自不同领域的图数据和不同分布变化。