English

Towards Understanding Link Predictor Generalizability Under Distribution Shifts

Machine Learning 2025-07-17 v3 Artificial Intelligence

Abstract

State-of-the-art link prediction (LP) models demonstrate impressive benchmark results. However, popular benchmark datasets often assume that training, validation, and testing samples are representative of the overall dataset distribution. In real-world situations, this assumption is often incorrect; uncontrolled factors lead new dataset samples to come from a different distribution than training samples. Additionally, the majority of recent work with graph dataset shift focuses on node- and graph-level tasks, largely ignoring link-level tasks. To bridge this gap, we introduce a novel splitting strategy, known as LPShift, which utilizes structural properties to induce a controlled distribution shift. We verify LPShift's effect through empirical evaluation of SOTA LP models on 16 LPShift variants of original dataset splits, with results indicating drastic changes to model performance. Additional experiments demonstrate graph structure has a strong influence on the success of current generalization methods. Source Code Available Here: https://github.com/revolins/LPShift

Keywords

Cite

@article{arxiv.2406.08788,
  title  = {Towards Understanding Link Predictor Generalizability Under Distribution Shifts},
  author = {Jay Revolinsky and Harry Shomer and Jiliang Tang},
  journal= {arXiv preprint arXiv:2406.08788},
  year   = {2025}
}

Comments

KDD 25' Datasets & Benchmarks Track, 14 pages, 8 figures, 18 tables

R2 v1 2026-06-28T17:04:02.823Z