mHC-GNN: Manifold-Constrained Hyper-Connections for Graph Neural Networks
Subhankar Mishra
TL;DR
mHC-GNN introduces manifold-constrained hyper-connections to graph neural networks by expanding node representations into multiple streams and constraining stream mixing to the Birkhoff polytope via Sinkhorn normalization. This architectural shift yields exponentially slower over-smoothing with a rate of $(1-\gamma)^{L/n}$ and enhances expressiveness beyond the 1-WL limit, enabling deep networks up to 128 layers while maintaining substantial accuracy. The approach is architecture-agnostic, improving a range of base GNNs across diverse datasets, with ablations confirming the necessity of the manifold constraint and demonstrating robust training stability. Practically, mHC-GNN offers a scalable, general solution to deep graph learning, producing meaningful gains on large graphs like ogbn-arxiv and enabling deeper models that better capture long-range dependencies.
Abstract
Graph Neural Networks (GNNs) suffer from over-smoothing in deep architectures and expressiveness bounded by the 1-Weisfeiler-Leman (1-WL) test. We adapt Manifold-Constrained Hyper-Connections (\mhc)~\citep{xie2025mhc}, recently proposed for Transformers, to graph neural networks. Our method, mHC-GNN, expands node representations across $n$ parallel streams and constrains stream-mixing matrices to the Birkhoff polytope via Sinkhorn-Knopp normalization. We prove that mHC-GNN exhibits exponentially slower over-smoothing (rate $(1-γ)^{L/n}$ vs.\ $(1-γ)^L$) and can distinguish graphs beyond 1-WL. Experiments on 10 datasets with 4 GNN architectures show consistent improvements. Depth experiments from 2 to 128 layers reveal that standard GNNs collapse to near-random performance beyond 16 layers, while mHC-GNN maintains over 74\% accuracy even at 128 layers, with improvements exceeding 50 percentage points at extreme depths. Ablations confirm that the manifold constraint is essential: removing it causes up to 82\% performance degradation. Code is available at \href{https://github.com/smlab-niser/mhc-gnn}{https://github.com/smlab-niser/mhc-gnn}
