计算机工程与应用 ›› 2018, Vol. 54 ›› Issue (19): 151-157.DOI: 10.3778/j.issn.1002-8331.1706-0165

• 模式识别与人工智能 • 上一篇    下一篇

融入用户社会关系的协同主题回归模型

胡检华1,李  平1,2   

  1. 1.长沙理工大学 计算机与通信工程学院,长沙 410114
    2.智能交通大数据处理湖南省重点实验室,长沙 410114
  • 出版日期:2018-10-01 发布日期:2018-10-19

User social relation incorporated into collaborative topic regression modeling

HU Jianhua1, LI Ping1,2   

  1. 1.School of Computer & Communication Engineering, Changsha University of Science & Technology, Changsha 410114, China
    2.Hunan Provincial Key Laboratory of Intelligent Processing of Big Data on Transportation, Changsha 410114, China
  • Online:2018-10-01 Published:2018-10-19

摘要: 协同主题回归(CTR)将概率矩阵分解和主题模型结合应用于推荐系统,在许多推荐应用中取得了成功,但该模型没有考虑用户社会关系对用户兴趣的影响。针对该问题,引入概率链接函数来评估社会关系网络对用户兴趣的影响,并以此约束目标函数。在CTR的基础之上,提出一种融入用户社会关系的协同主题回归模型(USRCTR),结合用户项目评分信息、项目内容和社会关系网络,构建一个基于分层贝叶斯模型的推荐引擎。在Lastfm数据集上实验表明,与其他几种CTR改进方法对比,该模型的训练时间更短,推荐精度更高。

关键词: 推荐系统, 协同过滤, 主题模型, 社会关系网络

Abstract: Collaborative Topic Regression(CTR) combines ideas of probabilistic matrix factorization and topic modeling for recommender systems, which has obtained successes in many application. But this model does not take into account the impact of user social relationships on user interests. In order to solve this problem, the probability link is incorporated in this algorithm to excavate the influence of social relation network on the latent structure for users, and it`s also in constraint on the object function. This paper proposes a novel hierarchical Bayesian model called User Social Relation incorporated into Collaborative Topic Regression(USRCTR), which extends CTR by integrating the user-item feedback information, item content information, and social relation network. Experiments on the Lastfm dataset show that this model can achieve better prediction accuracy than several improved methods of CTR with lower empirical training time.

Key words: recommender systems, collaborative filtering, topic models, social relations network