Title: FAST NODE EMBEDDINGS: LEARNING EGO-CENTRIC REPRESENTATIONS

Abstract: Representation learning is one of the foundations of Deep Learning and allowed important improvements on several Machine Learning tasks, such as Neural Machine Translation, Question Answering and Speech Recognition. Recent works have proposed new methods for learning representations for nodes and edges in graphs. Several of these methods are based on the SkipGram algorithm, and they usually process a large number of multi-hop neighbors in order to produce the context from which node representations are learned. In this paper, we propose an effective and also efficient method for generating node embeddings in graphs that employs a restricted number of permutations over the immediate neighborhood of a node as context to generate its representation, thus ego-centric representations. We present a thorough evaluation showing that our method outperforms state-of-the-art methods in six different datasets related to the problems of link prediction and node classification, being one to three orders of magnitude faster than baselines when generating node embeddings for very large graphs.

1. INTRODUCTION:

Many important problems involving graphs require the use of learning algorithms to make predictions about nodes and edges. These predictions and inferences on nodes and edges from a graph are typically done using classifiers with carefully engineered features (Grover & Leskovec, 2016) . These features, besides taking time and manual labor to be developed and acquired, usually do not generalize well to other problems or contexts.

The field of Natural Language Processing (NLP) has had many advances due to the use of algorithms that learn word representations, instead of manually extracted features. Originally proposed by Bengio et al. (2003) and commonly used with Word2Vec algorithms like CBOW and SkipGram (Mikolov et al., 2013a) , word embeddings are used in many state-of-the-art solutions for neural machine translation (Luong & Manning, 2016; Firat et al., 2016) , question answering (Andreas et al., 2016) and natural language generation (Wen et al., 2015) . Recent works have proposed new methods for learning representations for nodes and edges in graphs, based on random walks (Perozzi et al., 2014; Grover & Leskovec, 2016) or auto-encoding adjacency vectors (Wang et al., 2016) .

In this paper, we propose a new general purpose method for generating node embeddings in very large graphs, which we call Neighborhood Based Node Embeddings (or simply NBNE). NBNE is based on the SkipGram algorithm and uses nodes neighborhoods as contexts. NBNE outperforms state-of-the-art DeepWalk (Perozzi et al., 2014) and Node2Vec (Grover & Leskovec, 2016) for the tasks of link prediction and node classification on six collections, being one to three orders of magnitude faster. In this work, we considered DeepWalk and Node2Vec as baselines.

The main reason for this improvement on effectiveness and efficiency is that we concentrate learning on the "predictable" parts of the graph. A study by Facebook research (Edunov et al., 2016) found that each person in the world (at least among the people active on Facebook) is connected to every other person by an average 3.57 other people. In a graph of this magnitude and connectedness, learning node embeddings by maximizing the log-probability of predicting nearby nodes in a random walk (with a window size of 5) can be highly inefficient and make it 'harder' for the embeddings to be constructed, even if these random walks are biased like in Node2Vec. We suspect this can also make them more unstable, which would explain why they need more iterations before embedding convergence.

The main contributions of this work are:

• We present a new general purpose method that is more effective and more efficient than state-of-the-art methods for generating node embeddings in graphs.

• Experimental results in solving the link prediction and node classification problems for six graph sets show that our method outperforms, in terms of effectiveness and efficiency, the methods DeepWalk and Node2Vec.

• In spite of the fact that our method has the same time complexity of the two baselines DeepWalk and Node2Vec, we were able to improve the training time by one to three orders of magnitude, which is important when dealing with very large graphs. For instance, to learn node embeddings for a graph containing 317,080 nodes and 1,049,866 edges, collected from the DBLP1 repository (Yang & Leskovec, 2012) , NBNE took approximately 14m30s minutes, DeepWalk approximately 164m34s and Node2Vec approximately 3,285m59s (more than 54 hours).

• We provide a thorough evaluation of our method in real and synthetic graphs, motivating our choice for a semi supervised algorithm. Our method has a single tunable parameter (number of permutations, which will be explained later) that can be tuned at once on the training set to avoid overfitting the representations.

2. RELATED WORK:

The definition of node similarity and finding general purpose node and/or edge representations are non-trivial challenges (Lü & Zhou, 2011) . Many definitions of similarity in graphs use the notion of first and second order proximity. First-order proximity is the concept that connected nodes in a graph should have similar properties, while the second-order proximity indicates that nodes with similar neighborhoods should have common characteristics.

Some earlier works on finding these embeddings use various matrix representations of the graph, together with dimensionality reduction techniques, to obtain the nodes' representations (Roweis & Saul, 2000; Tenenbaum et al., 2000) . A problem with these approaches is that they usually depend on obtaining the matrix' eigenvectors, which is infeasible for large graphs (O(n 2.376 )) with the Coppersmith-Winograd algorithm (Coppersmith & Winograd, 1987) ). Recent techniques attempt to solve this problem by dynamically learning representations for nodes in a graph using non-linear techniques based either on first and second order proximities (Tang et al., 2015; Wang et al., 2016) or random walks (Perozzi et al., 2014; Grover & Leskovec, 2016) .

Other recent works focus on finding representations for specific types of graphs. TriDNR (Pan et al., 2016) uses a graph structure together with node content and labels to learn node representations in two citation networks. Their work can be directly applied to any graph where nodes have labels and/or text contents. TEKE (Wang & Li, 2016) and KR-EAR (Lin et al., 2016) find representations for entities in knowledge graphs and metapath2vec (Dong et al., 2017) finds node representations in heterogeneous networks. The method LINE (Tang et al., 2015) finds a d dimensional representation for each node based on first and second-order graph proximities, not being feasible for large graphs, because its cost function depends on the whole adjacency matrix (O(|V | 2 )).

Another method, Structural Deep Network Embedding (SDNE) (Wang et al., 2016) , is also based on first and second order proximities. It uses autoencoders to learn a compact representation for nodes based on their adjacency matrix (second-order proximity), while forcing representations of connected nodes to be similar (first-order proximity) by using an hybrid cost function. SDNE is also not feasible for large graphs, since the autoenconders are trained on the complete adjacency vectors. Each vector has size O(|V |) and is created at least once, creating a lower bound on time complexity

O(|V | 2 ).

The method DeepWalk (Perozzi et al., 2014) generates k random walks starting on each vertex in the graph to create sentences where each "word" is a node. These sentences are then trained using the SkipGram algorithm to generate node embeddings. This method has a time complexity bounded by O(|V | log |V |).

