arXiv:2411.01410cs.LGcs.AI2024-11NeurIPS被引 20

用强化学习思想解决链接预测中的探索与利用矛盾。

PageRank Bandits for Link Prediction

  • 将链接预测建模为序列决策问题,结合上下文老虎机与PageRank。
  • 在多个数据集上优于传统图模型和基于老虎机的方法。
  • 首次提供理论保证,适合动态推荐系统场景。

链接预测是图学习中的关键问题,广泛应用于推荐系统和知识图谱补全。现有方法多基于传统监督学习,难以适应用户兴趣变化,且无法有效平衡探索与利用。本文将链接预测重构为序列决策过程,提出首个融合上下文老虎机与PageRank的PRB算法,实现协同探索与利用。设计新型奖励机制并提供理论性能保证。在在线与离线设置下广泛评估,结果表明PRB显著优于基线方法。代码已开源。

原文摘要 · Abstract (English)

Link prediction is a critical problem in graph learning with broad applications such as recommender systems and knowledge graph completion. Numerous research efforts have been directed at solving this problem, including approaches based on similarity metrics and Graph Neural Networks (GNN). However, most existing solutions are still rooted in conventional supervised learning, which makes it challenging to adapt over time to changing customer interests and to address the inherent dilemma of exploitation versus exploration in link prediction. To tackle these challenges, this paper reformulates link prediction as a sequential decision-making process, where each link prediction interaction occurs sequentially. We propose a novel fusion algorithm, PRB (PageRank Bandits), which is the first to combine contextual bandits with PageRank for collaborative exploitation and exploration. We also introduce a new reward formulation and provide a theoretical performance guarantee for PRB. Finally, we extensively evaluate PRB in both online and offline settings, comparing it with bandit-based and graph-based methods. The empirical success of PRB demonstrates the value of the proposed fusion approach. Our code is released at https://github.com/jiaruzouu/PRB.

链接预测强化学习推荐系统

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。