We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

stat.ML

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Statistics > Machine Learning

Title: Asymptotics of $\ell_2$ Regularized Network Embeddings

Abstract: A common approach to solving prediction tasks on large networks, such as node classification or link prediction, begin by learning a Euclidean embedding of the nodes of the network, from which traditional machine learning methods can then be applied. This includes methods such as DeepWalk and node2vec, which learn embeddings by optimizing stochastic losses formed over subsamples of the graph at each iteration of stochastic gradient descent. In this paper, we study the effects of adding an $\ell_2$ penalty of the embedding vectors to the training loss of these types of methods. We prove that, under some exchangeability assumptions on the graph, this asymptotically leads to learning a graphon with a nuclear-norm-type penalty, and give guarantees for the asymptotic distribution of the learned embedding vectors. In particular, the exact form of the penalty depends on the choice of subsampling method used as part of stochastic gradient descent. We also illustrate empirically that concatenating node covariates to $\ell_2$ regularized node2vec embeddings leads to comparable, when not superior, performance to methods which incorporate node covariates and the network structure in a non-linear manner.
Comments: Accepted in Neural Information Processing Systems 2022. 44 pages, 2 figures, 2 tables
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG); Statistics Theory (math.ST)
Cite as: arXiv:2201.01689 [stat.ML]
  (or arXiv:2201.01689v3 [stat.ML] for this version)

Submission history

From: Andrew Davison [view email]
[v1] Wed, 5 Jan 2022 16:33:14 GMT (1326kb,D)
[v2] Tue, 31 May 2022 22:59:57 GMT (1391kb,D)
[v3] Sun, 18 Dec 2022 16:14:56 GMT (1512kb,D)

Link back to: arXiv, form interface, contact.