Node2Vec (Grover & Leskovec, 2016 ) also uses random walks with SkipGram and can be seen as a generalization of DeepWalk. The difference between the two methods is that Node2Vec's walks are random, but biased by two pre-assigned parameters p and q. During the creation of the walks, these parameters are used to increase the chance of the walk returning to a parent node or going farther from it. This method uses a semi-supervised approach which requires several models to be generated and a small sample of labeled nodes to be used so that the best parameters p and q can be chosen. Node2Vec is not efficient for densely connected graphs, since its time and memory dependencies on the graph's branching factor b are O(b 2 ).

In this work, we considered DeepWalk (Perozzi et al., 2014) and Node2Vec (Grover & Leskovec, 2016) 

3. NEIGHBORHOOD BASED NODE EMBEDDINGS:

The context of a word is not a straightforward concept, but it is usually approximated by the words surrounding it. In graphs, a node's context is an even more complex concept. As explained above, DeepWalk and Node2Vec use random walks as sentences and consequently as contexts in which nodes appear.

In this work, the contexts are based solely on the neighborhoods of nodes, defined here as the nodes directly connected to it, focusing mainly on the second-order proximities. Consequently, nodes' representations will be mainly defined by their neighborhoods and nodes with similar neighborhoods (contexts) will be associated with similar representations.

3.1. SENTENCE GENERATION:

In our Neighborhood Based Node Embedding's (NBNE) method, as the name implies, sentences are created based on the neighborhoods of nodes. There are two main challenges in forming sentences from neighborhoods, as follows:

• A sentence containing all the neighbors from a specific highly connected root node might be of little use. Most neighbors would be distant from each other in the sentence, not influencing each other's representations, and not directly influencing the root node.

• There is no explicit order in the nodes in a neighborhood. So there is no clear way to choose the order in which they would appear in a sentence.

In this work, the solution is to form small sentences, with only k neighbors in each, using random permutations of these neighborhoods. Algorithm 1 presents the code for generating sentences. As a trade-off between training time and increasing the training dataset the user can select the number of permutations n. Selecting a higher value for n creates a more uniform distribution on possible neighborhood sentences, but also increases training time.

Algorithm 1 Sentence Sampling 1: procedure GETSENTENCES(graph, n)

2:

sentences ← [∅]

3:

for j in 0 : n do 4:

for node in graph.nodes() do 5:

neighbors ← random permutation(node.neighbors())

6:

for i in 0 : len(neighbors)/k do 7:

sentence

← [node] + neighbors[i • k : i • (k + 1)] 8:

sentences.append(sentence)

3.2. LEARNING REPRESENTATIONS:

As described in Section 3.1, Algorithm 1 forms a set of sentences S, where each word is actually a node from the graph. We train the vector representations of nodes by maximizing the log probability of predicting a node given another node in a sentence and given a set of representations r. We use a window of size k which is equal to the size of the generated sentences, so that each node in a sentence predicts all the others. The log probability maximized by NBNE is given by:

EQUATION

where p (s|r) is the probability of each sentence, which is given by:

log (p (s|r)) = 1 |s| i∈s   j∈s,j =i (log (p (v j |v i , r)))   (2)

where v i is a vertex in the graph and v j are the other vertices in the same sentence. The probabilities in this model are learned using the feature vectors r vi , which are then used as the vertex representations. The probability p (v j |v i , r) is given by:

EQUATION

where r T vj is the transposed output feature vector of vertex j, used to make predictions. The representations r v and r v are learned simultaneously by optimizing Equation (1). This is done using stochastic gradient ascent with negative sampling (Mikolov et al., 2013b) .

By optimizing this log probability, the algorithm maximizes the predictability of a neighbor given a node, creating node embeddings where nodes with similar neighborhoods have similar representations. Since there is more than one neighbor in each sentence, this model also makes connected nodes have similar representations, because they will both predict each others neighbors, resulting in representations also having some first order similarities. A trade-off between first and second order proximity can be achieved by changing the parameter k, which simultaneously controls both the size of sentences generated and the size of the window used in the SkipGram algorithm. A further discussion on this effect can be seen in Appendix B.3.

3.3. AVOIDING OVERFITTING REPRESENTATIONS:

When using large values of n (i.e., number of permutations) on graphs with few edges per node, some overfitting can be seen on the representations, as shown in details in Section 5.1 and in Appendix B.2. This overfitting can be avoided by sequentially training on increasing values of n and testing the embeddings on a validation set every few iterations, stopping when performance stops improving, as shown in Algorithm 2. (Yang & Leskovec, 2012) 317,080 1,049,866 3.31 -1 used in Link Prediction 2 used in Node Classification brief description of each dataset can be found in Appendix A and an analysis of their assortativity properties in Appendix B.1. We present results for the link prediction problem in Section 4.1 and for the node classification problem in Section 4.2. For all experiments we used sentences of size k = 5 and embeddings of size d = 128, while the number of permutations was run for n ∈ {1, 5, 10}. The best value of n was chosen according to the precision on the validation set and we used early stopping, as described in Section 3.3.

On both these tasks, DeepWalk and Node2Vec were used as baselines, having been trained and tested under the same conditions as NBNE and using the parameters as proposed in (Grover & Leskovec, 2016) . More specifically, we trained them with the same training, validation and test sets as NBNE and used a window size of 10 (k), walk length (l) of 80 and 10 runs per node (r). For Node2Vec, which is a semi-supervised algorithm, we tuned p and q on the validation set, doing a grid search on values p, q ∈ {0.25; 0.5; 1; 2; 4}. We also evaluated NBNE on two synthetic graphs with different sizes and sparseness, which can be seen on Appendix C, and an author name disambiguation task, on Appendix F. A comparison between NBNE and SDNE can be seen on Appendix D.

4.1. LINK PREDICTION:

Setup. Link prediction attempts to estimate the likelihood of the existence of a link between two nodes, based on observed links and the nodes' attributes (Lü & Zhou, 2011) . Typical approaches to this task use similarity metrics, such as Common Neighbors or Adamic-Adar (Adamic & Adar, 2003) . Instead of these hand made similarity metrics, we propose to train a logistic classifier based on the concatenation of the embeddings from both nodes that possibly form an edge and predict the existence or not of the edge.

To train NBNE on this task, we first obtained a sub-graph with 90% randomly select edges from each dataset, and obtained the node embeddings by training NBNE on this sub-graph. We, then, separated a small part of these sub-graph edges as a validation set, using the rest to train a logistic regression with the learned embeddings as features.

