Research on Link Prediction Based on Improved Graph Convolutional Network

Authors

  • Ruijie Huang

DOI:

https://doi.org/10.54097/ykkpw239

Keywords:

Link Prediction, Graph Convolutional Network (GCN), Node Embedding, Negative Sampling, AUC Evaluation.

Abstract

Link prediction is a fundamental task in network analysis with extensive applications in social networks, recommendation systems, and biological networks. In this work, we propose an improved Graph Convolutional Network (IGCN) model that leverages an encoder–decoder architecture, integrating multi-layer GCN-based aggregation of node features and local structure with adaptive hard negative sampling and contrastive learning. Our framework is evaluated on three benchmark datasets—Cora, Citeseer, and Pubmed—and achieves state-of-the-art performance when compared with traditional methods and other graph neural network models. This study demonstrates that incorporating adaptive negative sampling and contrastive loss effectively enhances the discriminative power of node representations for link prediction. Future research will focus on model scalability and the integration of domain-specific knowledge to further broaden its application scope.

Downloads

Download data is not yet available.

References

[1] Zhang M, Chen Y. Link prediction based on graph neural networks[J]. Advances in neural information processing systems, 2018, 31.

[2] Cai L, Ji S. A multi-scale approach for graph link prediction[C]//Proceedings of the AAAI conference on artificial intelligence. 2020, 34(04): 3308-3315.

[3] Cai L, Li J, Wang J, et al. Line graph neural networks for link prediction[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2021, 44(9): 5103-5113.

[4] Cukierski W, Hamner B, Yang B. Graph-based features for supervised link prediction[C]//The 2011 International joint conference on neural networks. IEEE, 2011: 1237-1244.

[5] Arrar D, Kamel N, Lakhfif A. A comprehensive survey of link prediction methods[J]. The journal of supercomputing, 2024, 80(3): 3902-3942.

[6] Nguyen T K, Fang Y. Diffusion-based negative sampling on graphs for link prediction[C]//Proceedings of the ACM Web Conference 2024. 2024: 948-958.

[7] Deng W, Zhang Y, Yu H, et al. Knowledge graph embedding based on dynamic adaptive atrous convolution and attention mechanism for link prediction[J]. Information Processing & Management, 2024, 61(3): 103642.

[8] Dileo M, Zignani M, Gaito S. Temporal graph learning for dynamic link prediction with text in online social networks[J]. Machine Learning, 2024, 113(4): 2207-2226.

[9] Li M, Wang Z, Liu L, et al. Subgraph-aware graph kernel neural network for link prediction in biological networks[J]. IEEE Journal of Biomedical and Health Informatics, 2024.

[10] Zhang Y, Chen J, Cheng Z, et al. Edge propagation for link prediction in requirement-cyber threat intelligence knowledge graph[J]. Information Sciences, 2024, 653: 119770.

Downloads

Published

27-02-2025

Issue

Section

Articles