Improving Graph Out-of-distribution Generalization Beyond Causality
Not on the map. Why?
Abstract (arXiv)
Existing methods for graph out-of-distribution (OOD) generalization primarily rely on empirical studies on synthetic datasets. Such approaches tend to overemphasize the causal relationships between invariant sub-graphs and labels, thereby neglecting the non-negligible role of environment in real-world scenarios. In contrast to previous studies that impose rigid independence assumptions on environments and invariant sub-graphs, this paper presents the theorems of environment-label dependency and mutable rationale invariance, where the former characterizes the usefulness of environments in determining graph labels while the latter refers to the mutable importance of graph rationales. Based on analytic investigations, a novel variational inference based method named ``Probability Dependency on Environments and Rationales for OOD Graphs on Real-world Data'' (DEROG) is introduced. To alleviate the adverse effect of unknown prior knowledge on environments and rationales, DEROG utilizes generalized Bayesian inference. Further, DEROG employs an EM-based algorithm for optimization. Finally, extensive experiments on real-world datasets under different distribution shifts are conducted to show the superiority of DEROG. Our code is publicly available at https://github.com/LEOXC1571/DEROG.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 31 of its 41 references that are in the map. First posted on arXiv 14 Jul 2024.
Builds on
31 of 41 references.
References · 31
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2015Domain-Adversarial Training of Neural Networkscited by 3,023
- 2014Unsupervised Domain Adaptation by Backpropagationcited by 1,906
- 2019Invariant Risk Minimizationcited by 1,832
- 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,194
- 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,037
- 2021Generalizing to Unseen Domains: A Survey on Domain Generalizationcited by 762
- 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 737
- 2015Causal inference using invariant prediction: identification and confidence intervalscited by 533
- 2020Explainability in Graph Neural Networks: A Taxonomic Surveycited by 383
- 2020Environment Inference for Invariant Learningcited by 306
- 2017Advances in Variational Inferencecited by 236
19 more
- 2022Discovering Invariant Rationales for Graph Neural Networkscited by 186
- 2022Interpretable and Generalizable Graph Learning via Stochastic Attention Mechanismcited by 171
- 2022Correct-N-Contrast: A Contrastive Approach for Improving Robustness to Spurious Correlationscited by 171
- 2022Learning on Large-scale Text-attributed Graphs via Variational Inferencecited by 153
- 2022Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphscited by 144
- 2022GOOD: A Graph Out-of-Distribution Benchmarkcited by 140
- 2021Generalizing Graph Neural Networks on Out-Of-Distribution Graphscited by 69
- 2019Generalized Variational Inference: Three arguments for deriving new Posteriorscited by 57
- 2023Does Invariant Graph Learning via Environment Augmentation Learn Invariance?cited by 42
- 2022Unleashing the Power of Graph Data Augmentation on Covariate Distribution Shiftcited by 38
- 2023Mind the Label Shift of Augmentation-based Graph OOD Generalizationcited by 34
- 2024Graph Out-of-Distribution Generalization via Causal Interventioncited by 30
- 2023Joint Learning of Label and Environment Causal Independence for Graph Out-of-Distribution Generalizationcited by 29
- 2023Environment-Aware Dynamic Graph Learning for Out-of-Distribution Generalizationcited by 26
- 2023Learning Invariant Molecular Representation in Latent Discrete Spacecited by 19
- 2023Graph Invariant Learning with Subgraph Co-mixup for Out-Of-Distribution Generalizationcited by 18
- 2023Individual and Structural Graph Information Bottlenecks for Out-of-Distribution Generalizationcited by 15
- 2024Empowering Graph Invariance Learning with Deep Spurious Infomaxcited by 14
- 2023MARIO: Model Agnostic Recipe for Improving OOD Generalization of Graph Contrastive Learningcited by 7
Built on by
Not cited yet.