Graph Rewiring in GNNs to Mitigate Over-Squashing and Over-Smoothing: A Survey
Summary
A survey by Davide Buscaldi and Fragkiskos D. Malliaros reviews graph rewiring techniques to enhance Graph Neural Network (GNN) performance by mitigating over-squashing and over-smoothing. These phenomena, which degrade information flow and make node representations indistinguishable, stem from message passing interacting with input topology. The survey categorizes rewiring methods into structural-fix, which modify topology based on measures like discrete curvature, effective resistance, random walk-based metrics, and spectral gap, and feature-aware, which use node attributes for rewiring. It details specific techniques such as Ollivier Curvature, Augmented Forman Curvature, and Delaunay Rewiring. The authors also discuss practical trade-offs, including performance attribution, hyperparameter sensitivity, and computational complexity, emphasizing that rewiring should be evaluated beyond just downstream accuracy.
Key takeaway
For research scientists developing or deploying GNNs, understanding graph rewiring is crucial for overcoming performance limitations in heterophilic graphs or those with long-range dependencies. You should consider implementing rewiring strategies, either structural-fix or feature-aware, to explicitly modify graph topology. Evaluate these methods not only by downstream accuracy but also by their impact on structural diagnostics like effective resistance or commute times, ensuring the chosen technique aligns with the task's effective radius and computational constraints.
Key insights
Graph rewiring modifies GNN topology to combat over-squashing and over-smoothing, improving information flow.
Principles
- Topology directly impacts GNN information flow.
- Rewiring can be static (preprocessing) or dynamic (during training).
- Evaluation needs structure-level diagnostics, not just accuracy.
Method
Graph rewiring involves adding, removing, or reweighting edges based on topological descriptors (e.g., curvature, effective resistance, spectral gap) or node features to optimize information propagation.
In practice
- Use curvature-based rewiring to alleviate bottlenecks.
- Employ feature-aware methods when graph topology is noisy.
- Consider virtual nodes to reduce graph diameter.
Topics
- Graph Rewiring Techniques
- Over-squashing Mitigation
- Over-smoothing Mitigation
- Graph Topology Modification
- Discrete Curvature
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Student
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.LG updates on arXiv.org.