All Nodes Are Created Not Equal: Node-Specific Layer Aggregation and Filtration for GNN

Shilong Wang,Hao Wu,Yifan Duan,Guibin Zhang,Guohao Li,Yuxuan Liang,Shirui Pan,Kun Wang,Yang Wang
DOI: https://doi.org/10.48550/arxiv.2405.07892
2024-01-01
Abstract:The ever-designed Graph Neural Networks, though opening a promising path forthe modeling of the graph-structure data, unfortunately introduce two dauntingobstacles to their deployment on devices. (I) Most of existing GNNs areshallow, due mostly to the over-smoothing and gradient-vanish problem as theygo deeper as convolutional architectures. (II) The vast majority of GNNs adhereto the homophily assumption, where the central node and its adjacent nodesshare the same label. This assumption often poses challenges for many GNNsworking with heterophilic graphs. Addressing the aforementioned issue hasbecome a looming challenge in enhancing the robustness and scalability of GNNapplications. In this paper, we take a comprehensive and systematic approach toovercoming the two aforementioned challenges for the first time. We propose aNode-Specific Layer Aggregation and Filtration architecture, termed NoSAF, aframework capable of filtering and processing information from each individualnodes. NoSAF introduces the concept of "All Nodes are Created Not Equal" intoevery layer of deep networks, aiming to provide a reliable information filterfor each layer's nodes to sieve out information beneficial for the subsequentlayer. By incorporating a dynamically updated codebank, NoSAF dynamicallyoptimizes the optimal information outputted downwards at each layer. Thiseffectively overcomes heterophilic issues and aids in deepening the network. Tocompensate for the information loss caused by the continuous filtering inNoSAF, we also propose NoSAF-D (Deep), which incorporates a compensationmechanism that replenishes information in every layer of the model, allowingNoSAF to perform meaningful computations even in very deep layers.
What problem does this paper attempt to address?