AAAI Publications, Twenty-Ninth AAAI Conference on Artificial Intelligence

Font Size: 
Clustering-Based Collaborative Filtering for Link Prediction
Xiangyu Wang, Dayu He, Danyang Chen, Jinhui Xu

Last modified: 2015-02-09


In this paper, we propose a novel collaborative filtering approach for predicting the unobserved links in a network (or graph) with both topological and node features. Our approach improves the well-known compressed sensing based matrix completion method by introducing a new multiple-independent-Bernoulli-distribution model as the data sampling mask. It makes better link predictions since the model is more general and better matches the data distributions in many real-world networks, such as social networks like Facebook. As a result, a satisfying stability of the prediction can be guaranteed. To obtain an accurate multiple-independent-Bernoulli-distribution model of the topological feature space, our approach adjusts the sampling of the adjacency matrix of the network (or graph) using the clustering information in the node feature space. This yields a better performance than those methods which simply combine the two types of features. Experimental results on several benchmark datasets suggest that our approach outperforms the best existing link prediction methods.


Link Prediction; Compressed Sensing; Matrix Completion; Collaborative Filtering

Full Text: PDF