After the training was completed, the unused 10% of the edges were used as a test set to predict new links. 10-fold cross-validation was used on the entire training process to access the statistical significance of the results, analyzing statistical difference between the baselines and NBNE. To evaluate the results on this task, we used as metrics: AUC (area under the ROC curve) (Baeza-Yates & Ribeiro-Neto, 2011), and training time. 3 The logistic regressions were all trained and tested using all available edges (respectively in the training or test set), and an equal sized sample of negative samples, which, during training, included part of the 10% removed edges.

Results. Table 2 presents results for this task. Considering AUC scores on the Link Prediction task, NBNE was statistically better4 than both DeepWalk and Node2Vec on the Astro and PPI datasets, with more than 7% improvement, also showing a 4.67% performance gain in Wikipedia and a small, but statistically significant, gain on Blog. Only losing by a small percentage on Facebook, with a difference that was not statistically significant. In DBLP, NBNE again presents the best AUC score, although this difference was small and its statistical significance could not be verified due to the large training times of the baselines. This dataset contains the largest graph analyzed in this work (317,080 nodes and 1,049,866 edges) and in it, to train a single fold, Node2Vec took 3,285m59s (more than 54 hours) and DeepWalk took 164m34s (approximately 2 hours and 44 minutes), while NBNE took only 14m30s, which represents a 226/11 times improvement over Node2Vec and DeepWalk, respectively.

Considering training time for this task, NBNE presents the biggest improvements on sparser networks of medium size, like Astro, PPI and Wikipedia datasets. On these graphs, the best results are for n = 1, resulting in more than 50x faster training than DeepWalk and more than 1,500 times faster than Node2Vec, achieving a 6,049 times faster training than Node2Vec on Wikipedia. For the Blog and Facebook datasets the best results are for n = 5, resulting in larger training times, but still more than one order of magnitude faster than DeepWalk and more than 350 times faster than Node2Vec. For the DBLP dataset, the best results were achieved with n = 10, still much faster than the baselines.

4.2. NODE CLASSIFICATION:

Setup. Given a partially labeled graph, node classification is the task of inferring the classification of the unknown nodes, using the structure of the graph and/or the properties of the nodes. In this task, the node embeddings were trained using NBNE on the complete graph. After obtaining the node embeddings, 80% of the labeled nodes in the graph were used to train a logistic classifier that predicted the class of each node, while 5% of the nodes were used for validation and the remaining 15% nodes were used as a test set. This test was repeated for 10 different random seed initializations to access the statistical relevance of the results.

Results. Results on the Blog, PPI and Wikipedia datasets are shown in Table 3 and are presented in terms of Macro F1 scores and training times. NBNE produces statistically similar results to its baselines, in terms of Macro F1, on both PPI and Wikipedia, while showing a statistically significant 22.45% gain in the Blog dataset, indicating that NBNE's embeddings did not only get a better accuracy on Blog, but also that correct answers were better distributed across the 39 possible classes.

Considering training times, NBNE is more than 10 times faster than DeepWalk on these three datasets and is [300 ∼ 600] times faster than Node2Vec. NBNE didn't show statistically worse result in any dataset analyzed here5 , while having an order of magnitude faster training time than DeepWalk and more than two orders of magnitude faster training time than Node2Vec. 1 shows AUC scores versus embedding sizes for several values of n on the Facebook link prediction task. Quadratic functions approximating log(auc score) were plotted to allow for a better understanding of the results. It can be seen that for larger numbers of permutations (n > 100) results improve with embedding size, while for small ones (n = 1) they decrease. The plot also shows that n = 10 gives fairly robust values for all tested embedding sizes.

A further analysis can be seen in Table 4 , which shows that graphs with more edges per node tend to get better results with larger values of n, while graphs with a smaller branching factor have better results with only one permutation (n = 1). Other factors also enter into account when choosing n, like graph size. For example, link prediction on the DBLP graph had its best results for n = 10, although its branching size was only 3.31. Further experiments on this parameter can be seen in Appendices B.2 and C.1. 

5.2. TIME COMPLEXITY:

SkipGram's time complexity is linear on the number of sentences, embedding size (d) and logarithmic on the size of the vocabulary (Mikolov et al., 2013a) . Since the number of sentences is linear on the number of permutations (n), branching factor of the graph (b) and on the number of nodes, Figure 2 (Bottom-Right) shows that NBNE's time complexity is linear in the branching factor b, while Node2Vec's is quadratic. DeepWalk's running time is constant in this parameter, but for a graph with a larger branching factor, a higher number of walks per node should be used to train this algorithm, which would make it indirectly dependent on this factor.

6. CONCLUSIONS:

The proposed node embedding method NBNE shows results similar or better than the state-of-theart algorithms Node2Vec and DeepWalk on several different datasets. It shows promising results in two application scenarios: link prediction and node classification, while being efficient and easy to compute for large graphs, differently from other node embedding algorithms, such as LINE (Tang et al., 2015) or SDNE (Wang et al., 2016) . NBNE focuses learning on node's immediate neighbors, creating more ego-centric representations, which we suspect makes them more stable and faster to learn. Empirical results show that, although it has a similar time complexity, NBNE can be trained in a fraction of the time taken by DeepWalk (10 to 190 times faster) or Node2Vec (200 to 6,000 times faster), giving fairly robust results. Since embeddings are learned using only a node's immediate neighbors, we suspect it should also be easier to implement more stable asynchronous distributed algorithms to train them, and we leave this as future work.

A DATASETS:

We used a total of six graph datasets to evaluate NBNE, with DeepWalk and Node2Vec being used as baselines. Next we briefly describe these datasets:

1. Facebook (McAuley & Leskovec, 2012) : A snapshot of a subgraph of Facebook, where nodes represent users and edges represent friendships.

2. Astro (Leskovec et al., 2007) : A network that covers scientific collaborations between authors whose papers were submitted to the Astrophysics category in Arxiv.

3. Protein-Protein Interactions (PPI) (Breitkreutz et al., 2008) : We use the same subgraph of the PPI network for Homo Sapiens as in (Grover & Leskovec, 2016) . This subgraph contains nodes with labels from the hallmark gene sets (Liberzon et al., 2011) 

B ASSORTATIVITY:

Assortativity, also referred to as homophily in social network analysis, is a preference of nodes to attach themselves to others which are similar in some sense. In this section, we further investigate assortativity properties related to both the representations generated by our algorithm, as of the graphs themselves. In Section B.1, we do a quantitative analysis on the homophily inherent to the datasets considered in this work. In Section B.2, we make a qualitative analysis of how assortativity varies depending on the number of permutations n. In Section B.3, we make a qualitative analysis on the trade-off of first and second order proximities based on the choice of k.

