Out-of-Distribution Graph Models Merging
Not on the map. Why?
Abstract (arXiv)
This paper studies a novel problem of out-of-distribution graph models merging, which aims to construct a generalized model from multiple graph models pre-trained on different domains with distribution discrepancy. This problem is challenging because of the difficulty in learning domain-invariant knowledge implicitly in model parameters and consolidating expertise from potentially heterogeneous GNN backbones. In this work, we propose a graph generation strategy that instantiates the mixture distribution of multiple domains. Then, we merge and fine-tune the pre-trained graph models via a MoE module and a masking mechanism for generalized adaptation. Our framework is architecture-agnostic and can operate without any source/target domain data. Both theoretical analysis and experimental results demonstrate the effectiveness of our approach in addressing the model generalization problem.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 58 of its 69 references that are in the map. First posted on arXiv 4 Jun 2025.
Builds on
58 of 69 references.
References · 58
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2017Graph Attention Networkscited by 8,923
- 2018How Powerful are Graph Neural Networks?cited by 4,393
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,163
- 2019Invariant Risk Minimizationcited by 1,854
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
- 2022Editing Models with Task Arithmeticcited by 1,120
- 2016Paying More Attention to Attention: Improving the Performance of Convolutional Neural Networks via Attention Transfercited by 1,077
- 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 905
- 2021Merging Models with Fisher-Weighted Averagingcited by 621
- 2019Multi-scale Attributed Node Embeddingcited by 548
38 more
- 2019Dreaming to Distill: Data-free Knowledge Transfer via DeepInversioncited by 423
- 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 252
- 2020How Neural Networks Extrapolate: From Feedforward to Graph Neural Networkscited by 222
- 2021Invariance Principle Meets Information Bottleneck for Out-of-Distribution Generalizationcited by 220
- 2023ZipIt! Merging Models from Different Tasks without Trainingcited by 179
- 2021Graph Condensation for Graph Neural Networkscited by 150
- 2020Invariant Rationalizationcited by 131
- 2020Dynamic Model Pruning with Feedbackcited by 125
- 2022Fusing finetuned models for better pretrainingcited by 110
- 2021Causal Attention for Interpretable and Generalizable Graph Classificationcited by 108
- 2020Neural Networks Fail to Learn Periodic Functions and How to Fix Itcited by 100
- 2020Are Neural Nets Modular? Inspecting Functional Modularity Through Differentiable Weight Maskscited by 98
- 2022Out-Of-Distribution Generalization on Graphs: A Surveycited by 97
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalizationcited by 94
- 2020Masking as an Efficient Alternative to Finetuning for Pretrained Language Modelscited by 93
- 2024Mixture of A Million Expertscited by 72
- 2021Generalizing Graph Neural Networks on Out-Of-Distribution Graphscited by 71
- 2022A Survey on Deep Graph Generation: Methods and Applicationscited by 69
- 2023Graph Mixture of Experts: Learning on Large-Scale Graphs with Explicit Diversity Modelingcited by 64
- 2022Graph Condensation via Receptive Field Distribution Matchingcited by 60
- 2024A Survey on Model MoErging: Recycling and Routing Among Specialized Experts for Collaborative Learningcited by 52
- 2021Graph-Free Knowledge Distillation for Graph Neural Networkscited by 46
- 2023Does Invariant Graph Learning via Environment Augmentation Learn Invariance?cited by 42
- 2020How fine can fine-tuning be? Learning efficient language modelscited by 41
- 2023Graph Condensation for Inductive Node Representation Learningcited by 31
- 2023Joint Learning of Label and Environment Causal Independence for Graph Out-of-Distribution Generalizationcited by 29
- 2021Learngene: From Open-World to Your Learning Taskcited by 27
- 2023Semi-supervised Domain Adaptation in Graph Transfer Learningcited by 27
- 2023Environment-Aware Dynamic Graph Learning for Out-of-Distribution Generalizationcited by 26
- 2023Graph Invariant Learning with Subgraph Co-mixup for Out-Of-Distribution Generalizationcited by 18
- 2024Graph Learning under Distribution Shifts: A Comprehensive Survey on Domain Adaptation, Out-of-distribution, and Continual Learningcited by 18
- 2024A Survey of Deep Graph Learning under Distribution Shifts: from Graph Out-of-Distribution Generalization to Adaptationcited by 17
- 2022On Balancing Bias and Variance in Unsupervised Multi-Source-Free Domain Adaptationcited by 16
- 2024One Model for One Graph: A New Perspective for Pretraining with Cross-domain Graphscited by 12
- 2023Graph Structure and Feature Extrapolation for Out-of-Distribution Generalizationcited by 10
- 2023Graph Out-of-Distribution Generalization with Controllable Data Augmentationcited by 9
- 2024GALA: Graph Diffusion-based Alignment with Jigsaw for Source-free Domain Adaptationcited by 8
- 2025GCAL: Adapting Graph Models to Evolving Domain Shiftscited by 8
50 most cited of 58 All 58 (JSON)
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Graph Out-of-Distribution Generalization via Causal Intervention
- 2026Generalizing GNNs with Tokenized Mixture of Experts
- 2025Self-Adaptive Graph Mixture of Models
- 2025Training Diverse Graph Experts for Ensembles: A Systematic Empirical Study
- 2025GMoPE:A Prompt-Expert Mixture Framework for Graph Foundation Models
- 2026Training-Free Cross-Architecture Merging for Graph Neural Networks
- 2025Do Graph Diffusion Models Accurately Capture and Generate Substructure Distributions?
- 2023D4Explainer: In-Distribution GNN Explanations via Discrete Denoising Diffusion