Link Prediction via Subgraph Embedding-Based Convex Matrix Completion


  • Zhu Cao Tsinghua University, IIIS
  • Linlin Wang Tsinghua University, IIIS
  • Gerard de Melo Rutgers University



Link prediction is of fundamental importance in network science and machine learning. Early methods consider only simple topological features, while subsequent supervised approaches typically rely on human-labeled data and feature engineering. In this work, we present a new representation learning-based approach called SEMAC that jointly exploits fine-grained node features as well as the overall graph topology. In contrast to the SGNS or SVD methods espoused in previous representation-based studies, our model represents nodes in terms of subgraph embeddings acquired via a form of convex matrix completion to iteratively reduce the rank, and thereby, more effectively eliminate noise in the representation. Thus, subgraph embeddings and convex matrix completion are elegantly integrated into a novel link prediction framework. Experimental results on several datasets show the effectiveness of our method compared to previous work.




How to Cite

Cao, Z., Wang, L., & de Melo, G. (2018). Link Prediction via Subgraph Embedding-Based Convex Matrix Completion. Proceedings of the AAAI Conference on Artificial Intelligence, 32(1).