B.1 DATASETS' HOMOPHILY:

There are several ways to quantitatively capture the homophily present in a graph. Jensen & Neville describe relational auto-correlation, which is Pearson's contingency coefficient on the characteristics of nodes which share edges (Jensen & Neville, 2002; Neville & Jensen, 2007) . Park & Barabási (2007) define dyadicity and heterophilicity, which respectively measure how a graph's nodes share common/different characteristics in edges, compared to a random model.

Table 5 presents both degree and label assortativity properties of the six graphs analysed here, calculated using the definition of Newman (2003) . We can see in this table that the datasets analyzed in this work cover a broad specter of assortativity properties. PPI, Wikipedia and Blog graphs present negative degree assortativity, which means nodes in these graphs are more likely to connect with nodes of different connectivity degrees. At the same time, Facebook, Astro and DBLP present positive degree assortativity, which indicates that their nodes tend to connect to others with similar degrees. (Newman, 2003) We also analyze graphs with both positive and negative label assortativity in our label classification task. While PPI and Blog datasets present positive label assortativity, with connected nodes more frequently sharing classes, Wikipedia has a negative assortativity, with its connected nodes being more likely to have different classes.

B.2 MORE ON THE NUMBER OF PERMUTATIONS (n):

Here, we further analyze how the number of permutations (n) influences both homophily and overfitting in our learned representations. We qualitatively measure homophily by comparing either cosine or euclidean distances between nodes on edges to the distances in non-edges.

The cosine distances for the PPI dataset, shown by the box plots in Figure 4 (top-left), clearly show for larger values of n how the embeddings overfit to the specific graph structure, with the learned similarity on edges not generalizing to the links which were previously removed. In this graph, for larger numbers of permutation the removed edges have a distribution more similar to the non edges than to the edges used during training, which is a tendency that can be observed in the other graphs, although in a smaller scale.

The box plots in Figure 4 (top-right) show the cosine distance for Facebook nodes. We can see that for n = 5 there is a larger separation between removed edges and non edges, which justifies the algorithm's choice of this value. For larger values of n we can again see an overlap between the distributions, caused by the embeddings overfitting. On the other hand, the cosine distances for the DBLP in Figure 4 (bottom-left) show the largest separation for n = 10.

Finally, the box plots in Figure 4 (bottom-right) show cosine distances for the Blog dataset. We can see that for n = 1 and n = 5 there is actually a larger cosine distance between nodes in removed edges than in non edges, with this situation only inverting for n ≥ 10. This happens due to this graph's negative degree homophily. This is also observed for similar graphs in the PPI and Wikipedia datasets, though with a smaller intensity in the PPI graph, which has a smaller absolute value of degree assortativity and where only embeddings for n = 1 present this property.

The box plots from Figure 4 further support our intuition that graphs with larger branching factors should have larger values of n. At the same time, this choice also depends on the graph size and structure, as shown by the algorithms choice of n = 10 for the DBLP dataset, which contains the largest degree assortativity. The best choice of n depends on the analyzed task, but we believe that, at least for link prediction, this choice is both directly proportional to a graph's branching size and degree assortativity. Nonetheless, the difficulty in analyzing these graphs supports our choice for a semi-supervised approach, automatically choosing n on a per graph instance. Considering again the experiment on the PPI dataset with the number of permutations n = 1 in Figure 4 (top-left), in Figure 5 we present in detail the euclidean distances between nodes that share or not an edge for this number of permutations. We can see that the distribution of removed edges is a lot closer to the edges used for training than to the non edges.

Kept Edge:

Removed Edge Non Edge Connection type 

B.3 TRADE-OFF BETWEEN FIRST AND SECOND ORDER PROXIMITY:

The window size and the number of neighbors in a sentence are both adjusted by a single variable, k, and this variable also controls a trade-off between first and second order proximities in the node embeddings. This can be explained intuitively by analyzing both the sentence sampling method in Algorithm 1 and Equations 1, 2 and 3, in Section 3.2.

When a smaller k is chosen, each node's embedding r vi will be mainly used to predict its own neighbors. This causes nodes with shared neighbors to have closer representations (second order proximity). When larger values of k are chosen, nodes will appear more often in its neighbors sentences, and will predict not only its own neighbors, but his neighbors' neighbors. This will result in connected nodes to have more similar embeddings, increasing first order similarity.

We further analyze this by examining the distribution of cosine distances between nodes at different graph distances. For this analysis, we use three different synthetic graphs: Barabási-Albert (Barabási & Albert, 1999) ; Erdõs-Rényi (Erdos & Rényi, 1960) ; Watts-Strogatz (Watts & Strogatz, 1998) . We choose these graphs because of their structural differences, believing they cover an ample specter of different graphs' properties. These graphs were created with |V | = 2000 and b = 20, and Watts-Strogatz graphs had a probability β = 0.2 of generating non-lattice edges. To train our representations we used n = 10 and d = 128.

Figure 6 shows box plots of these cosine distances of nodes' representations versus their graph distance on these different artificial random graphs. In this figure, we can see that, for both Barabàsi-Albert and Erdõs-Rényi graphs, when using a sentence size (k) equal to 1, the cosine similarity is larger for nodes which are two steps away than for nodes which share an edge (second order proximity), while for larger values of k, nodes which share an edge have larger similarity (first order proximity). The box plots in Figure 6 also show that the difference in similarity increases with the value of k.

The larger the value of k, the larger the difference between similarities of nodes which share an edge and nodes with larger distances, as can be seen in detail in Figure 7 for the Barabási-Albert graph.

C GRAPH SIZE AND SPARSENESS ANALYSIS:

In this section, we analyze how a graph's sparseness (represented here by its branching factor) and size (represented here by its number of vertices) affect the choice of the number of permutations (n) and of the window/sentence size (k). With this purpose we ran several link prediction experiments on two different synthetic graphs: Watts-Stogratz and Barabási-Albert. 6 These graphs were generated for different sizes (|V |) and sparseness (b), and we ran experiments with the same setup as in Section 4.1, with Watts-Stogratz graphs having again β = 0.2.7 Section C.1 presents this analysis for the 

C.1 NUMBER OF PERMUTATIONS (n):

