acceptodds
Under review as a conference paper at ICLR 2027

GNNM: Parameter-Efficient BatchEnsembles for Graph Neural Networks

Abstract

Ensembles are an effective way to improve the predictive performance of neural networks, but their computational and memory costs limit their use in many settings. Modern parameter-efficient ensembles address this issue by sharing the main model weights across ensemble members and introducing only a small number of additional parameters, yet their effectiveness in message passing architectures remains unclear. We introduce GNNM, a lightweight ensemble modification for residual GNNs that replaces only the input and output linear projectors with BatchEnsemble parameterisations while leaving the graph backbone unchanged. This design allows the same modification to be applied to different GNN models with minimal parameter overhead and without altering their message passing layers. We evaluate GNNM on five standard node classification benchmarks and compare it with both individual model baselines and explicit ensemble baselines under the same protocol for hyperparameter selection. The results show that a GNNM variant achieves the best score on every dataset. On average, GNNM improves over the individual model for every backbone and performs on par with or better than an explicit ensemble of the same size while using about four times fewer parameters, while its training cost is comparable to that of the explicit ensemble. The gains are strongest for SAGE, GAT-sep, GT, and GT-sep, while for GAT the explicit ensemble remains stronger. These results show that projector level ensembling can serve as a simple and parameter-efficient mechanism for improving residual GNNs while preserving the structure of the underlying graph model.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.