首页 | 本学科首页   官方微博 | 高级检索  
     

深度矩阵分解推荐算法
引用本文:田震,潘腊梅,尹朴,王睿.深度矩阵分解推荐算法[J].软件学报,2021,32(12):3917-3928.
作者姓名:田震  潘腊梅  尹朴  王睿
作者单位:北京科技大学 计算机与通信工程学院,北京 100083;北京科技大学 计算机与通信工程学院,北京 100083;北京科技大学 顺德研究生院,广东 佛山 528300
基金项目:国家自然科学基金(62173158,61803391);北京科技大学顺德研究生院科技创新专项资金(BK19CF010,BK20BF012)
摘    要:协同过滤推荐算法中的矩阵分解因其简单、易于实现,得到了广泛的应用.但是矩阵分解通过简单的线性内积建模用户和物品之间的非线性交互关系,限制了模型的表达能力.为此,He等人提出了广义矩阵分解模型,通过非线性激活函数和连接权重,将矩阵分解推广到广义矩阵分解,为模型赋予建模用户和物品间的二阶非线性交互关系的能力.但是广义矩阵分解模型是一个浅层模型,并不能很好地建模用户和物品间高阶交互关系,一定程度上可能会影响模型性能.受广义矩阵分解模型启发,提出了深度矩阵分解模型(deep matrix factorization,简称DMF),在广义矩阵分解模型的基础上引入隐藏层,利用深层神经网络来学习用户和物品间高阶交互关系.深度矩阵分解模型不仅解决了简单内积的线性问题,同时还能够建模用户和物品间的高阶交互,具有很好的表达能力.此外,在MovieLens和Anime两个数据集上进行了大量丰富的对比实验,验证了模型的可行性和有效性;同时,通过实验确定了模型的最优参数.

关 键 词:协同过滤  线性内积  广义矩阵分解  隐藏层  高阶交互
收稿时间:2020/3/31 0:00:00
修稿时间:2020/6/11 0:00:00

Deep Matrix Factorization Recommendation Algorithm
TIAN Zhen,PAN La-Mei,YIN Pu,WANG Rui.Deep Matrix Factorization Recommendation Algorithm[J].Journal of Software,2021,32(12):3917-3928.
Authors:TIAN Zhen  PAN La-Mei  YIN Pu  WANG Rui
Affiliation:School of Computer & Communication Engineering, University of Science and Technology Beijing, Beijing 100083, China; School of Computer & Communication Engineering, University of Science and Technology Beijing, Beijing 100083, China;Shunde Graduate School, University of Science and Technology Beijing, Foshan 528300, China
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.
Keywords:collaborative filtering  linear inner product  generalized matrix factorization  hidden layers  high-order interaction
本文献已被 万方数据 等数据库收录!
点击此处可从《软件学报》浏览原始摘要信息
点击此处可从《软件学报》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号