
摘要
推荐系统的一个标准模型是矩阵补全设置:给定一个部分已知的用户(行)对项目(列)的评分矩阵,推断未知的评分。在过去的几十年中,尝试使用神经网络处理这一目标的研究较少,但最近基于自动编码器的架构被证明是一种有前景的方法。在本文中,我们通过以下两方面对该架构进行了改进:(i) 使用适应于包含缺失值输入数据的损失函数;(ii) 引入辅助信息。实验结果表明,虽然辅助信息仅略微改善了所有用户/项目的测试误差平均值,但它对冷启动用户/项目的性能提升更为显著。
代码仓库
fstrub95/Autoencoders_cf
官方
pytorch
GitHub 中提及
jowoojun/collaborative_filtering_keras
GitHub 中提及
SJD1882/Big-Data-Recommender-Systems
GitHub 中提及
Recvani/benchmark
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| collaborative-filtering-on-douban | U-CFN | RMSE: 0.7049 |
| collaborative-filtering-on-douban | I-CFN | RMSE: 0.6911 |
| collaborative-filtering-on-movielens-10m | U-CFN | RMSE: 0.7954 |
| collaborative-filtering-on-movielens-10m | I-CFN | RMSE: 0.7767 |
| collaborative-filtering-on-movielens-1m | U-CFN | RMSE: 0.8574 |
| collaborative-filtering-on-movielens-1m | I-CFN | RMSE: 0.8321 |