We analyze, in this section, how a graph's size and sparseness affect the choice of the number of permutations (n), for both Watts-Stogratz and Barabási-Albert graphs. Analyzing the graphs in Figure 8 , we see a correlation between the best choice of n and a graph's number of vertices (|V |) and branching factor (b). In Figure 8a , which contains the experiments in the most sparse graphs, results for n = 1 are better for all graph sizes. A random algorithm would return an AUC score of 0.5, so results bellow this value clearly expose a problem in the learning algorithm. This is the case for both n = 10 and n = 5 in these graphs, which overfit its representations. In Figure 8b we can see that, when considering a graph with a branching size of b = 4, for smaller graphs a smaller value of n is preferable, while for larger graphs a larger number of permutations gives better results (n = 10). In Figure8c we can see that, for a branching size of b = 8, results for larger values of n are always better than for n = 1. Notice also that, while results for b = 2 and b = 4 were around 0.55 ∼ 0.7, results for b = 8 are closer to 0.9, showing that this algorithm is better at learning with more information.

Our experiments in link prediction using synthetic Barabási-Albert graphs present slightly more complex results. Figure 9 shows that for smaller branching factors (b ≤ 8), n = 1 indeed generate better results for small graphs, but for larger graphs, a larger number of permutations is necessary.

For intermediary branch sizes the best value of n is harder to determine, and only for b = 64 we start to see a tendency of larger number of permutations consistently giving better results. We can also see from Figure 9 that edges in Barabási-Albert graphs are considerably more difficult to predict, specially for smaller branching sizes. Most of our results are around 60% and our best AUC scores in these graphs are around 70%.

Again, n's dependency on these graph properties (|V | and b) depends highly on the graph's structure, further supporting our choice of a semi-supervised approach, choosing n on a per graph instance by validating results on a small validation set. This can be considered as a form of early stopping when training these node embeddings.

C.2 WINDOW AND SENTENCE SIZES (k):

In this section, we again use Watts-Stogratz and Barabási-Albert graphs, this time to analyze how a graph's size and sparseness affect results for different window and sentence sizes (k) in our model.

For these experiments we keep n = 5 fixed.

Figure 10a shows that, for a small branching factor (b = 2), all choices of k clearly overfit for Watts-Strogatz graphs, but k = 5 overfits less than larger choices of k. For b = 8, k = 5 produces slightly better results in these graphs, while larger values of k produce better results for a larger branching size (b = 32). Barabási-Albert graphs' edges are considerably harder for our algorithm to predict, as shown in the previous section, so we only report results for larger values of b (the algorithm, with our choice of hyper-parameters, overfits for smaller values). We can see from Figure 11 that larger values of k usually produce better results for this graph, but are more propense to overfit, specially when being applied to larger sparse graphs (|V | ≥ 800 and b = 16). Further analysis on the representations' properties for different values of k could provide better motivation on its choice, but we leave this to future studies, keeping our choice of k = 5 constant in this work. Studying if geometric operations between representations have comprehensible meanings would also be interesting, such as was done for Word2Vec algorithms, but this is also left as future work.

D COMPARISON WITH SDNE:

Structural Deep Network Embedding (Wang et al., 2016) is another algorithm used for learning node embeddings in graphs. As described in Section 2, SDNE is based on first and second order proximities, using autoencoders to learn compact representations based on a node's adjacency matrix (second-order proximity), while forcing representations of connected nodes to be similar (first-order proximity) by using an hybrid cost function.

This algorithm has a time complexity of O(|V | 2 ), but its main computation, which is calculating the gradients of its cost function and updating model parameters, can be highly parallelized by using modern GPUs and Deep Learning frameworks. In this section, we compare NBNE and SDNE in terms of both efficiency and efficacy, analysing both AUC/Macro F1 scores and training time. With this objective, we trained SDNE embeddings using both a dedicated K40 GPU with CUDA 8.0 and a dedicated 16 core linux server.8 

In their original work, SDNE was run in a semi-supervised setting, finding the best value of α, β and ν by tuning them on a small validation set. In this work we fix α = 0.2 and β = 10, since in their work they state that these values commonly give the best results, while only choosing ν in a semi-supervised manner. We use SDNE's architecture with [10,300; 1,000; 128] nodes on each layer and test it on both Link Prediction and Node Classification tasks, using the same steps described in Sections 4.1 and 4.2. We train these embeddings using ν ∈ {0.1, 0.01, 0.001} and choose the best value on the same validation sets used to tune n for NBNE and p and q for Node2vec. 7 shows the results of running NBNE and SDNE on the Node Classification task. On this task NBNE gave statistically better results on two datasets, with an impressive gain of 29.27% on PPI and 46.94% on Blog, only losing on Wikipedia with an also large gain of -20.20%. We can again see that NBNE has a more than an order of magnitude faster training time than SDNE on a GPU in this dataset, being more than two orders of magnitude faster when SDNE is trained on a CPU. 

E FULL RESULTS:

In this section, we extend the results presented in Section 4, considering now the precision on the training and test sets. We present results for the link prediction problem in Table 8 and for the node classification problem in Table 9 .

NBNE produces statistically similar results to its baselines, in terms of Macro F1, on the node classification task using both PPI and Wikipedia datasets, while showing a statistically significant 

:

http://dblp.uni-trier.de

Link to NBNE code will be made available in final publication.

Training times were all obtained using 16 core processors, running NBNE, Node2Vec or DeepWalk on 12 threads, with all algorithms having been implemented using gensim( Řehůřek & Sojka, 2010). More detailed results including precision on the training and test sets can be seen in Appendix E.

In all experiments we performed Welch's t-tests with p = 0.01. The symbol * marks results which are statistically different from NBNE.

Except for test precision on Wikipedia, losing to DeepWalk. For more details, see Appendix E

Erdõs-Rényi graphs weren't analyzed in this section because, since they have a completely random structure, its removed edges would be unpredictable.

7 Results presented in this section are all averages of ten cross-validation executions in a single instance of each graph size.

SDNE code was implemented using Tensorflow(Abadi et al., 2015)

We tried running SDNE with the DBLP dataset, but after five days it hadn't reached half of the training, so we stopped it.

Figures and Tables:

Figure 2: (Color online) (Top) Facebook dataset with 30% edges removed: Training times vs (Left) embedding size (Right) number of permutations. (Bottom) Randomly generated graphs: Training times vs (Left) number of vertices with b = 10 (Right) branching factor with |V | = 500.

Figure3shows the distribution of classes in the three datasets used for node classification. While Wikipedia has a long tailed distribution, with one class being present in almost 50% of its nodes, PPI's probabilities are well distributed along the 49 different possible classes.

