Papers

SCARCE:无需标签的图分布外泛化框架

Implicit Feature Reweighting for Label-Free Graph Generalization

Why we picked it

一帮研究者做了个叫 SCARCE 的图神经网络框架,训练时完全不用类别标签,靠环境分区和对抗混淆就能扛住分布偏移,在 DrugOOD 上跑赢其它无标签方法。

arXiv 论文提出 SCARCE,一个不依赖类别标签的图表示学习框架,用于应对分布偏移下的图神经网络部署问题。该方法分两阶段:先用变分图自编码器进行图重建的预热训练,再通过对抗混淆机制让编码器输出无法被判别器区分环境的表示。论文给出理论分析,说明环境混淆会收缩表示空间中环境判别方向,等价于对环境相关特征施加更强的方向性惩罚。在合成数据集和 DrugOOD 基准上,SCARCE 在无标签方法中取得领先的 OOD 泛化成绩。

Original · arXiv cs.LG

Implicit Feature Reweighting for Label-Free Graph Generalization

Graph-level out-of-distribution (OOD) generalization is critical for deploying graph neural networks in real-world applications where distribution shifts are inevitable. Existing approaches typically require class labels during representation learning to decide which substructures are predictive, limiting their applicability. We propose SCARCE, a label-free representation-learning framework that yields distribution-shift-robust graph representations from an environment partition of the training graphs alone, whether that partition comes from dataset metadata or, when none is available, from clustering a self-supervised embedding. Our method operates in two stages: (1) a variational graph autoencoder is warm-started for graph reconstruction; and (2) an adversarial confusion mechanism forces the encoder to produce representations from which a discriminator cannot distinguish environments, while preserving graph-reconstruction fidelity. We provide theoretical analysis showing that environment confusion contracts environment-discriminative directions in the representation space, which, under regularized downstream classifiers, is equivalent to imposing a stronger directional penalty on environment-associated features. This feature-reweighting mechanism reduces reliance on spurious correlations without erasing information, offering a label-free, continuous relaxation of last-layer reweighting methods. Experiments on a controlled synthetic datasets and the DrugOOD datasets show that SCARCE achieves state-of-the-art OOD generalization among label-free methods, using no class labels during representation learning.