Pairwise Learning for Neural Link Prediction

12/06/2021
by   Zhitao Wang, et al.
0

In this paper, we aim at providing an effective Pairwise Learning Neural Link Prediction (PLNLP) framework. The framework treats link prediction as a pairwise learning to rank problem and consists of four main components, i.e., neighborhood encoder, link predictor, negative sampler and objective function. The framework is flexible that any generic graph neural convolution or link prediction specific neural architecture could be employed as neighborhood encoder. For link predictor, we design different scoring functions, which could be selected based on different types of graphs. In negative sampler, we provide several sampling strategies, which are problem specific. As for objective function, we propose to use an effective ranking loss, which approximately maximizes the standard ranking metric AUC. We evaluate the proposed PLNLP framework on 4 link property prediction datasets of Open Graph Benchmark, including ogbl-ddi, ogbl-collab, ogbl-ppa and ogbl-ciation2. PLNLP achieves Top 1 performance on ogbl-ddi, and Top 2 performance on ogbl-collab and ogbl-ciation2 only with basic neural architecture. The performance demonstrates the effectiveness of PLNLP.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
08/10/2022

Path-aware Siamese Graph Neural Network for Link Prediction

In this paper, we propose an algorithm of Path-aware Siamese Graph neura...
research
08/20/2020

A comparative study of similarity-based and GNN-based link prediction approaches

The task of inferring the missing links in a graph based on its current ...
research
11/13/2015

Handling Class Imbalance in Link Prediction using Learning to Rank Techniques

We consider the link prediction problem in a partially observed network,...
research
08/18/2020

NASE: Learning Knowledge Graph Embedding for Link Prediction via Neural Architecture Search

Link prediction is the task of predicting missing connections between en...
research
07/24/2017

Adversarial Sets for Regularising Neural Link Predictors

In adversarial training, a set of models learn together by pursuing comp...
research
02/02/2023

Neural Common Neighbor with Completion for Link Prediction

Despite its outstanding performance in various graph tasks, vanilla Mess...
research
11/19/2022

ReInform: Selecting paths with reinforcement learning for contextualized link prediction

We propose to use reinforcement learning to inform transformer-based con...

Please sign up or login with your details

Forgot password? Click here to reset