arXiv:2503.16661cs.IRcs.LG2025-03被引 1

将ContextGNN纳入基准测试框架,评估其在推荐任务中的表现

ContextGNN goes to Elliot: Towards Benchmarking Relational Deep Learning for Static Link Prediction (aka Personalized Item Recommendation)

  • 将ContextGNN接入Elliot框架,实现可复现的对比实验
  • 在3个标准数据集上与6个SOTA GNN推荐模型对比,验证其性能潜力
  • 适合关注推荐系统可比性研究和模型评估的研究者

关系深度学习(RDL)利用消息传递图神经网络(GNN)的表示能力,在通过主键-外键关联的表上进行知识挖掘与预测任务。近年来,RDL已成功应用于推荐系统,其中代表性架构为ContextGNN。尽管该模型在真实世界推荐数据集上表现优异,但在亚马逊图书数据集上的静态链接预测任务(即个性化物品推荐)中,相较于其他先进GNN推荐系统仍存在提升空间。为此,本文将ContextGNN集成至Elliot框架,该框架支持50多个现有推荐模型的可复现基准分析。我们在三个标准推荐数据集上,与六种SOTA GNN推荐模型进行对比实验,结果与原论文观察一致。代码已开源:https://github.com/danielemalitesta/Rel-DeepLearning-RecSys。

原文摘要 · Abstract (English)

Relational deep learning (RDL) settles among the most exciting advances in machine learning for relational databases, leveraging the representational power of message passing graph neural networks (GNNs) to derive useful knowledge and run predicting tasks on tables connected through primary-to-foreign key links. The RDL paradigm has been successfully applied to recommendation lately, through its most recent representative deep learning architecture namely, ContextGNN. While acknowledging ContextGNN's improved performance on real-world recommendation datasets and tasks, preliminary tests for the more traditional static link prediction task (aka personalized item recommendation) on the popular Amazon Book dataset have demonstrated how ContextGNN has still room for improvement compared to other state-of-the-art GNN-based recommender systems. To this end, with this paper, we integrate ContextGNN within Elliot, a popular framework for reproducibility and benchmarking analyses, counting around 50 state-of-the-art recommendation models from the literature to date. On such basis, we run preliminary experiments on three standard recommendation datasets and against six state-of-the-art GNN-based recommender systems, confirming similar trends to those observed by the authors in their original paper. The code is publicly available on GitHub: https://github.com/danielemalitesta/Rel-DeepLearning-RecSys.

推荐系统图神经网络基准测试

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