Journal of Physics: Complexity (Jan 2024)

Comparing discriminating abilities of evaluation metrics in link prediction

  • Xinshan Jiao,
  • Shuyan Wan,
  • Qian Liu,
  • Yilin Bi,
  • Yan-Li Lee,
  • En Xu,
  • Dong Hao,
  • Tao Zhou

DOI
https://doi.org/10.1088/2632-072X/ad46be
Journal volume & issue
Vol. 5, no. 2
p. 025014

Abstract

Read online

Link prediction aims to predict the potential existence of links between two unconnected nodes within a network based on the known topological characteristics. Evaluation metrics are used to assess the effectiveness of algorithms in link prediction. The discriminating ability of these evaluation metrics is vitally important for accurately evaluating link prediction algorithms. In this study, we propose an artificial network model, based on which one can adjust a single parameter to monotonically and continuously turn the prediction accuracy of the specifically designed link prediction algorithm. Building upon this foundation, we show a framework to depict the effectiveness of evaluating metrics by focusing on their discriminating ability. Specifically, a quantitative comparison in the abilities of correctly discerning varying prediction accuracies was conducted encompassing nine evaluation metrics: Precision, Recall, F1-Measure, Matthews correlation coefficient, balanced precision, the area under the receiver operating characteristic curve (AUC), the area under the precision-recall curve (AUPR), normalized discounted cumulative gain (NDCG), and the area under the magnified receiver operating characteristic. The results indicate that the discriminating abilities of the three metrics, AUC, AUPR, and NDCG, are significantly higher than those of other metrics.

Keywords