Skip to main navigation Skip to search Skip to main content

End-to-end deep reinforcement learning based recommendation with supervised embedding

  • Feng Liu
  • , Huifeng Guo
  • , Xutao Li
  • , Ruiming Tang
  • , Yunming Ye*
  • , Xiuqiang He
  • *Corresponding author for this work
  • Harbin Institute of Technology Shenzhen
  • Huawei Technologies Co., Ltd.

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

The research of reinforcement learning (RL) based recommendation method has become a hot topic in recommendation community, due to the recent advance in interactive recommender systems. The existing RL recommendation approaches can be summarized into a unified framework with three components, namely embedding component (EC), state representation component (SRC) and policy component (PC). We find that EC cannot be nicely trained with the other two components simultaneously. Previous studies bypass the obstacle through a pre-training and fixing strategy, which makes their approaches unlike a real end-to-end fashion. More importantly, such pre-trained and fixed EC suffers from two inherent drawbacks: (1) Pre-trained and fixed embeddings are unable to model evolving preference of users and item correlations in the dynamic environment; (2) Pre-training is inconvenient in the industrial applications. To address the problem, in this paper, we propose an End-to-end Deep Reinforcement learning based Recommendation framework (EDRR). In this framework, a supervised learning signal is carefully designed for smoothing the update gradients to EC, and three incorporating ways are introduced and compared. To the best of our knowledge, we are the first to address the training compatibility between the three components in RL based recommendations. Extensive experiments are conducted on three real-world datasets, and the results demonstrate the proposed EDRR effectively achieves the end-to-end training purpose for both policy-based and value-based RL models, and delivers better performance than state-of-the-art methods.

Original languageEnglish
Title of host publicationWSDM 2020 - Proceedings of the 13th International Conference on Web Search and Data Mining
PublisherAssociation for Computing Machinery, Inc
Pages384-392
Number of pages9
ISBN (Electronic)9781450368223
DOIs
StatePublished - 20 Jan 2020
Externally publishedYes
Event13th ACM International Conference on Web Search and Data Mining, WSDM 2020 - Houston, United States
Duration: 3 Feb 20207 Feb 2020

Publication series

NameWSDM 2020 - Proceedings of the 13th International Conference on Web Search and Data Mining

Conference

Conference13th ACM International Conference on Web Search and Data Mining, WSDM 2020
Country/TerritoryUnited States
CityHouston
Period3/02/207/02/20

Keywords

  • End-to-end
  • Recommendation
  • Reinforcement learning
  • Supervised embedding

Fingerprint

Dive into the research topics of 'End-to-end deep reinforcement learning based recommendation with supervised embedding'. Together they form a unique fingerprint.

Cite this