mlsanityData to 14 Oct 2026

One Model for One Graph: A New Perspective for Pretraining with Cross-domain Graphs

arXiv 2412.00315 ↗First posted 30 Nov 2024cs.LG cs.AI stat.MLcited by 12

Not on the map. Why?

Abstract (arXiv)

Graph Neural Networks (GNNs) have emerged as a powerful tool to capture intricate network patterns, achieving success across different domains. However, existing GNNs require careful domain-specific architecture designs and training from scratch on each dataset, leading to an expertise-intensive process with difficulty in generalizing across graphs from different domains. Therefore, it can be hard for practitioners to infer which GNN model can generalize well to graphs from their domains. To address this challenge, we propose a novel cross-domain pretraining framework, "one model for one graph," which overcomes the limitations of previous approaches that failed to use a single GNN to capture diverse graph patterns across domains with significant gaps. Specifically, we pretrain a bank of expert models, with each one corresponding to a specific dataset. When inferring to a new graph, gating functions choose a subset of experts to effectively integrate prior model knowledge while avoiding negative transfer. Extensive experiments consistently demonstrate the superiority of our proposed method on both link prediction and node classification tasks.

Key facts. Cited by 12 papers in the map (data through 14 Oct 2026). Builds on 34 of its 48 references that are in the map. First posted on arXiv 30 Nov 2024.

Builds on

34 of 48 references.

References · 34

  1. 2017Attention Is All You Needcited by 60,361
  2. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  3. 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
  4. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
  5. 2019Simplifying Graph Convolutional Networkscited by 1,477
  6. 2018Graph Convolutional Neural Networks for Web-Scale Recommender Systemscited by 1,138
  7. 2016Revisiting Semi-Supervised Learning with Graph Embeddingscited by 976
  8. 2019Graph Neural Networks for Social Recommendationcited by 714
  9. 2020Deep Graph Contrastive Representation Learningcited by 539
  10. 2020GCC: Graph Contrastive Coding for Graph Neural Network Pre-Trainingcited by 463
  11. 2022GraphMAE: Self-Supervised Masked Graph Autoencoderscited by 448
  12. 2023Exploring the Potential of Large Language Models (LLMs) in Learning on Graphscited by 274
22 more
  1. 2023One for All: Towards Training One Graph Model for All Classification Taskscited by 243
  2. 2023GraphGPT: Graph Instruction Tuning for Large Language Modelscited by 226
  3. 2024LLaGA: Large Language and Graph Assistantcited by 181
  4. 2023GraphPrompt: Unifying Pre-Training and Downstream Tasks for Graph Neural Networkscited by 167
  5. 2023All in One: Multi-task Prompting for Graph Neural Networkscited by 148
  6. 2021An Empirical Study of Graph Contrastive Learningcited by 113
  7. 2023PRODIGY: Enabling In-context Learning Over Graphscited by 98
  8. 2024All in One and One for All: A Simple yet Effective Method towards Cross-domain Graph Pretrainingcited by 86
  9. 2024ZeroG: Investigating Cross-dataset Zero-shot Transferability in Graphscited by 85
  10. 2023Graph Mixture of Experts: Learning on Large-Scale Graphs with Explicit Diversity Modelingcited by 64
  11. 2024AnyGraph: Graph Foundation Model in the Wildcited by 53
  12. 2021Automated Self-Supervised Learning for Graphscited by 50
  13. 2024Text-space Graph Foundation Models: Comprehensive Benchmarks and New Insightscited by 44
  14. 2023Demystifying Structural Disparity in Graph Neural Networks: Can One Size Fit All?cited by 41
  15. 2022Multi-task Self-supervised Graph Neural Networks Enable Stronger Task Generalizationcited by 35
  16. 2024LiGNN: Graph Neural Networks at LinkedIncited by 31
  17. 2020Knowing your FATE: Friendship, Action and Temporal Explanations for User Engagement Prediction on Social Appscited by 23
  18. 2024Node-wise Filtering in Graph Neural Networks: A Mixture of Experts Approachcited by 17
  19. 2021Graph Classification by Mixture of Diverse Expertscited by 16
  20. 2024GraphAlign: Pretraining One Graph Neural Network on Multiple Graphs via Feature Alignmentcited by 15
  21. 2023Better with Less: A Data-Active Perspective on Pre-Training Graph Neural Networkscited by 14
  22. 2023Mixture of Weak & Strong Experts on Graphscited by 7

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2024: 1; 2025: 7; 2026: 4.

12 papers.