PCE-GNN: a node feature-enhanced graph neural network with pre-clustering strategy
摘要
Graph Neural Networks (GNNs) exhibit excellent performance in extracting node features from graph-structured data. To enhance the representation of central node features and mitigate the over-smoothing issue, several models have refined their methods for acquiring information from distant neighbor nodes. However, most of these methods overlook the impact of distant same-type nodes on the central node and are unable to adequately mine the information contained in these distant neighbor nodes, which limits their performance. To address this, we propose a GNN model with a pre-clustering strategy, called PCE-GNN. Specifically, PCE-GNN enhances node representations through two collaborative modules: the local aggregation module effectively aggregates 1-hop neighbor information via a multi-head graph attention mechanism, while the long-distance similar neighbor aggregation module combines a pre-clustering strategy with GNN layers to utilize reconstructed star-shaped subgraphs for capturing information of distant neighbor nodes with similar features. Subsequently, these two parts of information are integrated via a max-pooling layer to form the final representation of the central node. Experimental results show that the dual-module collaborative approach of PCE-GNN possesses strong feature enhancement capabilities, outperforming baselines in node classification tasks on both public datasets and equipment maintenance datasets. The source code is available at http://github.cn/SanJinCabbage/PCE-GNN.