Graph Structure and Feature Extrapolation for Out-of-Distribution Generalization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, Xiner, Gui, Shurui, Luo, Youzhi, Ji, Shuiwang
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929373751279616
author Li, Xiner
Gui, Shurui
Luo, Youzhi
Ji, Shuiwang
author_facet Li, Xiner
Gui, Shurui
Luo, Youzhi
Ji, Shuiwang
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2306_08076
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Graph Structure and Feature Extrapolation for Out-of-Distribution Generalization
Li, Xiner
Gui, Shurui
Luo, Youzhi
Ji, Shuiwang
Machine Learning
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.
title Graph Structure and Feature Extrapolation for Out-of-Distribution Generalization
topic Machine Learning
url https://arxiv.org/abs/2306.08076