Link Prediction Algorithm Based On Dual-channel Graph Neural Network
The advancement of network science and technology has propelled humanity into the era of complex networks, and link prediction has emerged as a key focus in complex network research. With the gradual maturity of graph neural network technology, it can automatically learn the representation vectors of nodes based on downstream tasks. Researchers have developed various methods using graph neural networks (GNNs) to address the link prediction task, but these methods have not fully utilized multiple information sources. To tackle the challenges, we design a dual-channel GNN-based link prediction algorithm. This method extracts enclosing subgraphs from two different graphs constructed from adjacency matrix and node feature matrix. Then, it inputs the subgraph information into a Deep Graph Convolutional Neural Network (DGCNN) to learn embeddings of target node pairs. Finally, the embeddings obtained from both channels are merged and fed into a fully connected neural network (FCNN) for prediction. The method has been experimentally tested on multiple networks in this paper. Findings reveal that it outperforms comparative methods and delivers outstanding prediction outcomes.
Authors
- Longjie Li (ORCID: https://orcid.org/0000-0001-5866-617X)
- Lianfei Yu (ORCID: https://orcid.org/0009-0005-4877-6259)
- Haiwen Chen (ORCID: https://orcid.org/0000-0002-6470-3464)
- Jiaxuan Liu (ORCID: https://orcid.org/0009-0007-4238-2480)
- Qun Zhang (ORCID: https://orcid.org/0000-0002-2773-3437)
- Xiaoyun Chen (ORCID: https://orcid.org/0000-0003-1476-0951)
- Shiyu Fang (ORCID: https://orcid.org/0009-0000-2884-1504)
Institutions
- National University of Defense Technology (CN)
- Urumqi Vocational University (CN)
- Lanzhou University (CN)
Publication Details
- Published
- 2024-10-25
- DOI
- https://doi.org/10.1109/bigdia63733.2024.10808273
- Primary Topic
- Wireless Sensor Networks and IoT
- Type
- article
- Field-Weighted Citation Impact
- 0.00