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

计算机工程 ›› 2021, Vol. 47 ›› Issue (8): 69-77. doi: 10.19678/j.issn.1000-3428.0058466

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

基于注意力机制与改进TF-IDF的推荐算法

李昆仑, 于志波, 翟利娜, 赵佳耀   

  1. 河北大学 电子信息工程学院, 河北 保定 071000
  • 收稿日期:2020-05-28 修回日期:2020-07-16 发布日期:2020-07-26
  • 作者简介:李昆仑(1962-),男,教授、博士,主研方向为模式识别、智能信息处理;于志波、翟利娜、赵佳耀,硕士研究生。
  • 基金资助:
    国家自然科学基金(61672205)。

Recommendation Algorithm Based on Attention Mechanism and Improved TF-IDF

LI Kunlun, YU Zhibo, ZHAI Lina, ZHAO Jiayao   

  1. College of Electronic and Information Engineering, Hebei University, Baoding, Hebei 071000, China
  • Received:2020-05-28 Revised:2020-07-16 Published:2020-07-26

摘要: 针对传统推荐系统主要依赖用户对物品的评分数据而无法学习到用户和项目的深层次特征的问题,提出基于注意力机制与改进TF-IDF的推荐算法(AMITI)。通过将双层注意力机制引入并行的神经网络推荐模型,提高模型对重要特征的挖掘能力。基于用户评分及项目类别改进TF-IDF,依据项目类别权重将推荐结果分类以构建不同类型的项目组并完成推荐。实验结果表明,AMITI算法能提高对文本中重要内容的关注度以及项目分配的注意力权重,有效提升推荐精度并在实现项目组推荐后改善推荐效果。

关键词: 多层感知机, 注意力机制, 卷积神经网络, 推荐算法, 深度学习

Abstract: Traditional recommendation systems rely heavily on item rating data of users, and fail to learn the deep-seated features of users and items. To address the problem, a recommendation algorithm based on attention mechanism and improved TF-IDF(AMITI) is proposed. A two-level attention mechanism is introduced into the parallel neural network recommendation model to raise its ability of mining key features. Then TF-IDF is improved based on rating data and item category. The recommendation results are classified according to the weight of each item category to build different types of item groups, completing the recommendation process. Experimental results show that the proposed algorithm improves the attention acquired by the important parts in a text, and can allocate different weights to different items. It significantly improves the recommendation accuracy and thus performance by implementing item group recommendation.

Key words: Multilayer Perceptron(MLP), attention mechanism, Convolution Neural Network(CNN), recommendation algorithm, Deep Learning(DL)

中图分类号: