Journal of University of Science and Technology of China ›› 2019, Vol. 49 ›› Issue (1): 31-39.DOI: 10.3969/j.issn.0253-2778.2019.01.005

Previous Articles     Next Articles

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

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