How Wide and How Deep? Mitigating Over-squashing of GNNs via Channel Capacity Constrained Estimation

Read full text
Authors: Zinuo You, Jin Zheng, John Cartlidge
Cited by: 0

Abstract

Existing graph neural networks typically rely on heuristic choices for hidden dimensions and propagation depths, which often lead to severe information loss during propagation, known as over-squashing. To address this issue, we propose Channel Capacity Constrained Estimation (C³E), a novel framework that formulates the selection of hidden dimensions and depth as a nonlinear programming problem grounded in information theory. Through modeling spectral graph neural networks as communication channels, our approach directly connects channel capacity to hidden dimensions, propagation depth, propagation mechanism, and graph structure. Extensive experiments on nine public datasets demonstrate that hidden dimensions and depths estimated by C³E can mitigate over-squashing and consistently improve representation learning. Experimental results show that over-squashing occurs due to the cumulative compression of information in representation matrices. Furthermore, our findings show that increasing hidden dimensions indeed mitigates information compression, while the role of propagation depth is more nuanced, uncovering a fundamental balance between information compression and representation complexity.

Fields of Study & Subjects

Advanced Graph Neural Networks Ethics and Social Impacts of AI Explainable Artificial Intelligence (XAI)

Publication details

Journal / source
Proceedings of the AAAI Conference on Artificial Intelligence
Volume
40
Issue
33
Pages
27890–27898
Publication date
14 March 2026
Language
EN