Skip to content
Book Open access

One For All: Achieving Adaptive Graph Neural Networks via Mixture of Message Passing

Aug 2026 · Proceedings of the 32nd ACM SIGKDD Conference on Knowledge Discovery and Data Mining V.2 · pp. 3516-3526 · 0 citations · 25 references

TL;DR

This work proposes Mixture of Message Passing (MoMP), a novel mechanism that incorporates the Mixture of Experts (MoE) paradigm directly into the message passing mechanism of GNNs, treating different message passing mechanisms as ''experts''.

Abstract

The message passing mechanism, which updates node representations by exchanging messages with their neighbors, plays a critical role in graph neural networks (GNNs) for capturing structural patterns. Since the single message passing mechanism lacks the flexibility to handle graphs with differing node feature quality, GNNs with a fixed message passing mechanism, e.g., GCN, GAT, and GraphSAGE, exhibit inconsistent performance across different graphs. To address this issue, we propose Mixture of Message Passing (MoMP). Code is made publicly available at https://github.com/Tinmomo/MoMP, a novel mechanism that incorporates the Mixture of Experts (MoE) paradigm directly into the message passing mechanism of GNNs, treating different message passing mechanisms as ''experts''. Specifically, MoMP employs a gating network to perform edge-level routing, adaptively selecting the most suitable expert to process the message for each edge. To ensure training stability and efficacy, we also introduce a partially-shared expert architecture, reframing the learning task for sparse experts into a residual from a globally-learned general mechanism. Extensive experiments on several homophilic graph benchmarks demonstrate that the proposed method not only outperforms strong baselines but also provides significant interpretability. To the best of our knowledge, our work is the first to fuse the MoE paradigm directly with the GNN message passing mechanism, paving the way for developing adaptive GNNs.

Read PDF

Similar papers

#graph neural networks Preprint Aug 2026

CoRe-GNN: Multilevel Message passing on Coarsened graphs

CoRe-GNN is proposed, which performs both propagations in parallel at each layer: a coarsened inter-cluster term capturing long-range structure, and a local intra-cluster term preserving per-node discriminability.

Antonin Joly, Nicolas Keriven, Aline Roumy · 0 citations
Preprint Aug 2026

Rethinking Message Passing as Retrieval for Text-Attributed Graph Learning

Graph neural networks (GNNs) are typically conceptualized as message-passing neural networks, yet it remains unclear why neighborhood aggregation reliably outperforms node-wise multilayer perceptrons (MLPs). Despite its empirical success, this paradigm can be computationally expensive and sensitive to imperfect graph structures. In this work, we present a retrieval-augmented view of GNNs: each layer makes predictions by applying an MLP to a node representation together with a permutation-invariant summary of retrieved graph context. Motivated by this perspective, we propose RTA, a simple MLP-based framework that replaces structural message passing with label-aware retrieval and propagation. We provide theoretical insights that (i) connect retrieval-based aggregation to softmax-attention message passing, and (ii) establish the robustness of retrieved-context supervision to mis-retrieved outliers. Experiments on multiple text-attributed graph benchmarks show that RTA matches or even outperforms strong GNN and graph LLM baselines while improving efficiency and robustness across diverse scenarios.

Jintang Li, Yuhong Chen, Ruo-Fan Wu et al. · 0 citations
Open access Jan 2025

DeltaGNN: Graph Neural Network with Information Flow Control

