English

Training-Free Cross-Architecture Merging for Graph Neural Networks

Machine Learning 2026-02-24 v1

Abstract

Model merging has emerged as a powerful paradigm for combining the capabilities of distinct expert models without the high computational cost of retraining, yet current methods are fundamentally constrained to homogeneous architectures. For GNNs, however, message passing is topology-dependent and sensitive to misalignment, making direct parameter-space merging unreliable. To bridge this gap, we introduce H-GRAMA (Heterogeneous Graph Routing and Message Alignment), a training-free framework that lifts merging from parameter space to operator space. We formalize Universal Message Passing Mixture (UMPM), a shared operator family that expresses heterogeneous GNN layers in a common functional language. H-GRAMA enables cross-architecture GNN merging (e.g., GCN to GAT) without retraining, retaining high specialist accuracy in most cases in compatible depth settings and achieving inference speedups of 1.2x to 1.9x over ensembles.

Keywords

Cite

@article{arxiv.2602.19332,
  title  = {Training-Free Cross-Architecture Merging for Graph Neural Networks},
  author = {Rishabh Bhattacharya and Vikaskumar Kalsariya and Naresh Manwani},
  journal= {arXiv preprint arXiv:2602.19332},
  year   = {2026}
}
R2 v1 2026-07-01T10:46:33.532Z