中国科学技术大学学报 ›› 2019, Vol. 49 ›› Issue (1): 31-39.DOI: 10.3969/j.issn.0253-2778.2019.01.005

• 原创论文 • 上一篇    下一篇

基于成对交互张量分解的标签推荐

鲁亚男   

  1. 中国科学技术大学计算机科学技术学院,安徽合肥 230027
  • 收稿日期:2018-02-13 修回日期:2018-05-15 出版日期:2019-01-31 发布日期:2019-01-31
  • 通讯作者: 鲁亚男
  • 作者简介:鲁亚男(通讯作者),女,1993年生,硕士生.研究方向:推荐系统、迁移学习. E-mail: yananlu@mail.ustc.edu.cn

Pairwise interaction tensor factorization based tag recommendation

LU Yanan   

  1. School of Computer Science and Technology, USTC, Hefei 230027, China
  • Received:2018-02-13 Revised:2018-05-15 Online:2019-01-31 Published:2019-01-31

摘要: 标签推荐系统是为目标用户推荐最可能用来标记某个资源的一系列标签.目前基于塔克分解模型,相比传统的FolkRank等算法具有更好的预测质量,但它本身的时间复杂度很高,很难适用于大中型数据集;而正则分解模型的时间复杂度虽然为线性,但预测质量并不高.针对上述问题,在改进塔克分解模型的基础上首先提出成对交互张量分解模型PITD.该模型仅考虑用户、资源和标签3个特征之间的部分两两交互关系,减少了无关信息对模型性能以及效率的影响.进而,利用贝叶斯个性化排序方法对PITD 模型进行推导,并设计了相应的优化算法.最后,在真实数据集上的广泛实验表明,PITD 模型比对比算法具有更好的推荐性能.

关键词: 推荐系统, 标签推荐, 张量分解, BPR

Abstract: The tag recommendation system is a series of tags that are most likely to be used to tag a resource for the target user. Currently, the Tucker decomposition model has better prediction quality than the traditional FolkRank algorithm, but it has high time complexity and is difficult to apply to large and medium-sized data sets. Although the time complexity of the regular decomposition model is linear, its prediction quality is not high. To solve these problems, firstly, the paired interaction tensor decomposition model PITD on the basis of improving the Tucker decomposition model is proposed. The model considers only some of the two-to-two interactions between the three characteristics of users, resources, and tags, reducing the impact of irrelevant information on model performance and efficiency. Then, the PITD model is deduced by Bayesian personalization method, and the corresponding optimization algorithm is designed. Finally, extensive experiments on real data sets show that the PITD model has better recommendation performance than the comparison algorithm.

Key words: recommendation system, tag recommendation, tensor decomposition, BPR