Figure 4: (Color online) Cosine distances on the: PPI dataset (top-left); Facebook dataset (top-right); DBLP dataset (bottom-left); Blog dataset (bottom-right). All graphs had 10% of edges removes.

Figure 5: (Color online) Euclidean distances on PPI dataset for n = 1.

Figure 6: (Color online) NBNE features cosine similarities between nodes versus Graph Distance for different values of k for the graphs Barabási-Albert (left), Erdõs-Rényi (middle) and Watts-Strogatz (right).

Figure 7: (Color online) NBNE cosine vs graph distances in the Barabási-Albert graph.

Figure 8: (Color online) AUC Score vs Number of Vertices on a link prediction task on synthetic Watts-Strogatz graphs for different values of n.

Figure 9: (Color online) AUC Score vs Number of Vertices on a link prediction task for synthetic Barabási-Albert graphs for different values of n.

Figure 10: (Color online) AUC Score vs Number of Vertices on a link prediction task for synthetic Watts-Strogatz graphs for different values of k.

Figure 11: (Color online) AUC Score vs Number of Vertices on a link prediction task for synthetic Barabási-Albert graphs for different values of k.

as baselines, since they are scalable, having a time complexity (O(|V | log |V |)). The main differences between NBNE and the two baselines are: (i) we use a different sentence sampling strategy which is based in a node's neighborhood instead of random walks, (ii) NBNE is more effective than both Node2Vec and DeepWalk, as supported by our experiments in six different datasets, and (iii) NBNE is efficient for both dense and sparse graphs and scalable for very large applications, having a faster training time than both Node2Vec and DeepWalk.

Table content: <table/>

was evaluated on two different tasks: link prediction, and node classification.2 We used a total of six graph datasets to evaluate NBNE and Table1presents details about these datasets. A Statistics on the first six graph datasets

Table content: <table><tr><td colspan="2">Algorithm 2 NBNE without Overfitting</td></tr><tr><td colspan="2">1: procedure TRAINNBNE(graph, max n)</td></tr><tr><td>2:</td><td>sentences ← get sentences(graph, max n)</td></tr><tr><td>3:</td><td>model ← [initialize model()]</td></tr><tr><td>4: 5:</td><td>for j in 0 : log 2 (max n) do model ← train(model, sentences[2 j : 2 j+1 ])</td></tr><tr><td>6:</td><td>error ← test(new model, validation set)</td></tr><tr><td>7:</td><td>if error &gt; old error then</td></tr><tr><td>8:</td><td>break</td></tr><tr><td colspan="2">4 EXPERIMENTS</td></tr></table>

Link prediction results

Table content: <table><tr><td/><td colspan="2">Facebook</td><td colspan="2">Astro</td><td colspan="2">PPI</td></tr><tr><td/><td>AUC</td><td>Training Time</td><td>AUC</td><td>Training Time</td><td>AUC</td><td>Training Time</td></tr><tr><td>NBNE</td><td>0.9688</td><td>0m11s</td><td>0.8328</td><td>0m07s</td><td>0.8462</td><td>0m02s</td></tr><tr><td>DeepWalk</td><td>0.9730</td><td>2m26s</td><td>0.7548  *</td><td>6m55s</td><td>0.7741  *</td><td>2m30s</td></tr><tr><td>Node2vec</td><td>0.9762</td><td>69m33s</td><td colspan="2">0.7738  *  182m16s</td><td>0.7841  *</td><td>66m37s</td></tr><tr><td>Gain</td><td>-0.76%</td><td>12.96x 369.85x</td><td>7.62%</td><td>59.06x 1555.80x</td><td>7.91%</td><td>77.43x 2061.67x</td></tr><tr><td/><td colspan="2">Wikipedia</td><td/><td>Blog</td><td colspan="2">DBLP</td></tr><tr><td/><td>AUC</td><td>Training Time</td><td>AUC</td><td>Training Time</td><td>AUC</td><td>Training Time</td></tr><tr><td>NBNE</td><td>0.6853</td><td>0m02s</td><td>0.9375</td><td>1m11s</td><td>0.9335  †</td><td>14m30s</td></tr><tr><td colspan="2">DeepWalk 0.6534  *</td><td>7m38s</td><td>0.9098  *</td><td>28m13s</td><td>0.9242  ‡</td><td>164m34s</td></tr><tr><td colspan="3">Node2Vec 0.6547  *  236m60s</td><td colspan="2">0.9202  *  838m41s</td><td colspan="2">0.9322  ‡ 3,285m59s</td></tr><tr><td>Gain</td><td>4.67%</td><td>194.86x 6049.77x</td><td>1.88%</td><td>23.86x 709.24x</td><td>0.13%</td><td>11.34x 226.52x</td></tr><tr><td colspan="3">† average of 10 fold results</td><td/><td/><td/><td/></tr><tr><td colspan="6">‡ no statistical tests were run, due to the time necessary to run a single fold</td><td/></tr></table>

Node classification results

Table content: <table><tr><td/><td colspan="2">Blog</td><td>PPI</td><td/><td colspan="2">Wikipedia</td></tr><tr><td/><td>Macro F1</td><td>Training Time</td><td>Macro F1</td><td>Training Time</td><td>Macro F1</td><td>Training Time</td></tr><tr><td>NBNE</td><td>0.2004</td><td>1m57s</td><td>0.0978</td><td>0m16s</td><td>0.0727</td><td>0m41s</td></tr><tr><td>DeepWalk</td><td>0.1451  *</td><td>31m31s</td><td>0.0991</td><td>3m04s</td><td>0.0679</td><td>13m04s</td></tr><tr><td>Node2vec</td><td>0.1637  *</td><td>959m12s</td><td>0.0971</td><td>83m02s</td><td>0.0689</td><td>408m00s</td></tr><tr><td>Gain</td><td>22.45%</td><td>16.18x 492.57x</td><td>-1.35%</td><td>11.82x 319.78x</td><td>5.56%</td><td>19.04x 594.62x</td></tr><tr><td colspan="2">5 FURTHER ANALYSIS</td><td/><td/><td/><td/><td/></tr><tr><td colspan="3">5.1 NUMBER OF PERMUTATIONS (n)</td><td/><td/><td/><td/></tr><tr><td colspan="7">The quality of NBNE's embeddings depends on both the size of the embeddings (d) and the number</td></tr><tr><td colspan="7">10 3 AUC vs Embedding Size (Facebook) Embedding size (d) n=1 n=10 n=100 n=1000 of permutations (10 2 0.89 0.90 0.91 0.92 0.93 0.94 0.95 0.96 AUC (%) Figure</td></tr><tr><td/><td/><td/><td colspan="4">Figure 1: (Color online) NBNE AUC scores vs</td></tr><tr><td/><td/><td/><td colspan="4">embedding sizes on Facebook dataset with 50%</td></tr><tr><td/><td/><td/><td colspan="2">edges removes</td><td/><td/></tr></table>

