Research Article | Open Access
Link Prediction via Convex Nonnegative Matrix Factorization on Multiscale Blocks
Low rank matrices approximations have been used in link prediction for networks, which are usually global optimal methods and lack of using the local information. The block structure is a significant local feature of matrices: entities in the same block have similar values, which implies that links are more likely to be found within dense blocks. We use this insight to give a probabilistic latent variable model for finding missing links by convex nonnegative matrix factorization with block detection. The experiments show that this method gives better prediction accuracy than original method alone. Different from the original low rank matrices approximations methods for link prediction, the sparseness of solutions is in accord with the sparse property for most real complex networks. Scaling to massive size network, we use the block information mapping matrices onto distributed architectures and give a divide-and-conquer prediction method. The experiments show that it gives better results than common neighbors method when the networks have a large number of missing links.
As a fundamental problem in the network researches, link prediction attempts to estimate the likelihood of relationship between two individuals by the study of observed links and the property of nodes. Researches on the problem can benefit a variety of fields. For example, researchers in different areas can efficiently find their cooperative partners or assistants. Security agencies can more precisely focus their efforts on probable relationships in malicious networks. In social networks, people can find companions based on the prediction of their surrounding networks.
The natural framework of link prediction methods is the similarity-based algorithm. Simple similarity-based measures such as neighborhood-based measures, for example, Adamic-Adar score  and common neighbors , require consideration of the local structure of the networks. Recently, a considerable amount of work which draws attention to community structure and scalable proximity estimation [3, 4] gives good prediction accuracy. Some similarity-based measures such as the path based methods, for example, Katz  and Rooted PageRank [6, 7], which focus on the global structure of the networks, are more effective but have a high computational complexity. A new measure based on neighbor communities has a good performance with a low complexity . Maximum likelihood estimation, such as hierarchical structure model  and stochastic block model [9, 10], presuppose some organizing principles of the network structure. Some algorithms, such as probabilistic relational models , probabilistic entity-relationship models , and stochastic relational models , learn the underlying structure from the observed network and then predict the missing links. Lichtenwalter et al.  designed a flow based method for link prediction, which is more localized. Low rank matrices approximations can also be used in link prediction for network [15–17]. Based on the technique of cluster low rank approximation for massive graphs, Shin et al. proposed a multiscale link prediction method , which captures the information of global structure of network and can handle massive networks quickly.
In order to capture the information of both global structure and clustering structure of network, we consider low rank approximations as well as blocks in networks’ adjacent matrices. Low rank approximations algorithms are good techniques to get the global information of the matrices. Meanwhile block structure is an important feature of matrices and it is often true that links in the same blocks have similar properties. Indeed, links are easy to be found in dense blocks. Good block detection algorithms have error tolerance: they are unaffected by a few missing edges in a network. This suggests that the principle of block detection could be applied to edge prediction.
Theoretically, a probabilistic latent variable model is proposed that combines both the concepts of block structure and low rank approximations for matrices. The model provides a framework for predicting links. Firstly, any modularity clustering algorithm can be used to generate blocks, while the only limit is the computational complexity. Then different from the low rank matrices approximations algorithms already used for link predictions, we use a new low rank matrices approximations algorithm named convex nonnegative matrix factorization (CNMF)  to get the predicting results within the blocks. The reason we use CNMF is that the sparseness of solutions is in accord with the sparse property for most real complex networks, so the predicting results are more reliable. In small networks we use -means to detect the block structure of a network’s adjacency matrix and average the prediction matrices for some to get the predicting results. Experiments show that our method shows better performance. Scaling to the massive size networks, it is infeasible to use CNMF directly for the high computational complexity. In this case, we use the block information mapping matrices onto distributed architectures and give a divide-and-conquer prediction method to embrace distributed computing.
The network of nodes can be represented mathematically by an adjacency matrix . Here we set the diagonal entries to be 1, which means each node has a link to itself. This adjacency matrix can be treated as an object-feature matrix. Reduced rank method CNMF gives an approximation which has an interesting property: if is sparse, the factors and both tend to be sparse.
CNMF has a direct interpretation: matrix is a convex combinations of the columns of ; thus we could interpret its columns as weighted sums of certain objects’ coordinates (the coordinate of -object is given by -column of ). So the columns of can be treated as cluster centroids of objects and weights the association between objects and clusters. Meanwhile measure the strength of relationship between clusters and features; that is, if cluster has feature ; otherwise. So can measure the strength of relationship between object with feature and then can be used to predict link between and .
3. A Probabilistic Latent Variable Model
Although the background gives an intuitionistic interpretation of CNMF used in link prediction, we still need theoretical guarantee. Here we propose a probabilistic latent variable model, and the model ensures that the probability of a link between two nodes can be expressed as a combination of CNMF and the block structure of a given adjacent matrix.
In probabilistic view, the observed network data is a realization of an underlying probabilistic model, either because it is itself the result of a stochastic process, or because the sampling has uncertainty. We can think of the adjacent matrices of network data as the object-feature matrices for objects and features . In this paper, contains an adjacency matrix and its blocks found by clustering methods such as -means. The joint occurrence probability of an object and a feature can be factorized as where is a variable for the index of objects, is a variable for the index of features, and is a variable for the index of sampling. is the joint occurrence conditional probability given the observation , and is the priori probability that is observed.
Objects in real data are often organized into modules or clusters and the probability that a object has a feature depends on the groups to which they belong. These clusters memberships are unknown to us. In the language of statistical inference, they are latent variable. Assuming each cluster is a combination of objects, the joint occurrence probability can be factorized as where is the variable for the index of cluster. Here, we assume that the random variables , are conditional independent given .
Define a random variable If observing once, the expected value is Let be a random variable of the occurrence frequency of in observing times. Then the expected value is
For the reason of interpretability, we suppose the joint concurrent probability of th object, th cluster, and th data sampling is given by a combination of data as follows: where and Constraint (8) ensures that the probability defined by (7) is well defined.
This constraint has the advantage that we could interpret as weighted sums of certain joint concurrence probability of object, features, and data, given by Therefore, (6) can be expressed as where . Our goal is to compute two -order tensors and .
If we are only inputting the adjacency matrix, we can drop the index for sampling; then Equation (11) can be expressed by matrix as Now, we show that this factorization is equivalent to CNMF.
In fact, for any CNMF solution , does not hold and holds for any . Let be the diagonal matrix, containing the row sums of . We say that the matrix approximately satisfies (8). This can be proved as follows: So the factorization satisfies the condition of (11). If we are also inputting blocks, (10) can be solved as where and are solutions of CNMF for . This can be proved as the case of only inputting the adjacency matrix.
The algorithm of CNMF gives a global optimal solution to . The computational complexity of it (the most time-consuming step in our method) for matrix is of order for factor and is of order for factor , where is the number of iterations .
Inputting the network data, the missing link prediction by calculating (15) has three steps. First, partition the observed adjacent matrix into blocks, using any modularity clustering algorithms. Secondly, the predicting matrix is given by doing CNMF to approximate each block. Thirdly, sum the corresponding entities of predicting matrices for all to make the final prediction. In small networks, We call our method -CNMF, as we use -means to partition the matrix. The diagram of -CNMF for is shown in Figure 1.
The purpose of the first step is to use several scales structures information of the observed network. For small networks, CNMF approximation can be computed directly on the original matrix (block generated by -means with ) to use the global information. A simple interpretation of our method is that if an edge is predicted to exist in many scales, it should be a missing link with high probability. The input of -CNMF also needs two parameters: desired rank and scale parameter . Algorithm 1 shows the algorithm for -CNMF.
When predicting links in massive size network, -means is unsuitable for high dimensional data clustering. Meanwhile, the high computational complexity of CNMF makes it also infeasible to be used on the large adjacent matrix directly. So we use fast modularity clustering algorithm  to generate blocks. Based on block structures, we give a divideand-conquer algorithm (-CNMF) to predict links, which is shown in Algorithm 2. The algorithm works by partitioning a matrix into blocks which are small enough for CNMF directly. Then the predicting results for the small blocks are combined to give the final predicting result for the original matrix. In order to give a solution for CPU load balancing, the size of blocks should be similar, which is achieved by splitting the large blocks and combining the small blocks to make their sizes in the neighborhood of a given threshold.
5. Experiments and Comparison
In general, links between different nodes may have different weights in networks, representing their relative importance in the network. In our experiments, we set all weights to be one and get the original adjacency matrix of the network. The observed network is generated by removing a fraction of links randomly from the original network , which will be called the missing edges. Then we use the two algorithms -CNMF and -CNMF to get the probability of links between nodes, which appears to be links’ weight in the observed network.
5.1. Evaluation Method
To measure the accuracies of link prediction methods, the main metric we use is AUC , area under the receiver operating characteristic (ROC) curve, which is widely used in the machine learning and social network communities. If we rank pairs of nodes in order of decreasing, AUC is mean value of the probability that a missing link () has a higher ranking than a nonexistent link (). In practice, we do independent comparisons. At each time we randomly pick a missing link and a nonexistent link to compare their ranking. If there are times the missing link has a higher ranking than the missing one and times they have the same ranking, the AUC value is
The missing links fraction ranges from 0.05 to 0.95, and the interval is set at 0.05.
5.2. Methods Used to Compare
We compare our algorithm with three prediction methods: Common Neighbors, Block Model, and Hierarchical Random Graphs.
(1) Common Neighbors (CN) . If two nodes, and have many common neighbors, they are more likely to have a link. The measure of this is where is the set of neighbors of .
(2) Block Model (BM) . In block models, nodes are partitioned into groups and the connecting probability of two nodes only depends on the groups they belong to. Given a partition of the network, is the number of edges in the observed network between nodes in groups and , and is the maximum possible number of links between and . The reliability of an individual link is where the sum is over partitions in the space of all possible partitions of the network, is node ’s group, , and .
(3) Hierarchical Random Graphs (HRG) . The hierarchical structure of a network can be represented by a dendrogram with leaves (the vertices from the given network) and internal nodes. A probability is associated with internal node and the connecting probability of a pair of leaves is equal to , where is the deepest common ancestor of these two leaves. HRG combines the maximum likelihood method and Markov chain Monte Carlo method to sample the hierarchical structure with probability proportional to their likelihood from the observed network and then calculate .
5.3. Performance of -CNMF
We evaluate the performance of -CNMF using four high-quality small networks, and they are listed in Table 1: the social network of interactions between people in a karate club , the social network of frequent associations between dolphins , the air transportation network of USA, the coappearance network of characters in the novel Les Miserables , and a network of hyperlinks between weblogs on US politics . Each AUC is obtained by averaging over 100 independent realizations.
Communities are basic structure in networks, which is widely used to predict missing links. Using block structure, our combined method is also dependent on communities of the networks. As Figures 2(a) and 2(b) show, -CNMF (, ) performs much better than CNMF () alone on Karate and Les-Mis, because both of those networks have more than two communities. However the enhancement by block structure is small on PB with desired rank (see Figure 2(c)), which has two main communities. As the desired rank increases, the enhancement by block structure decreases. That is because the local information can be revealed by the richness of clustering structures given by CNMF with high desired rank. So the enhancement is also small on US-Air (nodes: 332) with desired rank (see Figure 2(d)). Will more block information usage bring more accuracy?
If partitioning matrix into too small blocks, -CNMF will have too many parameters relative to the observed data, and then overfitting will occur. An overfitted model describes noise instead of the underlying relationship and generally has poor predictive performance. From the performance of -CNMF () with different on Dolphins (see Table 2), we can see that -CNMF () has revealed enough local information, and increasing caused overfitting.
Figure 3 shows the comparison of -CNMF with CN, BM, and HRG on Karate (inputting parameters of -CNMF: , ) and Dolphins (, ), Les-Mis (, ), and US-Air (, ). -CNMF performs better than CN, because it concerns the property of both local and global information. The performance of -CNMF is comparable with BM and HRG, but faster, as it does not need Monte Carlo samplings.
5.4. Performance of -CNMF
We examine the performance of -CNMF by the main components of four real-world networks: Arxiv GR-QC collaboration network (inputting parameters: , ) , the Western States Power Grid of the United States (, ) , Enron email network (, ) , and the subnetwork of EU email communication network generated by email form the first 5000 nodes to first 10000 nodes (, ) .
Comparisons are made only between -CNMF and CN, for the reason that BM and HRG are not suitable for large networks. Figure 4 shows the comparison of -CNMF with CN. The performances of -CNMF are better than CN when the observed networks are much sparse, because common neighbors miss too much in sparse case and CN only concerns this property. In the power network, our method is obviously better than CN, because the original network is sparse.
Figures 5(a) and 5(b) show the comparison of AUC results between -CNMF and -CNMF on Karate and Dolphins, respectively, where , . There are no obvious rules that different modularity clustering algorithms will influence the results of AUC.
We have introduced a probabilistic latent variable model for finding missing edges, which combines convex nonnegative matrix factorization with block structures detection. It is inspired by two properties of block structure for matrices: the facts that entities in the same block tend to be similar and that good block detection algorithms have tolerance to missing edges. Scaling to massive size network, we use fast modularity clustering algorithm to generate blocks and give a divide-and-conquer algorithm (-CNMF) for predicting links. For the load balancing of CPU, we split the large blocks and combine the small blocks to make their sizes in the neighborhood of a given threshold.
Since most applications of link prediction are facing the problems of sparse data, such as personalized recommendation, we plan to combine other sparse low rank approximation algorithms with block detection methods to get effective link prediction algorithms for massive networks in the future.
Conflict of Interests
The authors declare that they do not have any commercial or associative interest that represents a conflict of interests in connection with the work.
- L. A. Adamic and E. Adar, “Friends and neighbors on the Web,” Social Networks, vol. 25, no. 3, pp. 211–230, 2003.
- M. E. J. Newman, “Clustering and preferential attachment in growing networks,” Physical Review E, vol. 64, Article ID 025102, 2001.
- A. Clauset, C. Moore, and M. E. J. Newman, “Hierarchical structure and the prediction of missing links in networks,” Nature, vol. 453, no. 7191, pp. 98–101, 2008.
- R. Guimerà and M. Sales-Pardo, “Missing and spurious interactions and the reconstruction of complex networks,” Proceedings of the National Academy of Sciences of the United States of America, vol. 106, no. 52, pp. 22073–22078, 2009.
- L. Katz, “A new status index derived from sociometric analysis,” Psychometrika, vol. 18, no. 1, pp. 39–43, 1953.
- D. Liben-Nowell and J. Kleinberg, “The link prediction problem for social networks,” in Proceedings of the 12th ACM International Conference on Information and Knowledge Management (CIKM '03), pp. 556–559, ACM, November 2003.
- L. Lü and T. Zhou, “Link prediction in complex networks: a survey,” Physica A: Statistical Mechanics and its Applications, vol. 390, no. 6, pp. 1150–1170, 2011.
- Z. Xie, E. Dong, J. Li, D. Kong, and N. Wu, “Potential links by neighbor communities,” Physica A, vol. 406, pp. 244–252, 2014.
- M. Girvan and M. E. J. Newman, “Community structure in social and biological networks,” Proceedings of the National Academy of Sciences of the United States of America, vol. 99, no. 12, pp. 7821–7826, 2002.
- R. Guimerà, M. Sales-Pardo, and L. A. N. Amaral, “Classes of complex networks defined by role-to-role connectivity profiles,” Nature Physics, vol. 3, no. 1, pp. 63–69, 2007.
- N. Friedman, L. Getoor, D. Koller, and A. Pfeffer, “Learning probabilistic relational models,” in Proceedings of the 16th International Joint Conference on Artificial Intelligence (IJCAI '99), pp. 1300–1307, Stockholm, Sweden, August 1999.
- D. Heckerman, C. Meek, and D. Koller, “Probabilistic entity-relationship models, PRMs, and plate models,” in Proceedings of the 21st International Conference on Machine Learning, p. 55, Banff, Canada, 2004.
- K. Yu, W. Chu, S. Yu, V. Tresp, and Z. Xu, “Stochastic relational models for discriminative link prediction,” in Proceedings of the Neural Information Precessing Systems, p. 1553, MIT Pres, Cambridge, Mass, USA, 2006.
- R. N. Lichtenwalter, J. T. Lussier, and N. V. Chawla, “New perspectives and methods in link prediction,” in Proceedings of the 16th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD '10), pp. 243–252, July 2010.
- B. Savas and I. S. Dhillon, “Clustered low rank approximation of graphs in information science applications,” in Proceedings of the 11th SIAM International Conference on Data Mining (SDM '11), pp. 164–175, April 2011.
- A. Menon and C. Elkan, “Link prediction via matrix factorization,” in Proceedings of the 2011 ECML PKDD European Conference, pp. 437–452, Athens, Greece, September 2011.
- D. M. Dunlavy, T. G. Kolda, and E. Acar, “Temporal link prediction using matrix and tensor factorizations,” ACM Transactions on Knowledge Discovery from Data, vol. 5, no. 2, article 10, 2011.
- D. Shin, S. Si, and I. S. Dhillon, “Multi-scale link prediction,” in Proceedings of the 21st ACM International Conference on Information and Knowledge Management (CIKM '12), pp. 215–224, ACM, November 2012.
- C. Ding, T. Li, and M. I. Jordan, “Convex and semi-nonnegative matrix factorizations,” IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 32, no. 1, pp. 45–55, 2010.
- U. Brandes, D. Delling, M. Gaertler et al., “On modularity clustering,” IEEE Transactions on Knowledge and Data Engineering, vol. 20, no. 2, pp. 172–188, 2008.
- D. J. Hand and R. J. Till, “A simple generalisation of the area under the ROC curve for multiple class classification problems,” Machine Learning, vol. 45, no. 2, pp. 171–186, 2001.
- W. W. Zachary, “An information flow model for conflict and fission in small groups,” Journal of Anthropological Research, vol. 33, pp. 452–473, 1977.
- D. Lusseau, K. Schneider, O. J. Boisseau, P. Haase, E. Slooten, and S. M. Dawson, “The bottlenose dolphin community of doubtful sound features a large proportion of long-lasting associations: can geographic isolation explain this unique trait?” Behavioral Ecology and Sociobiology, vol. 54, no. 4, pp. 396–405, 2003.
- D. E. Knuth, The Stanford GraphBase: A Platform for Combinatorial Computing, Addison-Wesley, Reading, Mass, USA, 1993.
- L. A. Adamic and N. Glance, “The political blogosphere and the 2004 US Election,” in Proceedings of the Workshop on the Weblogging Ecosystem (WWW '05), 2005.
- J. Leskovec, J. Kleinberg, and C. Faloutsos, “Graph evolution: densification and shrinking diameters,” ACM Transactions on Knowledge Discovery from Data, vol. 1, no. 1, Article ID 1217301, 2007.
- D. J. Watts and S. H. Strogatz, “Collective dynamics of ‘small-world’ networks,” Nature, vol. 393, no. 6684, pp. 440–442, 1998.
- B. Klimmt and Y. Yang, “Introducing the Enron corpus,” in Proceedings of the 1st Conference on Email and Anti-Spam (CEAS '04), 2004.
Copyright © 2014 Enming Dong et al. This is an open access article distributed under the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.