作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程 ›› 2021, Vol. 47 ›› Issue (11): 69-76. doi: 10.19678/j.issn.1000-3428.0059892

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

基于评论文本图表示学习的推荐算法

杨粟森1, 刘勇2, 张举勇1   

  1. 1. 中国科学技术大学 数学科学学院, 合肥 230000;
    2. 南洋理工大学及英属哥伦比亚大学百合卓越联合研究中心, 新加坡 639798
  • 收稿日期:2020-11-03 修回日期:2020-12-04 发布日期:2020-12-09
  • 作者简介:杨粟森(1996-),男,硕士研究生,主研方向为推荐系统、深度学习、图神经网络;刘勇,研究员、博士;张举勇,副教授、博士。
  • 基金资助:
    国家自然科学基金(61976198,61672481)。

Recommendation Algorithm Based on Graph Representation Learning of Review Texts

YANG Susen1, LIU Yong2, ZHANG Juyong1   

  1. 1. School of Mathematical Sciences, University of Science and Technology of China, Hefei 230000, China;
    2. Joint NTU-UBC Research Centre of Excellence in Active Living for the Elderly, Singapore 639798, Singapore
  • Received:2020-11-03 Revised:2020-12-04 Published:2020-12-09

摘要: 基于卷积或循环神经网络的推荐系统主要捕捉评论文本中相邻词之间的局部和连续依赖关系,对长期、全局、非连续的依赖关系的捕捉能力有限。针对该问题,提出一种基于评论文本图表示学习的推荐算法RGP。将每个用户或项目的评论文本表示成图,图的节点为评论文本的词,图的边为词与词的连接关系。针对图中的每个节点,使用基于连接关系的图注意力网络加权融合其邻点信息,利用基于交互关系的注意力机制对节点重新赋权,并加权融合图中所有节点的表征从而得到整个图的表征。在此基础上,将基于用户和项目ID的嵌入表征及其评论图表征耦合输入并采用因子分解机进行评分预测,以得到最终的推荐结果。实验结果表明,与NARRE、DAML等算法相比,RGP算法可有效提高推荐精度。

关键词: 推荐算法, 评论文本, 图神经网络, 注意力机制, 因子分解机

Abstract: Previous recommendation systems based on Convolutional Neural Network(CNN) or Recurrent Neural Network(RNN) mainly capture the local and consecutive dependency between neighboring words in review texts. Therefore,they may not be effective in capturing the long-term,global and non-consecutive dependency.To deal with the problem,a recommendation algorithm,RGP,is proposed based on graph representation learning of review texts.For each individual user/item,the method builds a specific review graph where nodes represent the review words,and edges describe the connections between words.For each node in the graph,a connection-based graph attention network is used to weight and fuse its neighboring information.Then an interaction-based attention mechanism is proposed to reweight the nodes in the graph,and the representation of all the nodes in the graph is fused to obtain the graph representation.On this basis,the embedded representations based on user and item ID are coupled with their review graph representations for input,and scored by a factorization machine to obtain the recommendation results.Experimental results show that compared with NARRE,DAML and other algorithms,the proposed algorithm effectively improves the recommendation accuracy.

Key words: recommendation algorithm, review text, graph neural network, attention mechanism, Factorization Machine(FM)

中图分类号: