3 个月前

GHRS:基于图的混合推荐系统及其在电影推荐中的应用

GHRS:基于图的混合推荐系统及其在电影推荐中的应用

摘要

过去十年中,推荐系统研究迅速发展,为众多企业提升收入提供了重要支持。在学术与工业界,针对论文推荐系统的研究已涌现出多种方法。尽管大多数现有推荐系统主要依赖内容相似性或协同过滤策略,但近年来涌现出融合两种方法的混合型推荐机制,能够进一步提升推荐准确性。尽管已有大量算法基于此类方法提出,但仍有进一步优化的空间。本文提出一种基于图模型的推荐系统方法(GHRS),该方法结合用户评分相似性、用户人口统计信息及地理位置数据,构建综合特征表示。通过引入自编码器(Autoencoder)进行特征提取,我们从所有融合属性中生成新的高阶特征。利用这些新特征对用户进行聚类,所提出的GHRS方法在冷启动问题上表现出显著优势,性能全面超越现有主流方法。在MovieLens数据集上的实验结果表明,该算法在推荐准确率方面优于多种现有推荐算法。

代码仓库

hadoov/GHRS
官方
tf

基准测试

基准方法指标
collaborative-filtering-on-movielens-100kGHRS
Precision: 0.771
RMSE (u1 Splits): 0.887
Recall: 0.799
collaborative-filtering-on-movielens-1mGHRS
Precision: 0.792
RMSE: 0.838
movie-recommendation-on-movielens-1mGHRS
RMSE: 0.833

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
GHRS:基于图的混合推荐系统及其在电影推荐中的应用 | 论文 | HyperAI超神经