English

Mixture of Message Passing Experts with Routing Entropy Regularization for Node Classification

Machine Learning 2025-11-13 v2 Social and Information Networks

Abstract

Graph neural networks (GNNs) have achieved significant progress in graph-based learning tasks, yet their performance often deteriorates when facing heterophilous structures where connected nodes differ substantially in features and labels. To address this limitation, we propose GNNMoE, a novel entropy-driven mixture of message-passing experts framework that enables node-level adaptive representation learning. GNNMoE decomposes message passing into propagation and transformation operations and integrates them through multiple expert networks guided by a hybrid routing mechanism. And a routing entropy regularization dynamically adjusts soft weighting and soft top-kk routing, allowing GNNMoE to flexibly adapt to diverse neighborhood contexts. Extensive experiments on twelve benchmark datasets demonstrate that GNNMoE consistently outperforms SOTA node classification methods, while maintaining scalability and interpretability. This work provides a unified and principled approach for achieving fine-grained, personalized node representation learning.

Keywords

Cite

@article{arxiv.2502.08083,
  title  = {Mixture of Message Passing Experts with Routing Entropy Regularization for Node Classification},
  author = {Xuanze Chen and Jiajun Zhou and Yadong Li and Jinsong Chen and Shanqing Yu and Qi Xuan},
  journal= {arXiv preprint arXiv:2502.08083},
  year   = {2025}
}
R2 v1 2026-06-28T21:41:07.076Z