Joint Learning of Label and Environment Causal Independence for Graph Out-of-Distribution Generalization
Not on the map. Why?
Abstract (arXiv)
We tackle the problem of graph out-of-distribution (OOD) generalization. Existing graph OOD algorithms either rely on restricted assumptions or fail to exploit environment information in training data. In this work, we propose to simultaneously incorporate label and environment causal independence (LECI) to fully make use of label and environment information, thereby addressing the challenges faced by prior methods on identifying causal and invariant subgraphs. We further develop an adversarial training strategy to jointly optimize these two properties for causal subgraph discovery with theoretical guarantees. Extensive experiments and analysis show that LECI significantly outperforms prior methods on both synthetic and real-world datasets, establishing LECI as a practical and effective solution for graph OOD generalization. Our code is available at https://github.com/divelab/LECI.
Key facts. Cited by 29 papers in the map (data through 7 Oct 2026). Builds on 96 of its 125 references that are in the map. First posted on arXiv 1 Jun 2023.
Builds on
96 of 125 references.
References · 96
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Graph Attention Networkscited by 8,854
- 2022Generative Adversarial Networkscited by 5,603
- 2018How Powerful are Graph Neural Networks?cited by 4,355
- 2017Neural Message Passing for Quantum Chemistrycited by 3,460
- 2015Domain-Adversarial Training of Neural Networkscited by 3,023
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,708
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
- 2014Unsupervised Domain Adaptation by Backpropagationcited by 1,906
- 2019Invariant Risk Minimizationcited by 1,832
38 more
- 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,366
- 2015Learning Transferable Features with Deep Adaptation Networkscited by 1,323
- 2017Adversarial Discriminative Domain Adaptationcited by 1,309
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,232
- 2020Graph Contrastive Learning with Augmentationscited by 1,203
- 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,194
- 2019A Comprehensive Survey on Transfer Learningcited by 1,154
- 2020In Search of Lost Domain Generalizationcited by 1,042
- 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,037
- 2017Deeper, Broader and Artier Domain Generalizationcited by 983
- 2020Learning to Simulate Complex Physics with Graph Networkscited by 867
- 2015Deep Learning and the Information Bottleneck Principlecited by 793
- 2021Generalizing to Unseen Domains: A Survey on Domain Generalizationcited by 762
- 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 737
- 2019GNNExplainer: Generating Explanations for Graph Neural Networkscited by 735
- 2019DropEdge: Towards Deep Graph Convolutional Networks on Node Classificationcited by 679
- 2018Deep Visual Domain Adaptation: A Surveycited by 634
- 2013Domain Generalization via Invariant Feature Representationcited by 622
- 2018Learning to Adapt Structured Output Space for Semantic Segmentationcited by 584
- 2015Causal inference using invariant prediction: identification and confidence intervalscited by 533
- 2016Interaction Networks for Learning about Objects, Relations and Physicscited by 503
- 2019Graph U-Netscited by 489
- 2020Principal Neighbourhood Aggregation for Graph Netscited by 477
- 2016Geometric deep learning on graphs and manifolds using mixture model CNNscited by 468
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 431
- 2019Contrastive Adaptation Network for Unsupervised Domain Adaptationcited by 407
- 2020Explainability in Graph Neural Networks: A Taxonomic Surveycited by 383
- 2018A Survey of Unsupervised Deep Domain Adaptationcited by 361
- 2018Learning Models with Uniform Performance via Distributionally Robust Optimizationcited by 334
- 2019Analysing Mathematical Reasoning Abilities of Neural Modelscited by 331
- 2020Environment Inference for Invariant Learningcited by 306
- 2021On Explainability of Graph Neural Networks via Subgraph Explorationscited by 280
- 2020The Risks of Invariant Risk Minimizationcited by 262
- 2018Graph networks as learnable physics engines for inference and controlcited by 242
- 2021Invariance Principle Meets Information Bottleneck for Out-of-Distribution Generalizationcited by 220
- 2020How Neural Networks Extrapolate: From Feedforward to Graph Neural Networkscited by 218
- 2020XGNN: Towards Model-Level Explanations of Graph Neural Networkscited by 211
- 2020GraphLIME: Local Interpretable Model Explanations for Graph Neural Networkscited by 209
50 most cited of 96 All 96 (JSON)
Built on by
Citing papers per year: 2023: 4; 2024: 13; 2025: 9; 2026: 3.
29 papers.
Sort:
By area:
- 2024A Survey of Graph Neural Networks in Real world: Imbalance, Noise, Privacy and OOD Challengescited by 54
- 2023Does Invariant Graph Learning via Environment Augmentation Learn Invariance?cited by 42
- 2024Pairwise Alignment Improves Graph Domain Adaptationcited by 25
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiencycited by 18
- 2024A Survey of Deep Graph Learning under Distribution Shifts: from Graph Out-of-Distribution Generalization to Adaptationcited by 16
- 2024Reconsidering Faithfulness in Regular, Self-Explainable and Domain Invariant GNNscited by 14
- 2024How Interpretable Are Interpretable Graph Neural Networks?cited by 13
- 2024Is the MMI Criterion Necessary for Interpretability? Degenerating Non-causal Features to Plain Noise for Self-Rationalizationcited by 11
- 2023Graph Structure and Feature Extrapolation for Out-of-Distribution Generalizationcited by 10
- 2023Identifying Semantic Component for Robust Molecular Property Predictioncited by 10
- 2025Beyond Topological Self-Explainable GNNs: A Formal Explainability Perspectivecited by 10
- 2024Out-of-Distribution Learning with Human Feedbackcited by 6
17 more
- 2025Breaking Free from MMI: A New Frontier in Rationalization by Probing Input Utilizationcited by 6
- 2025Adversarial Cooperative Rationalization: The Risk of Spurious Correlations in Even Clean Datasetscited by 6
- 2024Spatio-Temporal Fluid Dynamics Modeling via Physical-Awareness and Parameter Diffusion Guidancecited by 5
- 2023When Graph Neural Network Meets Causality: Opportunities, Methodologies and An Outlookcited by 4
- 2024A Survey of Out-of-distribution Generalization for Graph Machine Learning from a Causal Viewcited by 3
- 2024Virtual Nodes Can Help: Tackling Distribution Shifts in Federated Graph Learningcited by 3
- 2025Pruning Spurious Subgraphs for Graph Out-of-Distribution Generalizationcited by 3
- 2025Measure-Theoretic Anti-Causal Representation Learningcited by 3
- 2024Invariant Graph Learning Meets Information Bottleneck for Out-of-Distribution Generalizationcited by 2
- 2025Quantifying Distributional Invariance in Causal Subgraph for IRM-Free Graph Generalizationcited by 2
- 2025Evolving Graph Learning for Out-of-Distribution Generalization in Non-stationary Environmentscited by 2
- 2026SCL-GNN: Towards Generalizable Graph Neural Networks via Spurious Correlation Learningcited by 1
- 2024Improving Graph Out-of-distribution Generalization Beyond Causalitycited by 0
- 2025Catch Causal Signals from Edges for Label Imbalance in Graph Classificationcited by 0
- 2025Out-of-Distribution Graph Models Mergingcited by 0
- 2026Rethinking Graph Generalization through the Lens of Sharpness-Aware Minimizationcited by 0
- 2026Structure-agnostic Causal Representation Learningcited by 0
- 2026Structure-agnostic Causal Representation Learning1 Oct 2026
- 2026SCL-GNN: Towards Generalizable Graph Neural Networks via Spurious Correlation Learning9 Mar 2026
- 2026Rethinking Graph Generalization through the Lens of Sharpness-Aware Minimization9 Feb 2026
- 2025Evolving Graph Learning for Out-of-Distribution Generalization in Non-stationary Environments4 Nov 2025
- 2025Quantifying Distributional Invariance in Causal Subgraph for IRM-Free Graph Generalization23 Oct 2025
- 2025Measure-Theoretic Anti-Causal Representation Learning16 Oct 2025
- 2025Pruning Spurious Subgraphs for Graph Out-of-Distribution Generalization6 Jun 2025
- 2025Out-of-Distribution Graph Models Merging4 Jun 2025
- 2025Adversarial Cooperative Rationalization: The Risk of Spurious Correlations in Even Clean Datasets4 May 2025
- 2025Breaking Free from MMI: A New Frontier in Rationalization by Probing Input Utilization8 Mar 2025
- 2025Beyond Topological Self-Explainable GNNs: A Formal Explainability Perspective4 Feb 2025
- 2025Catch Causal Signals from Edges for Label Imbalance in Graph Classification3 Jan 2025
17 more
- 2024Virtual Nodes Can Help: Tackling Distribution Shifts in Federated Graph Learning26 Dec 2024
- 2024A Survey of Deep Graph Learning under Distribution Shifts: from Graph Out-of-Distribution Generalization to Adaptation25 Oct 2024
- 2024Is the MMI Criterion Necessary for Interpretability? Degenerating Non-causal Features to Plain Noise for Self-Rationalization8 Oct 2024
- 2024A Survey of Out-of-distribution Generalization for Graph Machine Learning from a Causal View15 Sep 2024
- 2024Out-of-Distribution Learning with Human Feedback14 Aug 2024
- 2024Invariant Graph Learning Meets Information Bottleneck for Out-of-Distribution Generalization3 Aug 2024
- 2024Improving Graph Out-of-distribution Generalization Beyond Causality14 Jul 2024
- 2024Reconsidering Faithfulness in Regular, Self-Explainable and Domain Invariant GNNs21 Jun 2024
- 2024How Interpretable Are Interpretable Graph Neural Networks?12 Jun 2024
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency11 Jun 2024
- 2024Spatio-Temporal Fluid Dynamics Modeling via Physical-Awareness and Parameter Diffusion Guidance18 Mar 2024
- 2024A Survey of Graph Neural Networks in Real world: Imbalance, Noise, Privacy and OOD Challenges7 Mar 2024
- 2024Pairwise Alignment Improves Graph Domain Adaptation2 Mar 2024
- 2023When Graph Neural Network Meets Causality: Opportunities, Methodologies and An Outlook19 Dec 2023
- 2023Identifying Semantic Component for Robust Molecular Property Prediction8 Nov 2023
- 2023Does Invariant Graph Learning via Environment Augmentation Learn Invariance?29 Oct 2023
- 2023Graph Structure and Feature Extrapolation for Out-of-Distribution Generalization13 Jun 2023
- 2024Out-of-Distribution Learning with Human Feedbackcited by 6
- 2024Out-of-Distribution Learning with Human Feedback14 Aug 2024
- 2024A Survey of Graph Neural Networks in Real world: Imbalance, Noise, Privacy and OOD Challengescited by 54
- 2023Does Invariant Graph Learning via Environment Augmentation Learn Invariance?cited by 42
- 2024Pairwise Alignment Improves Graph Domain Adaptationcited by 25
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiencycited by 18
- 2024A Survey of Deep Graph Learning under Distribution Shifts: from Graph Out-of-Distribution Generalization to Adaptationcited by 16
- 2024Reconsidering Faithfulness in Regular, Self-Explainable and Domain Invariant GNNscited by 14
- 2024How Interpretable Are Interpretable Graph Neural Networks?cited by 13
- 2024Is the MMI Criterion Necessary for Interpretability? Degenerating Non-causal Features to Plain Noise for Self-Rationalizationcited by 11
- 2023Graph Structure and Feature Extrapolation for Out-of-Distribution Generalizationcited by 10
- 2023Identifying Semantic Component for Robust Molecular Property Predictioncited by 10
- 2025Beyond Topological Self-Explainable GNNs: A Formal Explainability Perspectivecited by 10
- 2025Breaking Free from MMI: A New Frontier in Rationalization by Probing Input Utilizationcited by 6
12 most cited of 28 All 28 (JSON)
- 2026Structure-agnostic Causal Representation Learning1 Oct 2026
- 2026SCL-GNN: Towards Generalizable Graph Neural Networks via Spurious Correlation Learning9 Mar 2026
- 2026Rethinking Graph Generalization through the Lens of Sharpness-Aware Minimization9 Feb 2026
- 2025Evolving Graph Learning for Out-of-Distribution Generalization in Non-stationary Environments4 Nov 2025
- 2025Quantifying Distributional Invariance in Causal Subgraph for IRM-Free Graph Generalization23 Oct 2025
- 2025Measure-Theoretic Anti-Causal Representation Learning16 Oct 2025
- 2025Pruning Spurious Subgraphs for Graph Out-of-Distribution Generalization6 Jun 2025
- 2025Out-of-Distribution Graph Models Merging4 Jun 2025
- 2025Adversarial Cooperative Rationalization: The Risk of Spurious Correlations in Even Clean Datasets4 May 2025
- 2025Breaking Free from MMI: A New Frontier in Rationalization by Probing Input Utilization8 Mar 2025
- 2025Beyond Topological Self-Explainable GNNs: A Formal Explainability Perspective4 Feb 2025
- 2025Catch Causal Signals from Edges for Label Imbalance in Graph Classification3 Jan 2025
12 newest of 28 All 28 (JSON)