Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

AnyGraph: Graph Foundation Model in the Wild

About

The growing ubiquity of relational data structured as graphs has underscored the need for graph learning models with exceptional generalization capabilities. However, current approaches often struggle to effectively extract generalizable insights, frequently requiring extensive fine-tuning and limiting their versatility. Graph foundation models offer a transformative solution, with the potential to learn robust, generalizable representations from graph data. This enables more effective and adaptable applications across a wide spectrum of tasks and domains. In this work, we investigate a unified graph model, AnyGraph, designed to handle key challenges: i) Structure Heterogenity. Addressing distribution shift in graph structural information; ii) Feature Heterogenity. Handling diverse feature representation spaces across graph datasets; iii) Fast Adaptation. Efficiently adapting the model to new graph domains; iv) Scaling Law Emergence. Enabling the model to exhibit scaling law behavior, where its performance scales favorably with the amount of data and parameter sizes. To tackle these critical challenges, we build the AnyGraph upon a Graph Mixture-of-Experts (MoE) architecture. This approach empowers the model to effectively manage both the in-domain and cross-domain distribution shift concerning structure-level and feature-level heterogeneity. Furthermore, a lightweight graph expert routing mechanism is proposed to facilitate AnyGraph's fast adaptability to new data and domains. Our extensive experiments on diverse 38 graph datasets have demonstrated the strong zero-shot learning performance of AnyGraph across diverse graph domains with significant distribution shift. Furthermore, we have validated the model's fast adaptation ability and scaling law emergence, showcasing its versatility.

Lianghao Xia, Chao Huang• 2024

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy72.23
1252
Graph ClassificationMUTAG
Accuracy75.61
1103
Node ClassificationPubmed
Accuracy70.73
627
Node ClassificationCora
Accuracy62.6
583
Node Classificationogbn-arxiv (test)
Accuracy62.33
497
Graph ClassificationCOLLAB
Accuracy35.53
469
Node ClassificationCora (test)
Accuracy68.4
122
Node ClassificationAmazon Photo (test)
Accuracy32.42
112
Node ClassificationCiteseer
Accuracy19.32
51
Link PredictionarXiv--
29
Showing 10 of 27 rows

Other info

Follow for update