Learning to Recommend Links using Graph Structure and Node Content

Antonino Freno 1 Gemma Garriga 1, * Mikaela Keller 1
* Corresponding author
1 MOSTRARE - Modeling Tree Structures, Machine Learning, and Information Extraction
LIFL - Laboratoire d'Informatique Fondamentale de Lille, Inria Lille - Nord Europe
Abstract : 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.
Document type :
Conference papers
Complete list of metadatas

Cited literature [18 references]  Display  Hide  Download

https://hal.inria.fr/hal-00641419
Contributor : Mikaela Keller <>
Submitted on : Tuesday, November 15, 2011 - 5:00:11 PM
Last modification on : Thursday, February 21, 2019 - 10:52:49 AM
Long-term archiving on : Monday, December 5, 2016 - 10:54:12 AM

File

lp-v0.pdf
Publisher files allowed on an open archive

Identifiers

  • HAL Id : hal-00641419, version 1

Collections

Citation

Antonino Freno, Gemma 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⟩

Share

Metrics

Record views

543

Files downloads

581