arXiv · 2503.21579
Fusion of Graph Neural Networks via Optimal Transport
Abstract
In this paper, we explore the idea of combining GCNs into one model. To that end, we align the weights of different models layer-wise using optimal transport (OT). We present and evaluate three types of transportation costs and show that the studied fusion method consistently outperforms the performance of vanilla averaging. Finally, we present results suggesting that model fusion using OT is harder in the case of GCNs than MLPs and that incorporating the graph structure into the process does not improve the performance of the method.
Explore related subjects
Keep this discovery
Weronika Ormaniec, Michael Vollenweider, Elisa Hoskovec. 2025-03-27. Fusion of Graph Neural Networks via Optimal Transport. https://arxiv.org/abs/2503.21579
Cite the original work for its findings. Save a collection to share your selection of sources.