Link Prediction results for varying n with NBNE

Table content: <table><tr><td/><td>PPI (9.95  † )</td><td/><td colspan="3">Facebook (21.84  † )</td><td/><td>Blog (32.38  † )</td></tr><tr><td>n</td><td>Precision Train Test</td><td>AUC</td><td colspan="2">Precision Train Test</td><td>AUC</td><td colspan="2">Precision Train Test</td><td>AUC</td></tr><tr><td colspan="8">10 0.7071 0.7108 0.7795 0.8453  ‡ 0.9061 0.9642 0.8771  ‡ 0.8627</td><td>0.9348  ‡</td></tr><tr><td>5</td><td colspan="7">0.7280 0.7305 0.8071 0.8408  ‡ 0.9070 0.9688 0.8775  ‡ 0.8681</td><td>0.9375  ‡</td></tr><tr><td>1</td><td colspan="2">0.7822 0.7751 0.8462</td><td>0.8036</td><td colspan="2">0.8410 0.9150</td><td>0.8115</td><td>0.8374</td><td>0.9146</td></tr><tr><td colspan="2">† Edges per node</td><td/><td/><td/><td/><td/></tr></table>

Datasets homophily information

Table content: <table><tr><td/><td colspan="2">Assortativity</td></tr><tr><td/><td>Degree 1</td><td>Label 1</td></tr><tr><td>Facebook</td><td>0.0635</td><td>-</td></tr><tr><td>Astro</td><td>0.2051</td><td>-</td></tr><tr><td colspan="2">PPI -0.0930</td><td>0.0533</td></tr><tr><td colspan="2">Wikipedia -0.2372</td><td>-0.0252</td></tr><tr><td colspan="2">Blog -0.2541</td><td>0.0515</td></tr><tr><td>DBLP</td><td>0.2665</td><td>-</td></tr><tr><td colspan="2">1 Calculated as in</td><td/></tr></table>

shows results using both NBNE and SDNE embeddings on Link Prediction tasks. In this table we can see that both algorithms produce similar results in terms of AUC scores, with each having a statistically significant better result on two datasets, and NBNE having a non statistically significant, but slightly better result on the fifth. It is clear that even when training SDNE using a K40 GPU, NBNE still has more than an order of magnitude faster training time on all datasets, being more than two orders of magnitude faster on most. When comparing to SDNE trained on a CPU, NBNE has more than three orders of magnitude faster training time. On Astro, the dataset with the largest number of nodes analyzed here, NBNE had a 2,009 times faster training time compared to SDNE on a GPU and 44,896 times faster compared to SDNE on CPU. 9

Table content: <table/>

Link prediction results with SDNE

Table content: <table><tr><td/><td colspan="2">Facebook</td><td/><td>Astro</td><td/><td>PPI</td></tr><tr><td/><td>AUC</td><td>Training Time</td><td>AUC</td><td>Training Time</td><td>AUC</td><td>Training Time</td></tr><tr><td>NBNE</td><td>0.9688</td><td>0m11s  ‡</td><td>0.8328</td><td>0m07s  ‡</td><td>0.8462</td><td>0m02s  ‡</td></tr><tr><td colspan="2">SDNE 0.9510  *</td><td>20m34s  † 242m10s  ‡</td><td>0.8157  *</td><td>234m24s  † 5,237m59s  ‡</td><td>0.8751  *</td><td>16m10s  † 232m01s  ‡</td></tr><tr><td>Gain</td><td>1.87%</td><td>112.21x 1,320.91x</td><td>2.10%</td><td>2,009.17x 44,896.96x</td><td>-3.30%</td><td>485.10x 6,960.34x</td></tr><tr><td/><td colspan="2">Wikipedia</td><td/><td>Blog</td><td colspan="2">DBLP</td></tr><tr><td/><td>AUC</td><td>Training Time</td><td>AUC</td><td>Training Time</td><td>AUC</td><td>Training Time</td></tr><tr><td>NBNE</td><td>0.6853</td><td>0m02s  ‡</td><td>0.9375</td><td>1m11s  ‡</td><td>0.9335</td><td>14m30s  ‡</td></tr><tr><td>SDNE</td><td>0.6781</td><td>22m23s  † 337m47s  ‡</td><td>0.9462  *</td><td>81m33s  † 1,492m47s  ‡</td><td>-</td><td>--</td></tr><tr><td>Gain</td><td>1.06%</td><td>671.59x 10,133.46x</td><td>-0.92%</td><td>68.92x 1,261.51x</td><td>-</td><td>--</td></tr><tr><td colspan="3">‡ Training time on CPU</td><td/><td/><td/><td/></tr><tr><td/><td/><td/><td/><td/><td/><td/></tr></table>

Node classifications results with SDNE Training time on CPU † Training time on GPU Analyzing both these tables we can also see that the largest gains in training time occur when using NBNE on a large but sparse network, such as Astro. This agrees with our theoretical expectations, since SDNE's time complexity grows quadratically with the number of nodes O(|V 2 |) and NBNE's grows with O(|V | • log(|V |) • b), which is close to linear on the number of nodes for large graphs.

Table content: <table><tr><td/><td/><td>Blog</td><td/><td>PPI</td><td colspan="2">Wikipedia</td></tr><tr><td/><td>Macro F1</td><td>Training Time</td><td>Macro F1</td><td>Training Time</td><td>Macro F1</td><td>Training Time</td></tr><tr><td>NBNE</td><td>0.2005</td><td>1m57s  ‡</td><td>0.0978</td><td>0m16s  ‡</td><td>0.0727</td><td>0m41s  ‡</td></tr><tr><td>SDNE</td><td>0.1364  *</td><td>96m48s  † 1,476m33s  ‡</td><td>0.0757  *</td><td>16m52s  † 231m04s  ‡</td><td>0.0911  *</td><td>19m60s  † 338m40s  ‡</td></tr><tr><td>Gain</td><td>46.94%</td><td>49.64x 757.20x</td><td>29.27%</td><td>63.24x 866.48x</td><td>-20.20%</td><td>29.26x 495.60x</td></tr><tr><td>‡</td><td/><td/><td/><td/><td/><td/></tr></table>

