Principal Neighbourhood Aggregation (PNA)
One aggregator is not enough: mean, max, min and std aggregation with degree scalers for continuous features.
About
Most GNN layers pick a single aggregation function and hope it fits every neighborhood. PNA argues this is provably insufficient for continuous features: a single aggregator cannot distinguish distributions that differ only in statistics it does not compute. The paper extends the WL-style expressiveness framework to continuous features, shows multiple complementary aggregators are needed, and builds the layer to match: mean, max, min and standard deviation aggregations, each modulated by degree-based scalers (identity, amplification, attenuation) that let the network react to neighborhood size. A logarithmic moment normalization keeps the combination numerically stable.\n\nThe NeurIPS 2020 paper introduces a synthetic multi-task benchmark designed to isolate aggregation quality, where PNA scores -3.13 log-MSE against -1.99 for the best single-aggregator GIN and -2.04 for GCN, winning even with fewer parameters than enlarged single-aggregator baselines. PNA also performs strongly on larger-graph extrapolation, where the max aggregator's robustness carries over. The multiple-aggregator principle has since become common in expressive GNN design.
Specs
Context window
Pricing
Free quota
API access
Weights
Released
Benchmarks
Paper-reported: -3.13 log-MSE on the authors' multi-task benchmark vs -1.99 for the best single-aggregator GIN and -2.04 for GCN (Corso et al., NeurIPS 2020). Benchmark is synthetic and author-designed; treat as supporting rather than decisive evidence.
Provider-reported unless a source says otherwise. Benchmarks are marketing-adjacent by nature — treat them as directional.
Use cases
Trust score — 8/10
Scored 8/10 (verified tier, metrics-v1). Evidence: NeurIPS 2020 paper with a genuine theoretical contribution (expressiveness framework extended to continuous features); the multi-aggregator principle has been widely adopted. Strengths: principled diagnosis of single-aggregator limits, parameter-efficient wins over enlarged baselines. Docked for: (1) the headline benchmark is synthetic and author-designed, with no independent replication on real datasets of comparable clarity; (2) no official pretrained checkpoints. License: open-source code release; check the repo license before commercial use.
How the trust score is built
Full methodology →- Adoption · stars/downloads, log-scaled vs peers
- 7.0/10
- Maintenance · recency, releases, issue load
- 6.0/10
- Openness · license + docs
- 9.0/10
- Manual review · sanity read + red-flag check
- 8.0/10
- · NeurIPS 2020 paper
- · Extends WL expressiveness framework to continuous features
- · Author-designed synthetic benchmark: -3.13 vs -1.99 (GIN)
- · Multi-aggregator principle widely adopted since