Learning to Recommend Links using Graph Structure and Node Content - Inria - Institut national de recherche en sciences et technologies du numérique Access content directly
Conference Papers Year : 2011

Learning to Recommend Links using Graph Structure and Node Content


The link prediction problem for graphs is a binary classification task that estimates the presence or absence of a link between two nodes in the graph. Links absent from the training set, however, cannot be directly considered as the negative examples since they might be present links at test time. Finding a hard decision boundary for link prediction is thus unnatural. This paper formalizes the link prediction problem from the flexible perspective of preference learning: the goal is to learn a preference score between any two nodes---either observed in the network at training time or to appear only later in the test---by using the feature vectors of the nodes and the structure of the graph as side information. Our assumption is that the observed edges, and in general, shortest paths between nodes in the graph, can reinforce an existing similarity between the nodes feature vectors. We propose a model implemented by a simple neural network architecture and an objective function that can be optimized by stochastic gradient descent over appropriate triplets of nodes in the graph. Our first preliminary experiments in small undirected graphs show that our learning algorithm outperforms baselines in real networks and is able to learn the correct distance function in synthetic networks.
Fichier principal
Vignette du fichier
lp-v0.pdf (173.05 Ko) Télécharger le fichier
Origin : Publisher files allowed on an open archive

Dates and versions

hal-00641419 , version 1 (15-11-2011)


  • HAL Id : hal-00641419 , version 1


Antonino Freno, Gemma C. Garriga, Mikaela Keller. Learning to Recommend Links using Graph Structure and Node Content. Neural Information Processing Systems Workshop on Choice Models and Preference Learning, Dec 2011, Granada, Spain. ⟨hal-00641419⟩
386 View
478 Download


Gmail Facebook X LinkedIn More