HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Breaking the Expressive Bottlenecks of Graph Neural Networks

Mingqi Yang; Yanming Shen; Heng Qi; Baocai Yin

Breaking the Expressive Bottlenecks of Graph Neural Networks

Abstract

Recently, the Weisfeiler-Lehman (WL) graph isomorphism test was used to measure the expressiveness of graph neural networks (GNNs), showing that the neighborhood aggregation GNNs were at most as powerful as 1-WL test in distinguishing graph structures. There were also improvements proposed in analogy to $k$-WL test ($k>1$). However, the aggregators in these GNNs are far from injective as required by the WL test, and suffer from weak distinguishing strength, making it become expressive bottlenecks. In this paper, we improve the expressiveness by exploring powerful aggregators. We reformulate aggregation with the corresponding aggregation coefficient matrix, and then systematically analyze the requirements of the aggregation coefficient matrix for building more powerful aggregators and even injective aggregators. It can also be viewed as the strategy for preserving the rank of hidden features, and implies that basic aggregators correspond to a special case of low-rank transformations. We also show the necessity of applying nonlinear units ahead of aggregation, which is different from most aggregation-based GNNs. Based on our theoretical analysis, we develop two GNN layers, ExpandingConv and CombConv. Experimental results show that our models significantly boost performance, especially for large and densely connected graphs.

Code Repositories

qslim/epcb-gnns
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
graph-property-prediction-on-ogbg-ppaExpC
Ext. data: No
Number of params: 1369397
Test Accuracy: 0.7976 ± 0.0072
Validation Accuracy: 0.7518 ± 0.0080

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Breaking the Expressive Bottlenecks of Graph Neural Networks | Papers | HyperAI