GNN-VPA: A Variance-Preserving Aggregation Strategy for Graph Neural Networks
Abstract
Graph neural networks (GNNs), and especially message-passing neural networks, excel in various domains such as physics, drug discovery, and molecular modeling. The expressivity of GNNs with respect to their ability to discriminate non-isomorphic graphs critically depends on the functions employed for message aggregation and graph-level readout. By applying signal propagation theory, we propose a variance-preserving aggregation function (VPA) that maintains expressivity, but yields improved forward and backward dynamics. Experiments demonstrate that VPA leads to increased predictive performance for popular GNN architectures as well as improved learning dynamics. Our results could pave the way towards normalizer-free or self-normalizing GNNs.
Keywords
Cite
@article{arxiv.2403.04747,
title = {GNN-VPA: A Variance-Preserving Aggregation Strategy for Graph Neural Networks},
author = {Lisa Schneckenreiter and Richard Freinschlag and Florian Sestak and Johannes Brandstetter and Günter Klambauer and Andreas Mayr},
journal= {arXiv preprint arXiv:2403.04747},
year = {2024}
}
Comments
Accepted at ICLR 2024 (Tiny Papers Track)