深度矩阵分解推荐算法
作者:
作者单位:

作者简介:

田震(1996-),男,学士,主要研究领域为推荐系统,机器学习.
潘腊梅(1995-),女,硕士,主要研究领域为推荐系统,机器学习.
尹朴(1997-),男,学士,主要研究领域为推荐系统.
王睿(1975-),男,博士,教授,博士生导师,主要研究领域为大数据,物联网,信息融合,边缘智能.

通讯作者:

王睿,E-mail:wangrui@ustb.edu.cn

中图分类号:

TP311

基金项目:

国家自然科学基金(62173158,61803391);北京科技大学顺德研究生院科技创新专项资金(BK19CF010,BK20BF012)


Deep Matrix Factorization Recommendation Algorithm
Author:
Affiliation:

Fund Project:

National Natural Science Foundation of China (62173158, 61803391); Scientific and Technological Innovation Foundation of Shunde Graduate School, USTB (BK19CF010, BK20BF012)

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    协同过滤推荐算法中的矩阵分解因其简单、易于实现,得到了广泛的应用.但是矩阵分解通过简单的线性内积建模用户和物品之间的非线性交互关系,限制了模型的表达能力.为此,He等人提出了广义矩阵分解模型,通过非线性激活函数和连接权重,将矩阵分解推广到广义矩阵分解,为模型赋予建模用户和物品间的二阶非线性交互关系的能力.但是广义矩阵分解模型是一个浅层模型,并不能很好地建模用户和物品间高阶交互关系,一定程度上可能会影响模型性能.受广义矩阵分解模型启发,提出了深度矩阵分解模型(deep matrix factorization,简称DMF),在广义矩阵分解模型的基础上引入隐藏层,利用深层神经网络来学习用户和物品间高阶交互关系.深度矩阵分解模型不仅解决了简单内积的线性问题,同时还能够建模用户和物品间的高阶交互,具有很好的表达能力.此外,在MovieLens和Anime两个数据集上进行了大量丰富的对比实验,验证了模型的可行性和有效性;同时,通过实验确定了模型的最优参数.

    Abstract:

    Matrix factorization in collaborative filtering recommendation algorithms is widely used because of its simplicity and facility of implementation, but matrix factorization utilizes a simple linear inner product to model the non-linear interaction between the user and the item, which limits the model's expressive power. He et al. proposed a generalized matrix factorization model, which extended the matrix factorization to the generalized matrix factorization through a non-linear activation function and connection weights, and gave the model the ability to model second-order non-linear interactions between users and items. Nevertheless, the generalized matrix factorization model is a shallow model and does not model the high-order interaction between users and items, which may affect the performance of the model to a certain extent. Inspired by the generalized matrix factorization model, this study proposes a deep matrix factorization model, abbreviated as DMF. Based on the generalized matrix factorization model, a hidden layer is introduced, and a deep neural network is used to learn the higher-order interaction between users and items. The deep matrix factorization model, which has a good expression ability, not only solves the linear problem of simple inner product, but also models high-order interactions between users and items. In addition, a lot of rich comparative experiments are performed on two datasets, MovieLens and Anime, and the results confirm its feasibility and effectiveness. Meanwhile the optimal parameters of the model were determined through experiments.

    参考文献
    相似文献
    引证文献
引用本文

田震,潘腊梅,尹朴,王睿.深度矩阵分解推荐算法.软件学报,2021,32(12):3917-3928

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2020-03-31
  • 最后修改日期:2020-06-11
  • 录用日期:
  • 在线发布日期: 2021-12-02
  • 出版日期: 2021-12-06
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京市海淀区中关村南四街4号,邮政编码:100190
电话:010-62562563 传真:010-62562533 Email:jos@iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号