Escaping Oversquashing: Addressable and Support-Aware Global Memory for Message Passing Networks
cs.LG, cs.AI
Submitted: 2026-08-03
Updated: 2026-10-02
Comments: preliminary work
License: http://creativecommons.org/licenses/by/4.0/
The gist: Virtual nodes are a natural tool against oversquashing: they replace long message-passing paths by a two-hop global route.
Terminology
Abstract
Virtual nodes are a natural tool against oversquashing: they replace long message-passing paths by a two-hop global route. But when many nodes share one global state, that shortcut can become a bottleneck itself. We study two properties of this global memory. First, addressability: under constant-margin address codes and a nonlinearity that amplifies this margin, multiplicative write/read maps provide M selectable memory rows with only O(M) address-code dimensions. Cross-attention slots and a constrained ELU+1 bilinear memory both satisfy these conditions. Second, support awareness: normalized cross-attention has no self-key for a latent query to use as a reference. A learned private anchor supplies this reference, keeps the read bounded, and exposes the strength of the matching source mass. We demonstrate the merits of such properties on several instances of Two-Radius and Tree-NeighborsMatch: both addressable realizations solve the controlled tasks through depth 5, where pooled VNs of comparable or larger size reach about 10.6%.
Sources
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks