NeurIPS 2024

Sequential Signal Mixing Aggregation for Message Passing Graph Neural Networks

Mitchell Keren Taraday, Almog David, Chaim Baskin

Main figure for Sequential Signal Mixing Aggregation for Message Passing Graph Neural Networks

Abstract

Message Passing Graph Neural Networks (MPGNNs) have emerged as the preferred method for modeling complex interactions across diverse graph entities. While the theory of such models is well understood, their aggregation module has not received sufficient attention. Sum-based aggregators have solid theoretical foundations regarding their separation capabilities. However, practitioners often prefer using more complex aggregations and mixtures of diverse aggregations. In this work, we unveil a possible explanation for this gap. We claim that sum-based aggregators fail to "mix" features belonging to distinct neighbors, preventing them from succeeding at downstream tasks. To this end, we introduce Sequential Signal Mixing Aggregation (goldSSMA), a novel plug-and-play aggregation for MPGNNs. goldSSMA treats the neighbor features as 2D discrete signals and sequentially convolves them, inherently enhancing the ability to mix features attributed to distinct neighbors. By performing extensive experiments, we show that when combining goldSSMA with well-established MPGNN architectures, we achieve substantial performance gains across various benchmarks, achieving new state-of-the-art results in many settings. We published our code at

Benchmark Results

Tested on 1 dataset 2 baseline values reported too low

Tables show presented results in global context (best 3 papers + GCN + MLP).

LRGB

#ModelMAE ↓Paper
1
GCN
actual #1 vs paper #166 165 places worse
0.2421 ±0.0016
best reported
actual #1 vs paper #166 165 places worse
paper 0.2800 ±0.0100
paper 0.2800 ±0.0100 Δ at pub 0.0340 Δ today 0.0379
2
IPR-MPNN
0.2422 ±0.0007
3
TANGO-GPS
0.2422 ±0.0014
5
GIN
actual #5 vs paper #158 153 places worse
0.2700 ±0.0100
actual #5 vs paper #158 153 places worse
paper 0.2700 ±0.0100
paper 0.2700 ±0.0100 Δ at pub 0.0847 Δ today 0.0271
87
GraphGPS
actual #87 vs paper #166 79 places worse
0.2500 ±0.0005
best reported
actual #87 vs paper #166 79 places worse
paper 0.2800 ±0.0100
paper 0.2800 ±0.0100 Δ at pub 0.0300 Δ today 0.0300
141
GCN + SSMA
0.2600 ±0.0200
This paper
142
GAT + SSMA
0.2600 ±0.0100
This paper
143
GATv2 + SSMA
0.2600 ±0.0100
This paper
144
GIN + SSMA
0.2600 ±0.0200
This paper
159
GAT
actual #159 vs paper #159
0.2700 ±0.0100
actual #159 vs paper #159
paper 0.2700 ±0.0100
prev best: Graph Neural Networks Gone Hogwild (ICLR 2024)
paper 0.2700 ±0.0100 Δ at pub 0.0650
160
GATv2
0.2700 ±0.0100
This paper
161
GraphGPS + SSMA
0.2700 ±0.0100
This paper
162
PNA + SSMA
0.2700 ±0.0300
This paper
167
PNA
0.2800 ±0.0100
This paper
201
MLP
0.4351 ±0.0008
SOTA performance (not significantly different from the best) Baseline / comparison (this paper) Global context (other papers) Best at pub — reported result worse than the best known when published (best reported in green) Best at pub — reported value improved over the best known when published Improved at pub — later paper surpassed this result