4 个月前

基于外积的神经协同过滤

基于外积的神经协同过滤

摘要

在本研究中,我们提出了一种新的多层神经网络架构,命名为ONCF(Outer Product-based Neural Collaborative Filtering),用于执行协同过滤。该方法的核心思想是利用外积显式建模嵌入空间维度之间的成对相关性。与现有的神经推荐模型通过简单的拼接或逐元素乘法结合用户嵌入和项目嵌入不同,我们在嵌入层之上使用外积的方法生成了一个二维交互图,这种交互图更具表现力且语义上更为合理。在外积生成的交互图之上,我们建议采用卷积神经网络来学习嵌入维度之间的高阶相关性。在两个公开的隐式反馈数据集上进行的大量实验表明,所提出的ONCF框架具有有效性,特别是使用外积建模多层神经推荐模型低层次中嵌入维度之间相关性的积极效果。实验代码可在以下地址获取:https://github.com/duxy-me/ConvNCF

代码仓库

基准测试

基准方法指标
recommendation-systems-on-gowallaConvNCF
nDCG@20: 0.6019
recommendation-systems-on-yelpConvNCF
NDCG@20: 0.1939

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于外积的神经协同过滤 | 论文 | HyperAI超神经