When does global attention help: a unified empirical study on atomistic graph learning

Arindam Chowdhury, Massimiliano Lupo Pasini

Main figure for When does global attention help: a unified empirical study on atomistic graph learning

Abstract

Graph neural networks (GNNs) are widely used as surrogates for costly experiments and first-principles simulations to study the behavior of compounds at atomistic scale, and their architectural complexity is constantly increasing to enable the modeling of complex physics. While most recent GNNs combine more traditional message passing neural networks (MPNNs) layers to model short-range interactions with more advanced graph transformers (GTs) with global attention mechanisms to model long-range interactions, it is still unclear when global attention mechanisms provide real benefits over well-tuned MPNN layers due to inconsistent implementations, features, or hyperparameter tuning. We introduce the first unified, reproducible benchmarking framework–built on HydraGNN–that enables seamless switching among four controlled model classes: MPNN, MPNN with chemistry/topology encoders, GPS-style hybrids of MPNN with global attention, and fully fused localglobal models with encoders. Using seven diverse open-source datasets for benchmarking across regression and classification tasks, we systematically isolate the contributions of message passing, global attention, and encoder-based feature augmentation. Our study shows that encoder-augmented MPNNs form a robust baseline, while fused localglobal models yield the clearest benefits for properties governed by long-range interaction effects. We further quantify the accuracycompute trade-offs of attention, reporting its overhead in memory. Together, these results establish the first controlled evaluation of global attention in atomistic graph learning and provide a reproducible testbed for future model development.

Benchmark Results

Tested on 3 datasets

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

OGB

#ModelAP ↑Paper
1
GRPE-Large
0.3150 ±0.0010
2
Graphormer
0.3140
3
Graphormer-FLAG
0.3139 ±0.0032
31
GIN
0.2902 ±0.0017
103
S3
0.1840
This paper
104
S4
0.1820
This paper
108
S2
0.1670
This paper
109
S1
0.1550
This paper
#ModelMAE ↓Paper
1
DeMol
0.0603
2
TGT-At
0.0671
3
TGT-At (+RDKit)
0.0683
64
GIN
0.1195
67
S2
0.1247
This paper
74
S4
0.1351
This paper
79
S1
0.1412
This paper
81
S3
0.1467
This paper
#ModelAccuracy ↑Paper
1
A-NLSF
0.8149 ±0.0067
2
differential encoding
0.8096 ±0.0029
3
MPNN + VN + NoPE
0.8055 ±0.0038
16
GatedGCN
0.7531 ±0.0083
33
GIN
0.7037 ±0.0107
40
S4
0.6701
This paper
41
S1
0.6524
This paper
42
S2
0.6375
This paper
43
S3
0.5324
This paper
SOTA performance (not significantly different from the best) Baseline / comparison (this paper) Global context (other papers)