Graph Neural Networks (GNNs) are popular deep learning models designed to process graph-structured data through recursive neighborhood aggregations in the message passing process. When applied to semi-supervised node classification, the message-passing enables GNNs to understand short-range spatial interactions, but also causes them to suffer from over-smoothing and over-squashing. These challenges hinder model expressiveness and prevent the use of deeper models to capture long-range node interactions (LRIs) within the graph. Popular solutions for LRIs detection are either too expensive to process large graphs due to high time complexity or fail to generalize across diverse graph structures. To address these limitations, we propose a mechanism called information flow control, which leverages a novel connectivity measure, called information flow score, to address over-smoothing and over-squashing with linear computational overhead, supported by theoretical evidence. Building on this mechanism, we introduce DeltaGNN, to the best of our knowledge among the first scalable (featuring linear computational and memory complexity overhead) and generalizable (capable of effectively handling graphs with diverse homophily, density, and topology) architectures for long-range and short-range interaction detection. We benchmark our model across 10 real-world datasets, including graphs with varying sizes, topologies, densities, and homophilic ratios, showing superior performance with limited computational complexity.

Kevin Mancini, Islem Rekik · 2 citations
Preprint Jul 2026

CondPSE: A Polynomial-Filtered Structural Encoder with Conditional Modulation for Graphs

Message-passing graph neural networks are bounded by the 1-WL test and can miss topological structure that distinguishes non-isomorphic graphs. Positional and structural encodings (PSE) inject such topology-derived signals, and learned PSE encoders such as GPSE pretrain a single encoder to produce these signals from random node probes, which can then be frozen and reused as inputs across downstream graph models. We present CondPSE, a learned PSE encoder that applies a learnable polynomial graph filter bank to standard Gaussian node probes and refines the resulting structural-response branches through FiLM-style modulation conditioned on cross-filter, local message-passing, and graph-level signals. CondPSE is pretrained to reconstruct node-level positional/structural targets and graph-level invariants, and is then frozen for use as a downstream input encoding. On synthetic structural-discrimination benchmarks, CondPSE separates graph structures that 1-WL-bounded message passing cannot: it raises CSL accuracy from 42.9% to 97.3% and EXP accuracy from 68.3% to 99.9% relative to GPSE, and ablations show that the polynomial filter bank accounts for most of this gain. On real molecular property prediction, the picture is more limited. With a hybrid local-message-passing/global-attention backbone, CondPSE performs comparably to GPSE without surpassing it, and a ZINC backbone sweep shows no consistent ordering between the two encoders. We report these results and discuss why strong synthetic structural discrimination does not, on its own, yield a downstream advantage for frozen learned PSE encoders, including the role of downstream integration and possible mismatch between structural pretraining targets and molecular property labels.

Woohyung Lee, Hogun Park · 0 citations

ADC-GNN: Adaptive Dual-level Collaborative Graph Neural Networks for Graph Classification

This work proposes a novel Adaptive Dual-level Collaborative GNN associated with an adaptive dual-level collaborative mechanism, and shows that the ADC-GNN can inject the learned high-level information back into the node level, forming a closed-loop, bidirectional optimization process.

Wan Tang, Lu Bai, Lixin Cui et al. · 0 citations
Preprint Jul 2026

Examining the Efficacy of Graph Neural Network Message-Passing in Regression Contexts

Graph Neural Networks (GNN) facilitate effective prediction on graph data such as molecules, media networks and neural network blueprints. GNNs facilitate prediction through message passing techniques which define how information flows from a node to its neighbors. Due to the ubiquity of the graph data type, the development of newer and better GNNs has garnered much interest in the machine learning community. However, GNN evaluation and benchmarking is primarily driven by classification tasks. Thus, prospective GNN message passing layers are evaluated on their ability to outperform prior work in classification contexts. In contrast, GNNs are equally capable of performing scalar regression prediction, yet this class of problem is often overlooked when proposing new GNNs while the best classification GNNs are utilized in an a priori or off-the-shelf manner for regression problems. In response, this paper studies the efficacy of GNN layers in a slew of regression contexts from rank ordering, error minimization and insight extraction. Results show that deep convolutional GNNs, particularly GEN, are more effective at these tasks than attention-based GNNs, while other classical, theoretically-inspired GNNs remain competitive and efficient.

Keith G. Mills, Aedan J. DeFrates, Joong Ho Kim · 0 citations