Browse > Graphs > Link Prediction

Link Prediction

65 papers with code · Graphs

Link prediction is a task to estimate the probability of links between nodes in a graph.

State-of-the-art leaderboards

Greatest papers with code

LINE: Large-scale Information Network Embedding

12 Mar 2015tangjianpku/LINE

This paper studies the problem of embedding very large information networks into low-dimensional vector spaces, which is useful in many tasks such as visualization, node classification, and link prediction. Most existing graph embedding methods do not scale for real world information networks which usually contain millions of nodes.

GRAPH EMBEDDING LINK PREDICTION NETWORK EMBEDDING NODE CLASSIFICATION

Graph Convolutional Matrix Completion

7 Jun 2017tkipf/gae

We consider matrix completion for recommender systems from the point of view of link prediction on graphs. Interaction data such as movie ratings can be represented by a bipartite user-item graph with labeled edges denoting observed ratings.

COLLABORATIVE FILTERING LINK PREDICTION MATRIX COMPLETION

Modeling Relational Data with Graph Convolutional Networks

17 Mar 2017tkipf/gae

Knowledge graphs enable a wide variety of applications, including question answering and information retrieval. We demonstrate the effectiveness of R-GCNs as a stand-alone model for entity classification.

KNOWLEDGE BASE COMPLETION KNOWLEDGE GRAPHS LINK PREDICTION

Variational Graph Auto-Encoders

21 Nov 2016tkipf/gae

We introduce the variational graph auto-encoder (VGAE), a framework for unsupervised learning on graph-structured data based on the variational auto-encoder (VAE). This model makes use of latent variables and is capable of learning interpretable latent representations for undirected graphs.

LINK PREDICTION

Higher-Order Factorization Machines

NeurIPS 2016 geffy/tffm

Factorization machines (FMs) are a supervised learning approach that can use second-order feature combinations even when the data is very high-dimensional. Unfortunately, despite increasing interest in FMs, there exists to date no efficient training algorithm for higher-order FMs (HOFMs).

LINK PREDICTION

Convolutional 2D Knowledge Graph Embeddings

5 Jul 2017TimDettmers/ConvE

In this work, we introduce ConvE, a multi-layer convolutional network model for link prediction, and report state-of-the-art results for several established datasets. We also show that the model is highly parameter efficient, yielding the same performance as DistMult and R-GCN with 8x and 17x fewer parameters.

KNOWLEDGE GRAPH EMBEDDINGS KNOWLEDGE GRAPHS LINK PREDICTION

GraphGAN: Graph Representation Learning with Generative Adversarial Nets

22 Nov 2017hwwang55/GraphGAN

The goal of graph representation learning is to embed each vertex in a graph into a low-dimensional vector space. Existing graph representation learning methods can be classified into two categories: generative models that learn the underlying connectivity distribution in the graph, and discriminative models that predict the probability of edge existence between a pair of vertices.

GRAPH REPRESENTATION LEARNING LINK PREDICTION NODE CLASSIFICATION

Combining Two And Three-Way Embeddings Models for Link Prediction in Knowledge Bases

2 Jun 2015glorotxa/SME

This paper tackles the problem of endogenous link prediction for Knowledge Base completion. Knowledge Bases can be represented as directed graphs whose nodes correspond to entities and edges to relationships.

KNOWLEDGE BASE COMPLETION LINK PREDICTION

Jack the Reader - A Machine Reading Framework

20 Jun 2018uclmr/jack

For example, in Question Answering, the supporting text can be newswire or Wikipedia articles; in Natural Language Inference, premises can be seen as the supporting text and hypotheses as questions. Providing a set of useful primitives operating in a single framework of related tasks would allow for expressive modelling, and easier model comparison and replication.

LINK PREDICTION NATURAL LANGUAGE INFERENCE QUESTION ANSWERING READING COMPREHENSION

Knowledge Graph Completion via Complex Tensor Factorization

22 Feb 2017ttrouill/complex

In statistical relational learning, knowledge graph completion deals with automatically understanding the structure of large knowledge graphs---labeled directed graphs---and predicting missing relationships---labeled edges. State-of-the-art embedding models propose different trade-offs between modeling expressiveness, and time and space complexity.

KNOWLEDGE GRAPH COMPLETION LINK PREDICTION RELATIONAL REASONING