Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
The paper introduces ShareGNNs, a message-passing neural network architecture that incorporates graph structure information into weight sharing by indexing weights with user-chosen graph invariants. This allows the model to systematically reuse weights across structurally equivalent subgraphs, enhancing its ability to capture structural patterns beyond feature-based learning. Experiments demonstrate that ShareGNNs achieve improved performance over standard MPNNs on various tasks, including subgraph counting and real-world graph learning, while also offering explicit control over model complexity.
By indexing message passing weights with graph invariants, ShareGNNs unlock a new level of structure-awareness, outperforming standard MPNNs and even surpassing the 1-WL test in expressivity.
Message-passing neural networks (MPNNs) are a powerful framework for learning representations of graph-structured domains. However, weights in MPNNs act on features only, limiting their ability to capture structural patterns. We introduce a novel structure-aware weight sharing principle that explicitly incorporates information inherent to the graph structure. Weights are indexed directly by user-chosen graph invariants, i.e., functions preserved under node permutations, enabling systematic reuse across structurally equivalent subgraphs. We present ShareGNNs, which instantiate this principle within a simple encoder-decoder architecture, resulting in an MPNN with learnable adjacency and transformer-like connectivity. We show that their expressivity is at least as strong as the discriminative power of the chosen invariants, providing explicit control over the model complexity. Experiments on synthetic and real-world data, as well as subgraph counting tasks, demonstrate consistent improvements over standard MPNNs, competitive expressivity beyond the 1-WL test, and scalability to large datasets.