Tackling Long-tailed Distribution Issue in Graph Neural Networks via Normalization
Langzhang Liang,Zenglin Xu,Zixing Song,Irwin King,Yuan Qi,Jieping Ye
DOI: https://doi.org/10.1109/tkde.2023.3315284
IF: 9.235
2023-01-01
IEEE Transactions on Knowledge and Data Engineering
Abstract:Graph Neural Networks (GNNs) have attracted much attention due to their superior learning capability. Despite the successful applications of GNNs in many areas, their performance suffers heavily from the long-tailed node degree distribution. Most prior studies tackle this issue by devising sophisticated model architectures. In this paper, we aim to improve the performance of tail nodes (low-degree or hard-to-classify nodes) via a generic and light normalization method. In detail, we propose a novel normalization method for GNNs, termed as ResNorm, which Reshapes a long-tailed distribution into a normal-like distribution via Normalization. The ResNorm includes two operators. Firstly, the scale operator reshapes the distribution of the node-wise standard deviation (NStd) so as to improve the accuracy of tail nodes. Secondly, the analysis of the behavior of the standard shift indicates that the standard shift serves as a preconditioner on the weight matrix, increasing the risk of over-smoothing. To address this issue, we design a new shift operator for ResNorm, which simulates the degree-specific parameter strategy in a low-cost manner. Extensive experiments on various node classification benchmark datasets have validated the effectiveness of ResNorm in improving the performance of tail nodes as well as the overall performance.
computer science, information systems, artificial intelligence,engineering, electrical & electronic