Node Classification Meets Link Prediction on Knowledge Graphs

by   Ralph Abboud, et al.

Node classification and link prediction are widely studied tasks in graph representation learning. While both transductive node classification and link prediction operate over a single input graph, they are studied in isolation so far, which leads to discrepancies. Node classification models take as input a graph with node features and incomplete node labels, and implicitly assume that the input graph is relationally complete, i.e., no edges are missing from the input graph. This is in sharp contrast with link prediction models that are solely motivated by the relational incompleteness of the input graph which does not have any node features. We propose a unifying perspective and study the problems of (i) transductive node classification over incomplete graphs and (ii) link prediction over graphs with node features. We propose an extension to an existing box embedding model, and show that this model is fully expressive, and can solve both of these tasks in an end-to-end fashion. To empirically evaluate our model, we construct a knowledge graph with node features, which is challenging both for node classification and link prediction. Our model performs very strongly when compared to the respective state-of-the-art models for node classification and link prediction on this dataset and shows the importance of a unified perspective for node classification and link prediction on knowledge graphs.


page 1

page 2

page 3

page 4


Learning to Make Predictions on Graphs with Autoencoders

We examine two fundamental tasks associated with graph representation le...

End to end learning and optimization on graphs

Real-world applications often combine learning and optimization problems...

Graph Star Net for Generalized Multi-Task Learning

In this work, we present graph star net (GraphStar), a novel and unified...

Effective and Efficient Network Embedding Initialization via Graph Partitioning

Network embedding has been intensively studied in the literature and wid...

Beyond Observed Connections : Link Injection

In this paper, we proposed the link injection, a novel method that helps...

A Theory of Link Prediction via Relational Weisfeiler-Leman

Graph neural networks are prominent models for representation learning o...

End-to-End Learning on Multimodal Knowledge Graphs

Knowledge graphs enable data scientists to learn end-to-end on heterogen...

Please sign up or login with your details

Forgot password? Click here to reset