融合Node2Vec和负反馈强化学习的商品推荐算法

RECOMMENDATION METHOD WITH NODE2VEC AND NEGATIVE FEEDBACK REINFORCEMENT LEARNING

  • 摘要: 目前推荐系统普遍存在长尾问题,导致商品推荐覆盖率低、多样性差,为此提出一种融合有偏随机游走(Node2Vec)和负反馈强化学习的商品推荐算法GES4RL(Graph Embedding with Side Information for Reinforcement Learning)。对商品传播的有向加权图使用Node2Vec算法学习商品的编码表示;引入门控循环单元(GRU)对用户偏好的动态情况进行建模,并使用基于负反馈强化学习模型计算出长尾商品的最佳推荐策略。在TianChi电商数据集上的实验表明,该算法显著提高了商品推荐的多样性和命中率。

     

    Abstract: The long-tail problem is very common in recommendation system. It leads to recommending few and homogeneous products. We propose a new recommendation algorithm named GES4RL, which combines graph embedding with side information and reinforcement learning to solve long-tail problem. GES4RL is based on Node2Vec and negative feedback reinforcement learning. It constructs a weighted directed graph of product propagation and uses Node2Vec to learn the embedding of products. We used gated recurrent unit (GRU) to learn user’s dynamic preferences and designed a negative feedback reinforcement learning model to generate the best recommendation strategy for long-tail products. Experimental results on User Behavior Dataset provided by TianChi show that the algorithm improves the diversity and hit rate of recommendations significantly.

     

/

返回文章
返回