论文总结《Neural Collaborative Filtering(NCF)》

原文链接
Neural Collaborative Filtering(NCF)

论文复现代码

Motivation

  1. 传统的矩阵分解通过latent features的内积来估计user与item的交互,而内积仅仅只能表示latent features的线性关系

contributions

  1. 引入MLP来替代latent features的内积,从而增加矩阵分解的非线性特征捕获能力
  2. 提出Neural matrix factorization model(NeuMF),即Linear Collaborative Filtering + Non-linear Collaborative Filtering

问题分析

传统矩阵分解的线性性质无法很好的捕获相似关系
论文中的MF的局限性的图可以很好解释为什么MF无法很好的捕获相似关系
在这里插入图片描述

可以看到,u4与u1最相似,其次是与u3,最后与u2,如果将其映射到二维的隐空间,那么应该有 c o s i n e ( u 4 , u 1 ) > c o s i n e ( u 4 , u 3 ) > c o s i n e ( u 4 , u 2 ) cosine(u_4, u_1) > cosine(u_4, u3) > cosine(u_4, u_2) cosine(u4,u1)>cosine(u4,u3)>cosine(u4,u2)
而如果u1,u2,u3在隐空间的向量如图所示,那么u4放哪都不是最合适的,这归因于MF的内积的线性性质

为了解决这个问题,论文提出利用MLP来捕获非线性性质

模型结构

在这里插入图片描述

模型图坐标的GMF Layer表示的是latent feature的element-wise product,即捕获latent features的线性性质
右图MLP Layers则表示利用MLP捕获latent features的非线性性质
最后GMF Layer与MLP Layer的结果进行concatenation后预测最终的结果

Loss Function

其将user与item的交互视为二元,即1表示交互,0表示没交互。因此论文将该问题视为二分类问题,因此利用的是二元交叉熵损失

EXPERIMENTS

数据集

MovieLens和Pinterest
在这里插入图片描述

为了防止测试的复杂度太大,其对每一个用户的负样本进行random sample出了100个,再从中做topk排序

不同Factors之间的效果对比

factors指模型结构中NeuMF Layer的维度
在这里插入图片描述

与baseline的对比

在这里插入图片描述

迭代次数损失metrics的变化以及负采样率不同时的metrics value

在这里插入图片描述

对GMF Layer和MLP Layer进行预训练和不预训练的对比

在这里插入图片描述

不同层数MLP的对比

在这里插入图片描述

思考

  1. 其只是考虑了user ID与item ID之间的关系,学习的是ID pair的分布,可以考虑加入user和item的上下文信息
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
SQLAlchemy 是一个 SQL 工具包和对象关系映射(ORM)库,用于 Python 编程语言。它提供了一个高级的 SQL 工具和对象关系映射工具,允许开发者以 Python 类和对象的形式操作数据库,而无需编写大量的 SQL 语句。SQLAlchemy 建立在 DBAPI 之上,支持多种数据库后端,如 SQLite, MySQL, PostgreSQL 等。 SQLAlchemy 的核心功能: 对象关系映射(ORM): SQLAlchemy 允许开发者使用 Python 类来表示数据库表,使用类的实例表示表中的行。 开发者可以定义类之间的关系(如一对多、多对多),SQLAlchemy 会自动处理这些关系在数据库中的映射。 通过 ORM,开发者可以像操作 Python 对象一样操作数据库,这大大简化了数据库操作的复杂性。 表达式语言: SQLAlchemy 提供了一个丰富的 SQL 表达式语言,允许开发者以 Python 表达式的方式编写复杂的 SQL 查询。 表达式语言提供了对 SQL 语句的灵活控制,同时保持了代码的可读性和可维护性。 数据库引擎和连接池: SQLAlchemy 支持多种数据库后端,并且为每种后端提供了对应的数据库引擎。 它还提供了连接池管理功能,以优化数据库连接的创建、使用和释放。 会话管理: SQLAlchemy 使用会话(Session)来管理对象的持久化状态。 会话提供了一个工作单元(unit of work)和身份映射(identity map)的概念,使得对象的状态管理和查询更加高效。 事件系统: SQLAlchemy 提供了一个事件系统,允许开发者在 ORM 的各个生命周期阶段插入自定义的钩子函数。 这使得开发者可以在对象加载、修改、删除等操作时执行额外的逻辑。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值