Ordered GNN: Ordering Message Passing to Deal with Heterophily and Over-smoothing
Yunchong SongChenghu ZhouXinbing WangZhouhan Lin
Proposes Ordered GNN, a message-passing architecture that aligns neighborhood tree hierarchies with dedicated neuron blocks to prevent over-smoothing in deep networks while matching state-of-the-art accuracy across both homophilic and heterophilic graphs.
Graph neural networks are vital machine learning tools for analyzing interconnected data across domains like social networks, biology, and communication systems. However, standard models suffer from two major flaws: "over-smoothing," where node representations become indistinguishable and degrade in performance as models grow deeper, and "heterophily," where connected nodes have dissimilar attributes or labels, causing harmful mixing of conflicting information.
The article introduces and evaluates "Ordered GNN," a novel architecture designed to resolve both challenges by restructuring how information is merged across network neighborhoods. The approach aligns a node's local network hierarchy with specific blocks of artificial neurons in its representation. Using a specialized gating mechanism, the model assigns distinct neuron segments to specific neighborhood distances, preventing the conflation of local and distant features without requiring complex structural modifications.
The authors evaluated the framework across 11 benchmark datasets spanning citation and web networks, including large-scale graphs with over 160,000 nodes. The evaluation assessed classification accuracy across homophilic networks (where connected nodes are similar), heterophilic networks (where connected nodes differ), and deep network architectures scaling up to 64 layers.
The results show that Ordered GNN consistently matches or surpasses current state-of-the-art models across both network types. In heterophilic environments, it achieved major gains, such as reaching 62.44% accuracy on the dense Squirrel benchmark compared to 36.48%–38.47% from leading baselines. Crucially, the model maintained high classification accuracy even at depths of 32 and 64 layers, completely avoiding the severe performance drops typical of deep baseline architectures. Visualizations confirmed that the model automatically adapts its internal gating, preserving local identities on heterophilic data and filtering distant noise in deep configurations, all while maintaining competitive per-epoch training speeds comparable to standard models.
These findings demonstrate that managing the combination stage of message passing is an efficient, unified solution to multiple core graph learning limitations. By eliminating the need for ad-hoc heuristics, special edge-dropping training tricks, or separate models for different network types, Ordered GNN reduces engineering complexity and deployment risk in enterprise applications.
Organizations deploying graph machine learning should consider incorporating ordered gating principles into their model pipelines, particularly for deep networks or non-homophilic data. For future initiatives, the framework can be extended to link-level and graph-level tasks or tailored for complex graph types like knowledge graphs. However, practitioners should exercise caution on highly skewed, scale-free networks with extreme hub structures, where varying local topologies may require additional few-shot techniques or further data balancing.
- Paper: Beyond Homophily in Graph Neural Networks: Current Limitations and Effective Designs, Jiong Zhu et al. (2020). This seminal paper diagnoses why standard message-passing fails under heterophily and introduces core architectural designs, such as separating ego and neighbor embeddings, which Ordered GNN directly builds upon.
- Paper: Neural Sheaf Diffusion: A Topological Perspective on Heterophily and Oversmoothing in GNNs, Cristian Bodnar et al. (2022). It provides a topological perspective connecting the joint phenomena of heterophily and over-smoothing that Ordered GNN explicitly aims to resolve through ordered message passing.
- Paper: Geom-GCN: Geometric Graph Convolutional Networks, Hongbin Pei et al. (2020). This paper establishes standard benchmarks and analysis for heterophilic network evaluation that serve as essential context and baseline comparisons for Ordered GNN.
- Paper: Simple and Deep Graph Convolutional Networks, Ming Chen et al. (2020). It analyzes the over-smoothing bottleneck when scaling GCNs up to 64 layers, setting the primary depth benchmark and architectural challenge targeted by Ordered GNN.
- Paper: Measuring and Relieving the Over-smoothing Problem for Graph Neural Networks from the Topological View, Deli Chen et al. (2019). It formalizes quantitative metrics and topological causes of the over-smoothing problem across network layers, providing critical background for understanding representation collapse.
- Paper: DropEdge: Towards Deep Graph Convolutional Networks on Node Classification, Yu Rong et al. (2019). This work explores deep GCN training and techniques to prevent over-smoothing, representing the heuristic and edge-manipulation methods that Ordered GNN seeks to overcome without complex structural modifications.
- Paper: Representation Learning on Graphs with Jumping Knowledge Networks, Keyulu Xu et al. (2018). It introduces Jumping Knowledge Networks to adaptively combine multi-hop representations across layers, motivating layer-wise neighborhood gating mechanisms.
- Paper: Finding Global Homophily in Graph Neural Networks When Meeting Heterophily, Xiang Li et al. (2022). This study addresses non-local feature aggregation under heterophily, illustrating alternative strategies for handling disassortative graph structures.
- Paper: A Generalization of ViT/MLP-Mixer to Graphs, Xiaoxin He et al. (2023). This work extends beyond local message passing by adapting vision-style patch and channel mixing to graphs, presenting an alternative way to bypass neighborhood over-squashing and long-range bottlenecks.
- Paper: Simple and Efficient Heterogeneous Graph Neural Network, Xiaocheng Yang et al. (2023). It explores architectural simplifications and efficiency improvements for complex heterogeneous networks, continuing the investigation of tailored graph message passing.
