4 个月前

协作转化度量学习

协作转化度量学习

摘要

最近,基于矩阵分解的推荐方法因违反三角不等式的问题而受到批评。尽管已提出几种基于度量学习的方法来克服这一问题,但现有的方法通常将每个用户映射到度量空间中的一个单一点,因此无法充分建模隐式反馈中用户-项目关系的强度和异质性。在本文中,我们提出了TransCF,以发现隐含在隐式用户-项目交互中的潜在用户-项目关系。受知识图谱嵌入中流行的翻译机制启发,我们通过利用用户和项目的邻域信息构建了特定于用户-项目的翻译向量,并根据用户与项目的关联关系将每个用户向项目进行翻译。我们的提议方法在七组真实世界数据集上的Top-N推荐任务中,最高可将命中率提高17%,优于几种最先进的方法。我们还对所提出的模型学到的翻译向量进行了广泛的定性评估,以验证采用翻译机制对于基于隐式反馈的推荐的有效性。

代码仓库

hkuds/recdiff
pytorch
GitHub 中提及
pcy1302/TransCF
官方
pytorch

基准测试

基准方法指标
collaborative-filtering-on-flixsterTransCF
Hits@10: 0.7309
Hits@20: 0.8374
nDCG@10: 0.4986
nDCG@20: 0.5257
recommendation-systems-on-amazon-caTransCF
Hits@10: 0.3436
Hits@20: 0.4658
nDCG@10: 0.2019
nDCG@20: 0.2323
recommendation-systems-on-book-crossing-1TransCF
Hits@10: 0.3329
Hits@20: 0.4744
nDCG@10: 0.1865
nDCG@20: 0.2221
recommendation-systems-on-ciaoTransCF
Hits@10: 0.2292
Hits@20: 0.374
nDCG@10: 0.1167
nDCG@20: 0.1525
recommendation-systems-on-decliciousTransCF
Hits@10: 0.2586
Hits@20: 0.3786
nDCG@10: 0.1475
nDCG@20: 0.1781
recommendation-systems-on-pinterestTransCF
Hits@10: 0.5504
Hits@20: 0.8108
nDCG@10: 0.258
nDCG@20: 0.3242
recommendation-systems-on-tradesyTransCF
Hits@10: 0.3198
Hits@20: 0.4505
nDCG@10: 0.1767
nDCG@20: 0.2095

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
协作转化度量学习 | 论文 | HyperAI超神经