4 个月前

推荐系统中的协同相似性嵌入

推荐系统中的协同相似性嵌入

摘要

我们提出了协作相似性嵌入(Collaborative Similarity Embedding, CSE),这是一种统一框架,旨在利用用户-项目二部图中丰富的协作关系进行表示学习和推荐。在所提出的框架中,我们将两种类型的邻近关系区分开来:直接邻近关系和k阶邻域邻近关系。前者的学习利用了从图中可观察到的直接用户-项目关联,而后者的学习则利用了隐含的关联,如用户-用户相似性和项目-项目相似性,这些信息在图较为稀疏时尤为宝贵。此外,为了提高可扩展性和灵活性,我们提出了一种专门设计用于捕捉这两种邻近关系的采样技术。在八个基准数据集上进行的大量实验表明,CSE相比最先进的推荐方法显著提升了性能。

代码仓库

bdnf/SBX-Recommendation-Engine
pytorch
GitHub 中提及
cnclabs/smore
GitHub 中提及

基准测试

基准方法指标
collaborative-filtering-on-netflixRATE-CSE
Recall@10: 0.2014
mAP@10: 0.1039
recommendation-systems-on-citeulikeRATE-CSE
Recall@10: 0.2362
mAP@10: 0.1452
recommendation-systems-on-echonestRANK-CSE
Recall@10: 0.1358
mAP@10: 0.0679
recommendation-systems-on-epinions-extendRANK-CSE
Recall@10: 0.1767
mAP@10: 0.0921
recommendation-systems-on-frappeRATE-CSE
Recall@10: 33.47
mAP@10: 0.2047
recommendation-systems-on-lastfm-360kRANK-CSE
Recall@10: 0.1762
mAP@10: 0.097
recommendation-systems-on-movielens-latestRATE-CSE
Recall@10: 0.3225
mAP@10: 0.199

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
推荐系统中的协同相似性嵌入 | 论文 | HyperAI超神经