mlsanityData to 7 Oct 2026

Improving Graph Out-of-distribution Generalization Beyond Causality

arXiv 2407.10204 ↗First posted 14 Jul 2024cs.LGcited by 0

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

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  2. 2015Domain-Adversarial Training of Neural Networkscited by 3,023
  3. 2014Unsupervised Domain Adaptation by Backpropagationcited by 1,906
  4. 2019Invariant Risk Minimizationcited by 1,832
  5. 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,194
  6. 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,037
  7. 2021Generalizing to Unseen Domains: A Survey on Domain Generalizationcited by 762
  8. 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 737
  9. 2015Causal inference using invariant prediction: identification and confidence intervalscited by 533
  10. 2020Explainability in Graph Neural Networks: A Taxonomic Surveycited by 383
  11. 2020Environment Inference for Invariant Learningcited by 306
  12. 2017Advances in Variational Inferencecited by 236
19 more
  1. 2022Discovering Invariant Rationales for Graph Neural Networkscited by 186
  2. 2022Interpretable and Generalizable Graph Learning via Stochastic Attention Mechanismcited by 171
  3. 2022Correct-N-Contrast: A Contrastive Approach for Improving Robustness to Spurious Correlationscited by 171
  4. 2022Learning on Large-scale Text-attributed Graphs via Variational Inferencecited by 153
  5. 2022Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphscited by 144
  6. 2022GOOD: A Graph Out-of-Distribution Benchmarkcited by 140
  7. 2021Generalizing Graph Neural Networks on Out-Of-Distribution Graphscited by 69
  8. 2019Generalized Variational Inference: Three arguments for deriving new Posteriorscited by 57
  9. 2023Does Invariant Graph Learning via Environment Augmentation Learn Invariance?cited by 42
  10. 2022Unleashing the Power of Graph Data Augmentation on Covariate Distribution Shiftcited by 38
  11. 2023Mind the Label Shift of Augmentation-based Graph OOD Generalizationcited by 34
  12. 2024Graph Out-of-Distribution Generalization via Causal Interventioncited by 30
  13. 2023Joint Learning of Label and Environment Causal Independence for Graph Out-of-Distribution Generalizationcited by 29
  14. 2023Environment-Aware Dynamic Graph Learning for Out-of-Distribution Generalizationcited by 26
  15. 2023Learning Invariant Molecular Representation in Latent Discrete Spacecited by 19
  16. 2023Graph Invariant Learning with Subgraph Co-mixup for Out-Of-Distribution Generalizationcited by 18
  17. 2023Individual and Structural Graph Information Bottlenecks for Out-of-Distribution Generalizationcited by 15
  18. 2024Empowering Graph Invariance Learning with Deep Spurious Infomaxcited by 14
  19. 2023MARIO: Model Agnostic Recipe for Improving OOD Generalization of Graph Contrastive Learningcited by 7

Built on by

Not cited yet.