Expressive Graph Neural Networks via Equivariant Use of Noise
Xiyuan Wang, Muhan Zhang
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper45
- Open Graph Benchmark: Datasets for Machine Learning on GraphsWeihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong 等NeurIPS 2020 · 被引用 3,935 次
- Do Transformers Really Perform Badly for Graph Representation?Chengxuan Ying, Tianle Cai, Shengjie Luo, Shuxin Zheng 等NeurIPS 2021 · 被引用 1,632 次
- Geom-GCN: Geometric Graph Convolutional NetworksHongbin Pei, Bingzhe Wei, Kevin Chen-Chuan Chang, Yu Lei 等ICLR 2020 · 被引用 1,445 次
- E(n) Equivariant Graph Neural NetworksVictor Garcia Satorras, Emiel Hoogeboom, Max WellingICML 2021 · 被引用 1,432 次
- Recipe for a General, Powerful, Scalable Graph TransformerLadislav Rampásek, Michael Galkin, Vijay Prakash Dwivedi, Anh Tuan Luu 等NeurIPS 2022 · 被引用 1,216 次
相关 Paper
- What functions can Graph Neural Networks compute on random graphs? The role of Positional EncodingNicolas Keriven, Samuel VaiterNeurIPS 2023 · 被引用 24 次
- On the Expressive Power of Spectral Invariant Graph Neural NetworksBohang Zhang, Lingxiao Zhao, Haggai MaronICML 2024 · 被引用 20 次
- Universally Invariant Learning in Equivariant GNNsJiacheng Cen, Anyi Li, Ning Lin, Tingyang Xu 等NeurIPS 2025 · 被引用 7 次
- A Flexible, Equivariant Framework for Subgraph GNNs via Graph Products and Graph CoarseningGuy Bar-Shalom, Yam Eitan, Fabrizio Frasca, Haggai MaronNeurIPS 2024 · 被引用 9 次
- Learning Efficient Positional Encodings with Graph Neural NetworksCharilaos I. Kanatsoulis, Evelyn Choi, Stefanie Jegelka, Jure Leskovec 等ICLR 2025
