Implicit degree bias in the link prediction task
Rachith Aiyappa, Xin Wang, Munjung Kim, Ozgur Can Seckin, Yong-Yeol Ahn, Sadamori Kojaku
Abstract
Link prediction-the task of distinguishing actual hidden edges from random unconnected node pairs-is a quintessential task in graph machine learning. Despite being widely accepted as a universal benchmark and a downstream task for representation learning, its validity is seldom questioned. Here, we show that the common edge sampling procedure in link prediction introduces an implicit bias toward high-degree nodes and produces a skewed evaluation that favors methods overly reliant on node degree, to the extent that a "null" method based solely on node degree can nearly match optimal performance. To address this, we propose a degree-corrected link prediction task that offers a more accurate assessment that aligns better with performance in recommendation tasks. Finally, we demonstrate that this degree-corrected benchmark can more effectively train graph machine-learning models by reducing overfitting to node degrees and facilitating the learning of relevant structures in graphs.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 796ff4b6-2e40-4f71-9eb9-a2bfe10d5773Builds on10
- Open Graph Benchmark: Datasets for Machine Learning on GraphsWeihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong et al.NeurIPS 2020 · 3,935 citations
- Tail-GNN: Tail-Node Graph Neural NetworksZemin Liu, Trung-Kien Nguyen, Yuan FangKDD 2021 · 105 citations
- On Generalized Degree Fairness in Graph Neural NetworksZemin Liu, Trung-Kien Nguyen, Yuan FangAAAI 2023 · 42 citations
- Revisiting Link Prediction: a data perspectiveHaitao Mao, Juanhui Li, Harry Shomer, Bingheng Li et al.ICLR 2024 · 40 citations
- Residual2Vec: Debiasing graph embedding with random graphsSadamori Kojaku, Jisung Yoon, Isabel Constantino, Yong-Yeol AhnNeurIPS 2021 · 29 citations
Related papers
- Theoretical and Empirical Insights into the Origins of Degree Bias in Graph Neural NetworksArjun Subramonian, Jian Kang, Yizhou SunNeurIPS 2024 · 15 citations
- Graph Neural Networks Use Graphs When They Shouldn'tMaya Bechler-Speicher, Ido Amos, Ran Gilad-Bachrach, Amir GlobersonICML 2024 · 27 citations
- Breaking the Dyadic Barrier: Rethinking Fairness in Link Prediction Beyond Demographic ParityJoão Mattos, Debolina Halder Lina, Arlei SilvaAAAI 2026 · 1 citation
- Understanding Negative Sampling in Graph Representation LearningZhen Yang, Ming Ding, Chang Zhou, Hongxia Yang et al.KDD 2020 · 172 citations
- TAM: Topology-Aware Margin Loss for Class-Imbalanced Node ClassificationJaeyun Song, Joonhyung Park, Eunho YangICML 2022 · 90 citations
