Graph neural network with feature enhancement of isolated marginal groups

Jiahui Wang,Yi Guo,Zhihong Wang,Xinxiu Wen,Jianwei Ni
DOI: https://doi.org/10.1007/s10489-022-03277-x
IF: 5.3
2022-03-30
Applied Intelligence
Abstract:Graph neural networks (GNN) have achieved remarkable success by combining feature and structure information. However, the over-smoothing phenomenon has always been a crucial issue in GNN models since the node representation will easily converge to the full graph representation with the increasing of convolutional layers. Our investigation also indicates that the isolated part significantly restricts the capability due to the meaningless structure and local over-smoothing problem. While current models treated all nodes as equal with the absence of effectiveness of structure at different locations. To facilitate this line of research, we innovatively propose the graph neural network with feature enhancement of isolated parts (GNN-FEIP) consisting of graph partition, graph construction, and strategic label propagation procedures. In GNN-FEIP architecture, all the nodes are partitioned into several groups according to their position and connectivity. The feature-level similarity graph is reconstructed for subsequent feature enhancement of isolated nodes. Afterward, the preliminary prediction of the original GNN model has been adjusted with the strategic label propagation, which both balances the feature and structure of the nodes at different positions in a comprehensive manner. The exhaustive experiments indicate that GNN-FEIP achieves impressive performance than other off-the-shelf models, especially in the case that the isolated part is of large proportion.
computer science, artificial intelligence
What problem does this paper attempt to address?