Explaining and Adapting Graph Conditional Shift

06/05/2023
by   Qi Zhu, et al.
0

Graph Neural Networks (GNNs) have shown remarkable performance on graph-structured data. However, recent empirical studies suggest that GNNs are very susceptible to distribution shift. There is still significant ambiguity about why graph-based models seem more vulnerable to these shifts. In this work we provide a thorough theoretical analysis on it by quantifying the magnitude of conditional shift between the input features and the output label. Our findings show that both graph heterophily and model architecture exacerbate conditional shifts, leading to performance degradation. To address this, we propose an approach that involves estimating and minimizing the conditional shift for unsupervised domain adaptation on graphs. In our controlled synthetic experiments, our algorithm demonstrates robustness towards distribution shift, resulting in up to 10 algorithm. Furthermore, comprehensive experiments on both node classification and graph classification show its robust performance under various distribution shifts.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
05/24/2023

Size Generalizability of Graph Neural Networks on Biological Data: Insights and Practices from the Spectral Perspective

We investigate the question of whether the knowledge learned by graph ne...
research
07/28/2021

Adversarial Unsupervised Domain Adaptation with Conditional and Label Shift: Infer, Align and Iterate

In this work, we propose an adversarial unsupervised domain adaptation (...
research
10/04/2022

Tree Mover's Distance: Bridging Graph Metrics and Stability of Graph Neural Networks

Understanding generalization and robustness of machine learning models f...
research
10/19/2022

Towards Explaining Distribution Shifts

A distribution shift can have fundamental consequences such as signaling...
research
08/18/2023

Distribution shift mitigation at test time with performance guarantees

Due to inappropriate sample selection and limited training data, a distr...
research
02/06/2023

Energy-based Out-of-Distribution Detection for Graph Neural Networks

Learning on graphs, where instance nodes are inter-connected, has become...
research
08/06/2022

Generalizability Analysis of Graph-based Trajectory Predictor with Vectorized Representation

Trajectory prediction is one of the essential tasks for autonomous vehic...

Please sign up or login with your details

Forgot password? Click here to reset