Unifying Graph Neural Networks Through a Common Layer Equation
A single common equation is proposed to describe dozens of different graph neural network layers
Graph neural networks have traditionally been described with family-specific equations and notation, which hides the computations they actually share and where they truly differ structurally. This paper introduces a common layer equation that breaks a GNN layer into seven components. Its central idea is separating where information moves in the graph from what information actually moves.
METAL MEDIA explanatory visual
A single common equation is proposed to describe dozens of different graph neural network layers
- 01Different families of graph neural networks have each been described with their own notation, making it hard to see what computations are shared across models and where they structurally diverge
- 02The author decomposes a GNN layer into seven components: an update domain, channel set, propagation bank, per-channel message maps, a channel-fusion operator, an ego/residual map, and an update map
- 03The key factorization separates the propagation bank, which encodes where information moves across the graph, from the message maps, which encode what information moves
- 04This common equation is shown to represent a range of existing GNN architectures within the same unified framework
What they did
- Different families of graph neural networks have each been described with their own notation, making it hard to see what computations are shared across models and where they structurally diverge
- The author decomposes a GNN layer into seven components: an update domain, channel set, propagation bank, per-channel message maps, a channel-fusion operator, an ego/residual map, and an update map
- The key factorization separates the propagation bank, which encodes where information moves across the graph, from the message maps, which encode what information moves
- This common equation is shown to represent a range of existing GNN architectures within the same unified framework
Why it matters
Being able to describe GNN models from different papers in one shared notation lets researchers focus on real structural differences instead of translating between inconsistent equations when designing new models or comparing existing ones. It could also give a consistent basis for teaching GNNs or implementing them in libraries.
Terms in this paper
- Graph Neural Network (GNN) · A neural network designed to learn from graph-structured data made of nodes and edges
- propagation bank · The component defining where information travels between nodes in a graph layer
- message maps · Functions that define what value is computed and passed along for each channel of information
- channel-fusion operator · An operation that combines messages coming from multiple channels into one
- ego/residual map · A mapping that incorporates a node's own previous state into its update
Original abstract (English)
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.
Read on arXivLatest papers
- SWE-bench Science: Can Coding Agents Resolve Engineering Tasks in Science?AI coding agents were tested on fixing real scientific software, and even the best one failed more than half the time
- FlashPrefill V2: Block-Sparse Prefill Attention for Long-Context LLM ServingMaking sparse attention fast enough and accurate enough for real LLM serving, not just papers
- PolicyGuide: From Guarding One Action to Guiding the Whole Workflow for Policy-Compliant LLM AgentsMaking customer-service AI agents follow the whole procedure, not just avoid one bad action
- EXIMO: VLM Guided Exploration of VLA PoliciesTeaching a robot new chores without human teleoperation, by letting a chatty AI supervise it
- EnvHarness: Awakening Static Worlds for Agent LearningInstead of building new training worlds from scratch, this work adds a plug-in layer that reshapes existing ones around each agent's actual weaknesses
- Bounded Sovereignty and the Control Tax: Pricing AI Oversight When the Deployer Does Not Own the ModelCompanies that rent AI instead of owning it can only do half of AI safety oversight
- Beyond Imitation: Filtering On-Policy Distillation by Reasoning ProgressA fix for AI models that get penalized by their teacher even when they're reasoning correctly
- PersonalBench: Measuring the Authorship Gap in LLM PersonalizationAI can be prompted to write 'like someone,' but its own voice never fully disappears