Expressive Graph Neural Networks via Equivariant Use of Noise
Xiyuan Wang, Muhan Zhang
Abstract
Expressivity has been a major focus in the design of Graph Neural Networks (GNNs), yet a significant gap persists between theoretical universal expressivity and practical performance. While many expressive GNNs are efficient and achieve strong results, they often focus on specific graph properties and lack theoretical expressivity for general graph tasks. Conversely, theoretically universal-expressive models often suffer from high computational costs or poor generalization, limiting their real-world applicability. To bridge this gap, we introduce Equivariant Noise GNNs (ENGNNs), a framework that utilizes random noise features to enhance the expressivity of GNNs. Crucially, unlike prior methods that naively use noise, we enforce equivariance to nodewise noise transformations, such as orthogonal transformations. We prove that this property reduces the model's theoretical sample complexity, thereby improving generalization. Our framework simultaneously reaches theoretical universal expressivity, maintains the linear scalability of standard Message-Passing Neural Networks in practice, and achieves performance comparable to computationally expensive, high-expressivity models. Extensive experiments confirm strong performance across node, link, subgraph, and graph-level prediction tasks, demonstrating that the equivariant use of noise provides a powerful and practical pathway for building expressive GNNs. Our code is available at https://anonymous.4open.science/r/EquivNoiseGNN.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext f8d587ba-579b-44fc-8f7f-defc40e9f68aBuilds on45
- Open Graph Benchmark: Datasets for Machine Learning on GraphsWeihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong et al.NeurIPS 2020 · 3,935 citations
- Do Transformers Really Perform Badly for Graph Representation?Chengxuan Ying, Tianle Cai, Shengjie Luo, Shuxin Zheng et al.NeurIPS 2021 · 1,632 citations
- Geom-GCN: Geometric Graph Convolutional NetworksHongbin Pei, Bingzhe Wei, Kevin Chen-Chuan Chang, Yu Lei et al.ICLR 2020 · 1,445 citations
- E(n) Equivariant Graph Neural NetworksVictor Garcia Satorras, Emiel Hoogeboom, Max WellingICML 2021 · 1,432 citations
- Recipe for a General, Powerful, Scalable Graph TransformerLadislav Rampásek, Michael Galkin, Vijay Prakash Dwivedi, Anh Tuan Luu et al.NeurIPS 2022 · 1,216 citations
Related papers
- What functions can Graph Neural Networks compute on random graphs? The role of Positional EncodingNicolas Keriven, Samuel VaiterNeurIPS 2023 · 24 citations
- On the Expressive Power of Spectral Invariant Graph Neural NetworksBohang Zhang, Lingxiao Zhao, Haggai MaronICML 2024 · 20 citations
- Universally Invariant Learning in Equivariant GNNsJiacheng Cen, Anyi Li, Ning Lin, Tingyang Xu et al.NeurIPS 2025 · 7 citations
- A Flexible, Equivariant Framework for Subgraph GNNs via Graph Products and Graph CoarseningGuy Bar-Shalom, Yam Eitan, Fabrizio Frasca, Haggai MaronNeurIPS 2024 · 9 citations
- Learning Efficient Positional Encodings with Graph Neural NetworksCharilaos I. Kanatsoulis, Evelyn Choi, Stefanie Jegelka, Jure Leskovec et al.ICLR 2025
