atlas

Research map / Graph learning / Graph self-supervised learning

Graph Structure Extrapolation for Out-of-Distribution Generalization

ICML 2024 · Graph learning › Graph self-supervised learning · approach: Graph out-of-distribution generalization

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes G-Splice and FeatX, environment-aware non-Euclidean linear extrapolation data augmentations that generate OOD graphs by splicing graph structures and extrapolating non-causal node features, for graph OOD generalization.

Problem

Graph OOD generalization under structure shifts (size, scaffold/base) and feature shifts; existing graph augmentations mostly interpolate or only produce subgraphs, so they cannot reach OOD regions.

Approach

Defines graph addition (splicing) and subtraction (largest common subgraph removal) to form structural linear extrapolation, uses environment- and label-aware pair learning to extract causal/environmental subgraphs, and generates connecting bridges with a cVAE; feature extrapolation perturbs selected non-causal node features with modulo wrapping into the training domain.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
apply G-Splice instead of the strongest listed baselinegraph classification models · GOOD-HIV sizeROC-AUC: 62.38\textpm2.65 -> 64.46\textpm1.38Helped
apply G-Splice instead of the strongest listed baselinegraph classification models · GOOD-HIV scaffoldROC-AUC: 71.94\textpm1.31 -> 72.82\textpm1.16Helped
apply G-Splice instead of the strongest listed baselinegraph classification models · GOOD-SST2 lengthaccuracy: 82.00\textpm0.73 -> 82.31\textpm0.59Helped
apply G-Splice instead of the strongest listed baselinegraph classification models · Twitter lengthaccuracy: 57.67\textpm1.11 -> 58.02\textpm0.40Helped
apply G-Splice instead of the strongest listed baselinegraph classification models · GOOD-Motif sizeaccuracy: 54.14\textpm3.11 -> 86.53\textpm2.66Helped
apply G-Splice instead of the strongest listed baselinegraph classification models · GOOD-Motif baseaccuracy: 75.58\textpm0.75 -> 79.86\textpm13.00Helped
apply G-Splice instead of the strongest listed baselinegraph classification models · DD sizeMCC: 0.42\textpm0.03 -> 0.45\textpm0.09Helped
apply G-Splice instead of the strongest listed baselinegraph classification models · NCI1 sizeMCC: 0.30\textpm0.05 -> 0.37\textpm0.08Helped
add invariant regularization to G-Splice (ablation)G-Splice · GOOD-HIV sizeROC-AUC: 64.46\textpm1.38 -> 65.56\textpm0.34Helped
add invariant regularization to G-Splice (ablation)G-Splice · GOOD-Motif sizeaccuracy: 86.53\textpm2.66 -> 85.07\textpm4.50Hurt
add invariant regularization to G-Splice (ablation)G-Splice · GOOD-Motif baseaccuracy: 79.86\textpm13.00 -> 83.96\textpm7.38Helped
combine FeatX with G-Splice (ablation)G-Splice · FSMotif size-coloraccuracy: 72.11\textpm6.18 -> 75.11\textpm 2.99Helped

Surprising findings

Limitations

Experimental setup

Datasets: GOOD-HIV GOOD-SST2 Twitter GOOD-Motif DD NCI1 FSMotif

Benchmarks: GOOD-HIV GOOD-SST2 Twitter GOOD-Motif DD NCI1 FSMotif

Metrics: ROC-AUC Matthews correlation coefficient accuracy

Baselines: ERM IRM VREx GroupDRO DANN DIR CIGA GIL LiSA DropNode DropEdge Graph Mixup

Builds on: Mixup invariant risk minimization DIR CIGA LiSA

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy