Handling Class Imbalance in Link Prediction using Learning to Rank Techniques

11/13/2015
by   Bopeng Li, et al.
0

We consider the link prediction problem in a partially observed network, where the objective is to make predictions in the unobserved portion of the network. Many existing methods reduce link prediction to binary classification problem. However, the dominance of absent links in real world networks makes misclassification error a poor performance metric. Instead, researchers have argued for using ranking performance measures, like AUC, AP and NDCG, for evaluation. Our main contribution is to recast the link prediction problem as a learning to rank problem and use effective learning to rank techniques directly during training. This is in contrast to existing work that uses ranking measures only during evaluation. Our approach is able to deal with the class imbalance problem by using effective, scalable learning to rank techniques during training. Furthermore, our approach allows us to combine network topology and node features. As a demonstration of our general approach, we develop a link prediction method by optimizing the cross-entropy surrogate, originally used in the popular ListNet ranking algorithm. We conduct extensive experiments on publicly available co-authorship, citation and metabolic networks to demonstrate the merits of our method.

READ FULL TEXT
research
01/17/2022

SigGAN : Adversarial Model for Learning Signed Relationships in Networks

Signed link prediction in graphs is an important problem that has applic...
research
10/10/2019

Link Prediction via Deep Learning

Link prediction aims to infer the missing links or predicting future one...
research
12/06/2021

Pairwise Learning for Neural Link Prediction

In this paper, we aim at providing an effective Pairwise Learning Neural...
research
11/02/2016

Limitations and Alternatives for the Evaluation of Large-scale Link Prediction

Link prediction, the problem of identifying missing links among a set of...
research
01/29/2013

Link prediction for partially observed networks

Link prediction is one of the fundamental problems in network analysis. ...
research
07/23/2020

On a Bernoulli Autoregression Framework for Link Discovery and Prediction

We present a dynamic prediction framework for binary sequences that is b...
research
11/29/2020

Conditional Link Prediction of Category-Implicit Keypoint Detection

Keypoints of objects reflect their concise abstractions, while the corre...

Please sign up or login with your details

Forgot password? Click here to reset