arXiv · 2308.16800
Rank Collapse Causes Over-Smoothing and Over-Correlation in Graph Neural Networks
Abstract
Our study reveals new theoretical insights into over-smoothing and feature over-correlation in graph neural networks. Specifically, we demonstrate that with increased depth, node representations become dominated by a low-dimensional subspace that depends on the aggregation function but not on the feature transformations. For all aggregation functions, the rank of the node representations collapses, resulting in over-smoothing for particular aggregation functions. Our study emphasizes the importance for future research to focus on rank collapse rather than over-smoothing. Guided by our theory, we propose a sum of Kronecker products as a beneficial property that provably prevents over-smoothing, over-correlation, and rank collapse. We empirically demonstrate the shortcomings of existing models in fitting target functions of node classification tasks.
Explore related subjects
Keep this discovery
Andreas Roth, Thomas Liebig. 2023-08-31. Rank Collapse Causes Over-Smoothing and Over-Correlation in Graph Neural Networks. https://arxiv.org/abs/2308.16800
Cite the original work for its findings. Save a collection to share your selection of sources.