Unifying Graph Neural Networks Through a Common Layer Equation
Organizations: University of Texas at Dallas · Pacific Northwest National Laboratory · Vanderbilt University · Dolby Laboratories ·
Abstract
Graph neural networks are commonly described through family-specific equations whose notation obscures shared computations and structural differences. We introduce a common layer equation that represents covered architectures through seven components: an update domain, channel set, propagation bank, per-channel message maps, channel-fusion operator, ego/residual map, and update map. The central factorization separates where information moves, encoded by the propagation bank, from what moves, encoded by the message maps. Function-valued fillings extend the same equation across local message passing, attention, spectral filtering, global communication, relation-specific channels, higher-order domains, and geometric messages. We make this unification explicit and checkable through worked reductions of canonical layers and component assignments spanning seven nonexclusive architectural families. A fixed slot discipline assigns operations by computational role and defines the framework's coverage boundary. The decomposition also yields component-level theoretical insights: under endpoint-local messages and node-local updates, operator support bounds one-layer dependencies, and one-layer global mixing requires a full effective operator row under the stated hypotheses. The resulting framework organizes more than 200 architectures in a common design space, enables component-wise comparison and generation of structurally consistent architectures, and connects propagation choices to oversmoothing, oversquashing, heterophily, and expressivity. It further exposes the empirical inverse problem of mapping measurable graph and task properties to validated component choices.
Figures & tables
| Method | Propagation | Message | Mixing | Ego/Update | ||
|---|---|---|---|---|---|---|
| 1 Spatial local support in with differences in aggregation and ego/update | ||||||
| GCN Kipf and Welling ( 2017 ) | ||||||
| GraphSAGE- Mean Hamilton et al. ( 2017 ) | ||||||
| GIN Xu et al. ( 2019b ) | ||||||
| 2 Attention state-dependent in | ||||||
| GAT Veličković et al. ( 2018 ) | (avg. at final layer) | |||||
| Formalism | Canonical computational objects | Not separately exposed | Exact relation or boundary |
| MPNN ( Gilmer et al., 2017 ) | Node states, a pairwise message , summation over neighbors, a node update , and a graph-level readout | The canonical equations expose a single neighbor-aggregated message . They do not separately expose a channel index set, a decomposition of propagation support from message values, a distinct cross-channel mixing operator, or an independently named ego/residual construction | Exact node-domain specialization of the pairwise branch with , , , , , , and . The graph-level readout lies outside the base-layer equation |
| Graph Network block ( Battaglia et al., 2018 ) | A graph ; edge update ; edge-to-node aggregation ; node update ; edge-to-global and node-to-global aggregations and ; and global update | The GN formalism explicitly types edge , node, and global domain updates and aggregations, but it does not separately parameterize them by a channel inventory with distinct support operators , value maps , and a cross-channel mixing operator | Its node-update slice is represented exactly by taking edge incidence as the support of , , , , , and , with supplied as shared context. A full GN block additionally updates edge and global states: in the canonical ordering, consumes aggregated updated edges and consumes aggregated updated edges and updated nodes. Representing the full block therefore requires multiple typed domains and staged updates rather than a single node-domain base-layer application |
| Augmented message-passing view ( Veličković, 2022 ) | Pairwise message passing applied to a suitably augmented computational graph, where features, connectivity, or represented entities may be modified so that a richer computation is realized through pairwise communication | This is a broad representational viewpoint rather than a prescribed layer tuple. It does not require an explicit channel inventory, support value factorization, or separately named cross channel mixing component | Broader in representational scope. A computation is captured by one seven-component base-layer application only when the relevant synchronous step admits the stated channel-wise normalization. Computations requiring graph augmentation, replicated entities, multiple dependent stages, or other transformations may still fall within augmented message passing while requiring more than one base-layer application or an augmented computational graph |
| Seven-component equation (this work) | , , , , , , and | The training objective, post-stack readout, external domain transitions such as pooling, and multi-stage control flow are outside the base layer | A typed reporting form for the covered layer class. It refines the canonical MPNN description by separately exposing support, channel-wise value computation, mixing, and base-state construction. It captures the node-update slice of a Graph Network block but not an entire staged edge node global block in one node-domain application, and it is intentionally narrower than the broader augmented-message-passing viewpoint |
| Method | Propagation | Message | Mixing | Ego/Update | New Slot | ||
| Level 1 Recombination equation and inventory fixed, with fillers combined across families | |||||||
| Spectrally-Gated Propagation | (node-adaptive) | – | |||||
| Relation- Equivariant Network | (relations) | (equiv. ) | (scalar/vector) | – | |||
| Subgraph- Attentive Layer | (subgr.) | (heads) | (state-dependent) | (external fold ) | – | ||
| Spectral-Biased Transformer | (global) | – | |||||
| Level 2 New fillers equation fixed, with a filler introduced that the inventory lacks | |||||||
| Remedy class (instance) | Targets | Slot | Quantity moved (trade-off) | Key refs |
| Curvature rewiring (SDRF, BORF) | oversquashing | support/weights | raises on negative-curvature edges; pruning can slow smoothing | Topping et al.,2022 ; Nguyen et al.,2023 ; Liu et al.,2023a |
| Spectral-gap rewiring (FoSR, GTR) | oversquashing | support; relation channels | raises / lowers ; relational split avoids induced smoothing | Karhadkar et al.,2023 ; Black et al.,2023 |
| Spectral pruning (Braess) | squashing and smoothing | support/weights | selected edge deletions can raise and co-mitigate both under the cited criterion | Jamadandi et al.,2024 ; Liang et al.,2025 |
| Learned rewiring (PR-MPNN) | oversquashing, underreaching | sampled support | differentiable -subset edges; conditions for random rewiring | Qian et al.,2024 |
| Virtual / global channel (MPNN VN) | oversquashing, long-range | and for VN; for dense attention | VN adds a two-edge path (two synchronous propagation steps); classical VN forces uniform importance | Southern et al.,2025b ; Cai et al.,2023 |
| Signed / high-pass bank (FAGCN, GPR-GNN, BernNet, ChebNetII) | heterophily, oversmoothing | bank ( folded in) | shifts the response high or uses signed ; can suppress the dominant smoothing response | Bo et al.,2021 ; Chien et al.,2021 ; He et al.,2021 ; He et al.,2022b |
| Phenomenon | Diagnostic measure | Governing quantity (and bound) | Key refs |
| Oversmoothing (forward) | Dirichlet energy ; MAD (Mean Average Distance); numerical rank | largest nontrivial eigenvalue magnitude: ; under Theorem 2 ; rank under the cited conditions | Oono and Suzuki,2020 ; Cai and Wang,2020 ; Roth and Liebig,2024 ; Zhang et al.,2026b |
| Oversharpening (dual) | high-frequency energy growth | sign of channel-mixing eigenvalues: most-negative dominant repulsion | Giovanni et al.,2023b |
| Oversmoothing (backward) | smoothness of backprop. error; flat high-loss regions | linear backward smoothing; per-layer Lipschitz bound | Keriven,2026 ; Park and Kim,2024 |
| Oversquashing | Jacobian ; mixing | ; ; capacity | Topping et al.,2022 ; Giovanni et al.,2023a ; Black et al.,2023 ; Giovanni et al.,2024 |
| Underreaching | receptive radius task distance | depth : | Giovanni et al.,2023a ; Errica et al.,2025 |
| Heterophily / band mismatch | aggregation homophily; Contextual Stochastic Block Model (CSBM) Bayes error | pass-band of vs. signal band; low-pass response of is | NT and Maehara,2019 ; Luan et al.,2022 ; Ma et al.,2022 ; Luan et al.,2023 |
| Model class (instance) | WL power | Counting power | Memory | Key refs |
| Mean/max/attention-aggregation MPNN (GCN, GraphSAGE-mean, GAT) | -WL (strict for mean/max) | loses multiset multiplicities (mean/attention) or repeats (max) | Xu et al.,2019b | |
| Injective sum-aggregation MPNN (GIN) | -WL (the ceiling, attained iff aggregate and readout are injective) | stars only; no induced connected pattern nodes; no girth/diameter/ -clique/biconnectivity | Xu et al.,2019b ; Chen et al.,2020c ; Garg et al.,2020 ; Zhang et al.,2023b | |
| Polynomial spectral GNN (GPR-GNN, BernNet, ChebNetII) | -WL under the structure-blind node-domain hypotheses of Proposition 4 ; fixed-graph reachability is a separate axis | filter-band dependent; reaches any -D target on one fixed graph if the spectrum is simple and no frequency is missing | Wang and Zhang,2022 | |
| Spectral-invariant GNN (eigenbasis invariants) | strictly between -WL and the subgraph-GNN band above it | homomorphism-counts (hom-counts) exactly “parallel trees”; class grows with depth | Gai et al.,2025 | |
| Hom-count-augmented MPNN ( -MPNN; spasm basis) | -WL; gain over -WL depends on ; for every fixed some escapes -WL | counts patterns via | Barceló et al.,2021 ; Jin et al.,2024 | |
| Node-based subgraph GNN (ESAN, GNN-AK, SUN, SSWL) | -WL, -FWL ( -WL) for the six ordered classes | the analyzed subgraph-MPNN class node-counts cycles only through length | Frasca et al.,2022 ; Zhang et al.,2023a |
| Benchmark family | Task / domain | Documented design or finding | Component-level reading | Key refs |
| Citation networks (Cora, Citeseer, Pubmed; Planetoid splits) | Citation graphs | Node classification on homophilous graphs. Adjusted homophily is preferable for cross-dataset comparison, label informativeness agrees more closely with GNN performance, and rankings depend on splits and tuning | encodes topology, but no layer component is isolated | Yang et al.,2016 ; Shchur et al.,2018 ; Platonov et al.,2023a ; Luo et al.,2024b |
| OGB node / link ( ogbn , ogbl ) | Multiple domains and tasks | Large datasets with application-specific splits, metrics, and standardized pipelines. Scale and out-of-distribution generalization are common challenges | Dataset- and task-dependent; scale, split, and prediction head † | Hu et al.,2020a |
| Heterophily suite, legacy (Geom-GCN: Texas, Wisconsin, Cornell, Actor, Chameleon, Squirrel) | Node classification | Geom-GCN evaluated geometric aggregation on this suite. Duplicate nodes in Chameleon and Squirrel create train–test leakage and materially change results when removed | No reliable component attribution; and are relevant design axes | Pei et al.,2020 ; Platonov et al.,2023b |
| Heterophily suite, corrected (Roman-empire, Amazon-ratings, Minesweeper, Tolokers, Questions) | Node classification | Diverse heterophilous graphs without the identified duplicate-node leakage. Standard GNNs are strong baselines, and connectivity is better characterized jointly by adjusted homophily and label informativeness | and are relevant but not isolated | Platonov et al.,2023b ; Platonov et al.,2023a |
| Large non-homophilous (LINKX datasets) | Social and web node classification | Substantially larger non-homophilous graphs. Existing scalable and mini-batch methods degrade on these datasets | ; scale and mini-batching † | Lim et al.,2021 |
| TUDataset | Graph classification and regression across more than 120 datasets | A repository with standardized loaders, baselines, and evaluation rather than a controlled test of one architectural mechanism | Dataset-dependent; graph-level readout | Morris et al.,2020a |
| Open problem | Why it is hard | What would count as progress | Slot | Key refs |
|---|---|---|---|---|
| Generalization beyond expressivity | More expressive fillings provably inflate the complexity term (WL classes bound Rademacher complexity), and no theory ties SGD implicit bias to generalization | Non-vacuous, structure-aware bounds naming which statistics control error at a reasonable computational cost, valid for sparse unbounded-size graphs, and connected to scaling the operator (Section 7.5 ) | Morris et al.,2024 ; Vasileiou et al.,2025 ; Carrasco et al.,2025 ; Amran et al.,2026 ; Rauchwerger et al.,2025 | |
| Distribution / size shift | Failure is a shift over local structure. Causal-invariance remedies rest on unverifiable assumptions, and rankings reverse across shift types | A measurable predictor of which statistic governs transfer and validated environment-free causal-subgraph discovery | Yehudai et al.,2021 ; Li et al.,2025b ; Loveland et al.,2024 ; Chen et al.,2022c ; Gui et al.,2022 | |
| Dataset-adaptive slot selection | Homophily is sensitive to class count/balance and unreliable for model selection. No mechanistic property-to-slot map exists | A robust measured property predicting the right across datasets | You et al.,2020 ; Platonov et al.,2023a ; Luan et al.,2026 ; Kormann et al.,2026 | |
| Slots as a search space (graph neural architecture search, graph-NAS) | Graph-NAS learns shift-fragile non-causal maps, is non-reproducible without a shared space, and uses language-model controllers that lack guarantees | Causal, data-conditioned search whose discovered tuple generalizes under shift | all seven components | Wang et al.,2022b ; Qin et al.,2022 ; Li et al.,2025c ; Gao et al.,2025 |
| Scaling the operator | Whether dense attention is even needed is unproven. It is unknown how much expressivity sub-quadratic attention sacrifices, sparsification and condensation guarantees do not transfer, and no clean long-range benchmark exists | A characterized expressivity-efficiency frontier and a structured operator provably matching dense attention | Deng et al.,2024 ; Sancak et al.,2025 ; Chen et al.,2023d ; Tönshoff et al.,2024 ; Gupta et al.,2026 | |
| Temporal / time-indexed carry slot | Dominant temporal families are incomparable, standard messages cannot express moving averages, and component-to-gain links are untested | A temporal WL/counting ladder and a principled continuous-time memory operator with known limits | Souza et al.,2022 ; Tjandra et al.,2024 ; Cong et al.,2023 ; Li et al.,2025a ; Tieu et al.,2025 |
Appendix figures & tables11 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Meaning |
|---|---|
| Part I | Standard notation. Graph objects, matrices, spectral quantities, features, and learning conventions following established usage in spectral graph theory and graph neural networks. | |
| Graph structure | |
| Graph with node set and edge set | |
| , | Number of nodes and edges |
| Neighborhood of node | |
| Degree of node | |
| Operation | Primary location | Time | Effect |
| Positional / structural encoding | ; external input for pre-stack encoding; for pairwise score biases | infer | Adds position/structure descriptors; can break neighborhood-isomorphism symmetry. |
| Rewiring / virtual node | (support/weights; if nodes added) | infer | Changes which nodes communicate; can relieve oversquashing bottlenecks. |
| Skip / residual | infer | Re-injects earlier state; enables depth, can counter oversmoothing. | |
| Normalization | primarily; for state pre-norm | infer | Rescales/recenters states; stabilizes training, can oppose collapse. |
| Pooling / readout | external for flat readout; / via external if hierarchical | infer | Produces graph-level vector; coarsens support in stages. |
| Continuous-depth / ODE | primarily; for augmented dynamics; only if objects are added | infer | Reads depth as integration time over a chosen graph vector field. |
| Result | Reference | Description |
|---|---|---|
| Spectral subsumption | (Lemma 1 ) | A layer applying scalar functions of a fixed shift operator through its spectral calculus is the unified equation with ; ChebNet, SGC, APPNP, GPR-GNN, and BernNet are each an exact special case, separated by the operator/basis, the coefficient constraint, and the coupled-versus-decoupled message slot. |
| Concatenation is additive mixing | (Prop. 1 ) | Multi-head concatenation equals a sum of head outputs embedded into disjoint output blocks, introducing no separate mixing mechanism beyond summation. |
| Concatenation expressivity | (Cor. 1 ) | In the linear regime concatenation matches additive mixing only after projection to a common width; concatenation itself creates explicit head separation in disjoint coordinates, and downstream weights (or a blockwise nonlinearity) decide whether it is exploited or erased. |
| Attention subsumption | (Lemma 3 ) | A single-head layer whose neighbor weight is a scalar function of node states times a linear value is the unified equation with a state-dependent operator; restricted to the single-head, edge-feature-free case, GAT and GATv2 differ only in the score inside . |
| Channel count not identifiable | (Prop. 2 ) | The linear pre-activation depends only on , so the channel count is not recoverable and the invariant is the channel rank. |
| Receptive field and underreaching | (Prop. 3 ) | Locally computed one-hop channels confine the depth- receptive field to the -hop ball, so a node output is independent of inputs farther than hops; a globally state-dependent operator escapes this even at one-hop support. |
| Order | Primary family | Assignment rule | How secondary mechanisms are handled |
| 1 | Family 1. Spatial message passing | Choose this family when the layer is best described as ordinary neighbor aggregation with shared message and update maps, and no more specific rule below is dominant. | Fixed-point solvers, readouts, sampling procedures, and routing loops are marked as external machinery when they are not one synchronous layer update. |
| 2 | Family 2. Attention and adaptive weighting | Choose this family when the main contribution is state-dependent attention, adaptive edge weights, learned masks, gates, routing coefficients, or generated aggregation weights. | Spatial supports and relation features remain secondary unless another family better describes the cited contribution. |
| 3 | Family 3. Spectral and diffusion | Choose this family when the main contribution is a Laplacian or adjacency polynomial, spectral filter, diffusion kernel, hop basis, signed propagation basis, or learned graph filter. | Ordinary neighbor aggregation is treated as the implementation of an operator basis rather than the primary family. |
| 4 | Family 4. Graph transformer | Choose this family when the defining layer performs global token mixing, transformer self-attention, structural attention biasing, sparse global attention, or tokenized graph communication. | Positional encodings, structural encodings, and local message passing are treated as supporting components. |
| 5 | Family 5. Heterogeneous and multi-relational | Choose this family when typed nodes, typed edges, relation channels, meta-paths, or knowledge-graph relation states define the update. | Attention over relations remains secondary when relation typing is the defining axis. |
| 6 | Family 6. Higher-order and subgraph | Choose this family when the updated objects are tuples, rooted subgraphs, simplices, cells, hyperedges, motifs, or another lifted domain rather than ordinary nodes alone. | Attention or spectral operations on the lifted domain are treated as secondary mechanisms. |
| Method | {\color[rgb]{1,1,1}\bm{\mathcal{X}}} | {\color[rgb]{1,1,1}\bm{\mathcal{K}}} | Propagation | Message | Mixing | Ego/Update |
|---|---|---|---|---|---|---|
| 1a Anchor models fixed one-hop spatial propagation | ||||||
| GCN Kipf and Welling (2017) | ||||||
| GraphSAGE-Mean Hamilton et al. (2017) | ||||||
| GIN Xu et al. (2019b) | ||||||
| MPNN Gilmer et al. (2017) | ||||||
| 1b Aggregators, gates, message maps, and implicit fixed points one-hop support with nonlinear message or update maps | ||||||
| Method | Propagation | Message | Mixing | Ego/Update | ||
|---|---|---|---|---|---|---|
| 2a Query–key attention static and dynamic attention operators | ||||||
| GAT Veličković et al. (2018) | ||||||
| GATv2 Brody et al. (2022) | ||||||
| GAT-POS Ma et al. (2021) | ||||||
| AGNN Thekumparampil et al. (2018) | ||||||
| GaAN Zhang et al. (2018) | ||||||
| Method | Propagation | Message | Mixing | Ego/Update | ||
|---|---|---|---|---|---|---|
| 3a Explicit spectral and polynomial filters fixed/learned filter banks, linear in the signal | ||||||
| SpectralCNN Bruna et al. (2014) | (input/output channel pair) | |||||
| ChebNet Defferrard et al. (2016) | ( -rescaled , distinct from ) | |||||
| CayleyNet Levie et al. (2019) | (complex coefficients folded into ) | |||||
| GWNN Xu et al. (2019a) | (wavelet; input/output channel pair) | |||||
| LanczosNet Liao et al. (2019) | (Ritz) | |||||
| Method | Propagation | Message | Mixing | Ego/Update | ||
|---|---|---|---|---|---|---|
| 4a Dense global attention with structural bias full or subgraph all-pairs softmax; bias inside the operator Graph Trans. Dwivedi and Bresson (2021) is the one exception, restricting to (genuinely local, not all-pairs), grouped here for its Transformer-block lineage rather than its support | ||||||
| Graph-BERT Zhang et al. ( 2020 ) | ||||||
| Graph Trans. Dwivedi & Bresson ( 2021 ) | ||||||
| Graphormer Ying et al. ( 2021 ) | ||||||
| SAN Kreuzer et al. (2021) | ||||||
| GraphiT Mialon et al. (2021) | ||||||
| Method | Propagation | Message | Mixing | Ego/Update | ||
|---|---|---|---|---|---|---|
| 5a Per-relation operators and weights explicit per-relation supports, weights, and relation/KG scoring | ||||||
| R-GCN Schlichtkrull et al. (2018) | ||||||
| RGAT Busbridge et al. (2019) | ||||||
| CompGCN Vashishth et al. (2020) | ||||||
| KBGAT Nathani et al. (2019) | ||||||
| 5b Meta-path supports channels are (learned or fixed) meta-paths; semantic-level mixing | ||||||
| Method | Propagation | Message | Mixing | Ego/Update | ||
|---|---|---|---|---|---|---|
| 6a Tuple-based ( -order / -WL) models states over -tuples; equivariant linear maps and FWL | ||||||
| IEGN Maron et al. (2019b) | (equality-pattern equivariant basis; ) | (including equivariant bias basis) | ||||
| -GNN Morris et al. (2019) | (or optional loc/glob split) | (support ; local variant uses only) | (optional loc/glob split sums channel contributions) | |||
| PPGN Maron et al. (2019a) | (product channels; shown as ) | (state-dep. operator; matrix prod. 2-FWL) | ( gives ; concat in ) | |||
| Ring-GNN Chen et al. (2019) (partial slice) | (equiv. linear maps; biases included) | |||||
| - -LGNN Morris et al. (2020b) | (selects with ) | (value ) | (replacement multisets feed joint tuple aggregation) | |||
| Method | Propagation | Message | Mixing | Ego/Update | ||
|---|---|---|---|---|---|---|
| 7a Invariant (distance / angle / torsion) messages -invariant scalars; coordinates enter only through invariants | ||||||
| MoNet Monti et al. ( 2017 ) | ||||||
| SchNet Schütt et al. ( 2017 ) | ||||||
| DimeNet Gasteiger et al. ( 2020 ) | ||||||
| SphereNet Liu et al. ( 2022b ) ; chirality-sensitive | ||||||
| GemNet Gasteiger et al. ( 2021 ) ; apart from reflections | ||||||
Graph Neural Networks
Graph Neural Networks learn from graph-structured data by combining information across connected nodes, edges, and larger structures.
See all 714 papers