Link prediction using low-dimensional node embeddings: The measurement problem.

ABSTRACT: Graph representation learning is a fundamental technique for machine learning (ML) on complex networks. Given an input network, these methods represent the vertices by low-dimensional real-valued vectors. These vectors can be used for a multitude of downstream ML tasks. We study one of the most important such task, link prediction. Much of the recent literature on graph representation learning has shown remarkable success in link prediction. On closer investigation, we observe that the performance is measured by the AUC (area under the curve), which suffers biases. Since the ground truth in link prediction is sparse, we design a vertex-centric measure of performance, called the VCMPR@k plots. Under this measure, we show that link predictors using graph representations show poor scores. Des

SUBMITTER: Menand N 

PROVIDER: S-EPMC10895345 | biostudies-literature | 2024 Feb

REPOSITORIES: biostudies-literature