Link prediction complete results † average of 10 fold results ‡ no statistical tests were run, due to the time necessary to run one fold 22.45% gain in the Blog dataset. Node classification results in the PPI dataset had the smallest precision among all datasets, with only approximately 14%, but there are 49 classes in it and a classifier which always guessed the most common class would only get 2.95% precision.NBNE only shows a statistically worse result in test precision for node classification on the Wikipedia dataset, losing to DeepWalk, but having an order of magnitude faster training time than DeepWalk and more than two orders of magnitude faster training time than Node2Vec. On all other experiments in either node classification or link prediction it presented either statistically better or similar results to its baselines, while showing much faster training times.

Table content: <table><tr><td>Facebook</td><td colspan="2">Precision Train Test</td><td>AUC</td><td>Training Time</td></tr><tr><td>NBNE</td><td>0.8408</td><td>0.9070</td><td>0.9688</td><td>0m11s</td></tr><tr><td>DeepWalk</td><td>0.8770  *</td><td>0.9218</td><td>0.9730</td><td>2m26s</td></tr><tr><td>Node2vec</td><td>0.8844  *</td><td>0.9251</td><td>0.9762</td><td>69m33s</td></tr><tr><td>Gain</td><td>-4.93%</td><td>-1.95%</td><td>-0.76%</td><td>(12.96x, 369.85x)</td></tr><tr><td>Astro</td><td colspan="2">Precision Train Test</td><td>AUC</td><td>Training Times</td></tr><tr><td>NBNE</td><td>0.7640</td><td>0.7552</td><td>0.8328</td><td>0m07s</td></tr><tr><td>DeepWalk</td><td colspan="3">0.6957  *  0.6836  *  0.7548  *</td><td>6m55s</td></tr><tr><td>Node2vec</td><td colspan="3">0.7223  *  0.7163  *  0.7738  *</td><td>182m16s</td></tr><tr><td>Gain</td><td>5.78%</td><td>5.43%</td><td>7.62%</td><td>(59.06x, 1555.80x)</td></tr><tr><td>PPI</td><td colspan="2">Precision Train Test</td><td>AUC</td><td>Training Times</td></tr><tr><td>NBNE</td><td>0.7822</td><td>0.7751</td><td>0.8462</td><td>0m02s</td></tr><tr><td>DeepWalk</td><td colspan="3">0.7124  *  0.7078  *  0.7741  *</td><td>2m30s</td></tr><tr><td>Node2Vec</td><td colspan="3">0.7332  *  0.7253  *  0.7841  *</td><td>66m37s</td></tr><tr><td>Gain</td><td>6.69%</td><td>6.86%</td><td>7.91%</td><td>(77.43x, 2061.67x)</td></tr><tr><td>Wikipedia</td><td colspan="2">Precision Train Test</td><td>AUC</td><td>Training Times</td></tr><tr><td>NBNE</td><td>0.6823</td><td>0.6223</td><td>0.6853</td><td>0m02s</td></tr><tr><td>DeepWalk</td><td colspan="3">0.8024  *  0.5245  *  0.6534  *</td><td>7m38s</td></tr><tr><td>Node2Vec</td><td colspan="3">0.8129  *  0.5317  *  0.6547  *</td><td>236m60s</td></tr><tr><td>Gain</td><td colspan="2">-16.07% 17.04%</td><td>4.67%</td><td>(194.86x, 6049.77x)</td></tr><tr><td>Blog</td><td colspan="2">Precision Train Test</td><td>AUC</td><td>Training Times</td></tr><tr><td>NBNE</td><td>0.8775</td><td>0.8681</td><td>0.9375</td><td>1m11s</td></tr><tr><td>DeepWalk</td><td colspan="3">0.8560  *  0.8337  *  0.9098  *</td><td>28m13s</td></tr><tr><td>Node2Vec</td><td colspan="3">0.8664  *  0.8460  *  0.9202  *</td><td>838m41s</td></tr><tr><td>Gain</td><td>1.28%</td><td>2.61%</td><td>1.88%</td><td>(23.86x, 709.24x)</td></tr><tr><td>DBLP</td><td colspan="2">Precision Train Test</td><td>AUC</td><td>Training Times</td></tr><tr><td>NBNE</td><td>0.9724  †</td><td colspan="2">0.8781  † 0.9335  †</td><td>14m30s</td></tr><tr><td>DeepWalk</td><td>0.9344  ‡</td><td colspan="2">0.8369  ‡ 0.9242  ‡</td><td>164m34s</td></tr><tr><td>Node2Vec</td><td>0.9449  ‡</td><td colspan="2">0.8498  ‡ 0.9322  ‡</td><td>3,285m59s</td></tr><tr><td>Gain</td><td>2.91%</td><td>3.33%</td><td>0.13%</td><td>(11.34x, 226.52x)</td></tr><tr><td/><td/><td/><td/><td/></tr></table>

Peer review (iclr2018):
 The authors propose a method for learning node representations which, like previous work (e.g. node2vec, DeepWalk), is based on the skip-gram model. However, unlike previous work, they use the concept of shared neighborhood to define context rather than applying random walks on the graph. The paper is well-written and it is quite easy to follow along with the discussion. This work is most similar, in my opinion, to node2vec. In particular, when node2vec has its restart probability set pretty high, the random walks tend to stay within the local neighborhood (near the starting node). The main difference is in the sentence construction strategy. Whereas node2vec may sample walks that have context windows containing the same node, the proposed method does not as it uses a random permutation of a node*s neighbors. This is the main difference between the proposed method and node2vec/DeepWalk. Pros: Outperforms node2vec and DeepWalk on 5 of the 6 tested datasets and achieves comparable results on the last one. Proposes a simple yet effective way to sample walks from large graphs. Cons: The description on the experimental setup seems to lack some important details. See more detailed comments in the paragraph below. While LINE or SDNE, which the authors cite, may not run on some of the larger datasets they can be tested on the smaller datasets. It would be helpful if the authors tested against these methods as well. For instance, on page 5 footnote 4 the authors state that DeepWalk and node2vec are tested under similar conditions but do not elaborate. In NBDE, when k=5 a node u*s neighbors are randomly permuted and these are divided into subsets of five and concatenated with u to form sentences. Random walks in node2vec and DeepWalk can be longer, instead they use a sliding context window. For instance a sentence of length 10 with context window 5 gives 6 contexts. Do the authors account for this to ensure that skip-gram for all compared methods are tested using the same amount of information. Also, what exactly does the speedup in time mean. The discussion on this needs to be expounded.