Graph Structure and Feature Extrapolation for Out-of-Distribution Generalization
Not on the map. Why?
Abstract (arXiv)
Out-of-distribution (OOD) generalization deals with the prevalent learning scenario where test distribution shifts from training distribution. With rising application demands and inherent complexity, graph OOD problems call for specialized solutions. While data-centric methods exhibit performance enhancements on many generic machine learning tasks, there is a notable absence of data augmentation methods tailored for graph OOD generalization. In this work, we propose to achieve graph OOD generalization with the novel design of non-Euclidean-space linear extrapolation. The proposed augmentation strategy extrapolates both structure and feature spaces to generate OOD graph data. Our design tailors OOD samples for specific shifts without corrupting underlying causal mechanisms. Theoretical analysis and empirical results evidence the effectiveness of our method in solving target shifts, showing substantial and constant improvements across various graph OOD tasks.
Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). Builds on 69 of its 95 references that are in the map. First posted on arXiv 13 Jun 2023.
Builds on
69 of 95 references.
References · 69
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Graph Attention Networkscited by 8,854
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 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
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,148
- 2019Invariant Risk Minimizationcited by 1,832
- 2016Variational Graph Auto-Encoderscited by 1,373
- 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
38 more
- 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
- 2017Deeper, Broader and Artier Domain Generalizationcited by 983
- 2021Generalizing to Unseen Domains: A Survey on Domain Generalizationcited by 762
- 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 737
- 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
- 2019GraphSAINT: Graph Sampling Based Inductive Learning Methodcited by 618
- 2015Causal inference using invariant prediction: identification and confidence intervalscited by 533
- 2016Interaction Networks for Learning about Objects, Relations and Physicscited by 503
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 431
- 2020Explainability in Graph Neural Networks: A Taxonomic Surveycited by 383
- 2018Learning Models with Uniform Performance via Distributionally Robust Optimizationcited by 334
- 2019Analysing Mathematical Reasoning Abilities of Neural Modelscited by 331
- 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
- 2020Graph Random Neural Network for Semi-Supervised Learning on Graphscited by 213
- 2022Improving Out-of-Distribution Robustness via Selective Augmentationcited by 202
- 2022Discovering Invariant Rationales for Graph Neural Networkscited by 186
- 2021Large-Scale Representation Learning on Graphs via Bootstrappingcited by 184
- 2022Interpretable and Generalizable Graph Learning via Stochastic Attention Mechanismcited by 171
- 2019Understanding Attention and Generalization in Graph Neural Networkscited by 169
- 2018Measuring abstract reasoning in neural networkscited by 165
- 2022Handling Distribution Shifts on Graphs: An Invariance Perspectivecited by 163
- 2019What Can Neural Networks Reason About?cited by 155
- 2022G-Mixup: Graph Data Augmentation for Graph Classificationcited by 145
- 2022Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphscited by 144
- 2022GOOD: A Graph Out-of-Distribution Benchmarkcited by 140
- 2022Equivariant 3D-Conditional Diffusion Models for Molecular Linker Designcited by 121
- 2020From Local Structures to Size Generalization in Graph Neural Networkscited by 119
- 2019Neural Execution of Graph Algorithmscited by 118
- 2020Deep Visual Domain Adaptationcited by 91
- 2021Heterogeneous Risk Minimizationcited by 90
- 2021Shift-Robust GNNs: Overcoming the Limitations of Localized Graph Training Datacited by 86
50 most cited of 69 All 69 (JSON)
Built on by
Citing papers per year: 2023: 2; 2024: 5; 2025: 3.
10 papers.
- 2023Does Invariant Graph Learning via Environment Augmentation Learn Invariance?cited by 42
- 2023Deep Generative Models for Detector Signature Simulation: A Taxonomic Reviewcited by 18
- 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
- 2024How Interpretable Are Interpretable Graph Neural Networks?cited by 13
- 2024Dissecting the Failure of Invariant Learning on Graphscited by 5
- 2025Pruning Spurious Subgraphs for Graph Out-of-Distribution Generalizationcited by 3
- 2024Invariant Graph Learning Meets Information Bottleneck for Out-of-Distribution Generalizationcited by 2
- 2025Out-of-Distribution Generalization on Graphs via Progressive Inferencecited by 2
- 2025Out-of-Distribution Graph Models Mergingcited by 0
- 2025Pruning Spurious Subgraphs for Graph Out-of-Distribution Generalization6 Jun 2025
- 2025Out-of-Distribution Graph Models Merging4 Jun 2025
- 2025Out-of-Distribution Generalization on Graphs via Progressive Inference4 Mar 2025
- 2024Dissecting the Failure of Invariant Learning on Graphs5 Nov 2024
- 2024A Survey of Deep Graph Learning under Distribution Shifts: from Graph Out-of-Distribution Generalization to Adaptation25 Oct 2024
- 2024Invariant Graph Learning Meets Information Bottleneck for Out-of-Distribution Generalization3 Aug 2024
- 2024How Interpretable Are Interpretable Graph Neural Networks?12 Jun 2024
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency11 Jun 2024
- 2023Deep Generative Models for Detector Signature Simulation: A Taxonomic Review15 Dec 2023
- 2023Does Invariant Graph Learning via Environment Augmentation Learn Invariance?29 Oct 2023