在稀疏二分知识图中通过真实边重采样提升链接预测性能。
AEGIS: Authentic Edge Growth In Sparsity for Link Prediction in Edge-Sparse Bipartite Knowledge Graphs
- 仅重采样现有边,不新增节点,保持原始结构真实性。
- 语义KNN方法在Amazon、MovieLens和GDP数据集上显著提升AUC与降低Brier得分。
- 适合缺乏标注数据的垂直领域,尤其有文本描述节点时效果更优。
针对小众领域中的二分知识图谱普遍存在的数据稀疏问题,本文提出AEGIS(Authentic Edge Growth In Sparsity)——一种仅基于现有边进行重采样的增强框架,支持均匀或按逆度加权的重采样策略,保留原节点集,避免虚构端点。通过自然稀疏的‘游戏设计模式’图和高比例键渗透诱导的Amazon、MovieLens图验证方法有效性。评估采用AUC-ROC(越高越好)与Brier分数(越低越好),并使用双尾配对t检验对比稀疏基线。在Amazon和MovieLens上,复制型AEGIS表现持平,而语义KNN是唯一恢复AUC与校准性的方法;随机与合成边仍具破坏性。在文本丰富的GDP图中,语义KNN实现最大AUC提升与Brier分数下降,简单重采样亦优于稀疏对照组。结果表明,真实性约束的重采样是一种高效的数据增强策略,当节点描述信息丰富时,语义增强可带来额外增益。
原文摘要 · Abstract (English)
Bipartite knowledge graphs in niche domains are typically data-poor and edge-sparse, which hinders link prediction. We introduce AEGIS (Authentic Edge Growth In Sparsity), an edge-only augmentation framework that resamples existing training edges -either uniformly simple or with inverse-degree bias degree-aware -thereby preserving the original node set and sidestepping fabricated endpoints. To probe authenticity across regimes, we consider naturally sparse graphs (game design pattern's game-pattern network) and induce sparsity in denser benchmarks (Amazon, MovieLens) via high-rate bond percolation. We evaluate augmentations on two complementary metrics: AUC-ROC (higher is better) and the Brier score (lower is better), using two-tailed paired t-tests against sparse baselines. On Amazon and MovieLens, copy-based AEGIS variants match the baseline while the semantic KNN augmentation is the only method that restores AUC and calibration; random and synthetic edges remain detrimental. On the text-rich GDP graph, semantic KNN achieves the largest AUC improvement and Brier score reduction, and simple also lowers the Brier score relative to the sparse control. These findings position authenticity-constrained resampling as a data-efficient strategy for sparse bipartite link prediction, with semantic augmentation providing an additional boost when informative node descriptions are available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。