Skip to content
← Models
ClosedGabriele Corso et al. (University of Cambridge)/10 trustVerified

Principal Neighbourhood Aggregation (PNA)

One aggregator is not enough: mean, max, min and std aggregation with degree scalers for continuous features.

Visit official page →

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

N/A - graph-structured model; no token context window

Pricing

Free (open-source research code)

Free quota

Unlimited (self-hosted)

API access

No hosted API; self-host the authors' open-source PNA implementation

Weights

Closed — API only

Released

Not disclosed

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

graph-regressionmolecular-property-predictiongraph-classificationresearch

Trust score — 8/10

/10 trustVerified

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

Sources