There are great interests as well as many challenges in applyingreinforcement learning (RL) to recommendation systems. In this setting, anonline user is the environment; neither the reward function nor the environmentdynamics are clearly defined, making the application of RL challenging. In thispaper, we propose a novel model-based reinforcement learning framework forrecommendation systems, where we develop a generative adversarial network toimitate user behavior dynamics and learn her reward function. Using this usermodel as the simulation environment, we develop a novel Cascading DQN algorithmto obtain a combinatorial recommendation policy which can handle a large numberof candidate items efficiently. In our experiments with real data, we show thisgenerative adversarial user model can better explain user behavior thanalternatives, and the RL policy based on this model can lead to a betterlong-term reward for the user and higher click rate for the system.
Quick Read (beta)
Generative Adversarial User Model for
Reinforcement Learning Based Recommendation System
There are great interests as well as many challenges in applying reinforcement learning (RL) to recommendation systems. In this setting, an online user is the environment; neither the reward function nor the environment dynamics are clearly defined, making the application of RL challenging. In this paper, we propose a novel model-based reinforcement learning framework for recommendation systems, where we develop a generative adversarial network to imitate user behavior dynamics and learn her reward function. Using this user model as the simulation environment, we develop a novel Cascading DQN algorithm to obtain a combinatorial recommendation policy which can handle a large number of candidate items efficiently. In our experiments with real data, we show this generative adversarial user model can better explain user behavior than alternatives, and the RL policy based on this model can lead to a better long-term reward for the user and higher click rate for the system.
Recommendation systems have become a crucial part of almost all online service platforms. A typical interaction between the system and its users is — users are recommended a page of items and they provide feedback, and then the system recommends a new page of items. A common way of building recommendation systems is to estimate a model which minimizes the discrepancy between the model prediction and the immediate user response according to some loss function. In other words, these models do not explicitly take into account the long-term user interest. However, user’s interest can evolve over time based on what she observes, and the recommender’s action may significantly influence such evolution. In some sense, the recommender is guiding users’ interest by displaying particular items and hiding the rest. Thus, it is more favorable to design a recommendation strategy, such as one based on reinforcement learning (RL), which can take users’ long-term interest into account. However, it is challenging to apply RL framework to recommendation system setting since the environment will correspond to the logged online user.
First, a user’s interest (reward function) driving her behavior is typically unknown, yet it is critically important for the use of RL algorithms. In existing RL algorithms for recommendation systems, the reward functions are manually designed (e.g. for click/no-click) which may not reflect a user’s preference over different items.
Second, model-free RL typically requires lots of interactions with the environment in order to learn a good policy. This is impractical in the recommendation system setting. An online user will quickly abandon the service if the recommendation looks random and do not meet her interests. Thus, to avoid the large sample complexity of the model-free approach, a model-based RL approach is more preferable. In a related but a different setting where one wants to train a robot policy, recent works showed that model-based RL is much more sample efficient (Nagabandi et al., 2017; Deisenroth et al., 2015; Clavera et al., 2018). The advantage of model-based approaches is that potentially large amount of off-policy data can be pooled and used to learn a good environment dynamics model, whereas model-free approaches can only use expensive on-policy data for learning. However, previous model-based approaches are typically designed based on physics or Gaussian processes, and not tailored for complex sequences of user behaviors.
To address the above challenges, we propose a novel model-based RL framework for recommendation systems, where a user behavior model and the associated reward function are learned in unified mini-max framework, and then RL policies are learned using this model. Our main technical contributions are:
We develop a generative adversarial learning (GAN) formulation to model user behavior dynamics and recover her reward function. These two components are estimated simultaneously via a joint mini-max optimization algorithm. The benefits of our formulation are: (i) a better predictive user model can be obtained, and the reward function are learned in a consistent way with the user model; (ii) the learned reward allows later reinforcement learning to be carried out in a more principled way, rather than relying on manually designed reward; (ii) the learned user model allows us to perform model-based RL and online adaptation for new users to achieve better results.
Using this model as the simulation environment, we develop a cascading DQN algorithm to obtain a combinatorial recommendation policy. The cascading design of action-value function allows us to find the best subset of items to display from a large pool of candidates with time complexity linear in number of candidates.
In our experiments with real data, we showed that this generative adversarial model is a better fit to user behavior in terms of held-out likelihood and click prediction. Based on the learned user model and reward, we show that the estimated recommendation policy leads to better cumulative long-term reward for the user. Furthermore, in the case of model mismatch, our model-based policy can also quickly adapt to the new dynamics with a much fewer number of user interactions compared to model-free approaches.
2 Related Work
Commonly used recommendation algorithms use a simple user model. For instance, Wide&Deep networks (Cheng et al., 2016) and other methods such as XGBOOST (Chen & Guestrin, 2016) and DFM (Guo et al., 2017) based on logistic regression assume a user chooses each item independently; Collaborative competitive filtering (Yang et al., 2011) takes into account the context of user’s choice but assumes that user’s choices in each page view are independent. Session-based RNN (Hidasi et al., 2016) and session-based KNN (Jannach & Ludewig, 2017) improve upon previous approaches by modeling users’ history, but this model does not recover a users’ reward function and can not be used subsequently for reinforcement learning. Bandit based approaches, such as LinUCB (Li et al., 2010), can deal with adversarial user behaviors, but the reward is updated in a Bayesian framework and can not be directly in a reinforcement learning framework.
Zhao et al. (2018b, a); Zheng et al. (2018) used model-free RL for recommender systems, which may require many user interactions and the reward function is manually designed. Model-based reinforcement learning has been commonly used in robotics applications and resulted in reduced sample complexity to obtain a good policy (Deisenroth et al., 2015; Nagabandi et al., 2017; Clavera et al., 2018). However, user behaviors are sequences of discrete choices under a complex session context, very different from robotics models. Views resonating our work also emerge from literature (Shi et al., 2019) that user model estimation is essential for the success of RL in recommendation systems.
3 Setting and RL Formulation
We focus on a simplistic yet typical setting where the recommendation system (RS) and its user interact as follows:
Setting: RS displays items in one page to a user. The user provides feedback by clicking on one or none of these items, and then the system recommends a new page of items.
Our model can be easily extended to settings with more complex page views and user interactions.
Since reinforcement learning can take into account long-term reward, it holds the promise to improve users’ long-term engagement with an online platform. In the RL framework, a recommendation system aims to find a policy to choose from a set of items in user state , such that the cumulative expected reward is maximized,
where , , , . The overall RL framework for recommendation is illustrated in Figure 1. Several key aspects are as follows:
Environment: will correspond to a logged online user who can click on one of the items displayed by the recommendation system in each page view (or interaction);
State : will correspond to an ordered sequence of a user’s historical clicks;
Action of the recommender: will correspond to a subset of items chosen by the recommender from to display to the user. means the set of all subsets of items of , where are available items to recommend at time .
State Transition : will correspond to a user behavior model which returns the transition probability for given previous state and the set of items displayed by the system. It is equivalent to the distribution over a user’s actions, which is defined in our user model in section 4.1.
Reward Function : will correspond to a user’s utility or satisfaction after making her choice in state . Here we assume that the reward to the recommendation system is the same as the user’s utility. Thus, a recommendation algorithm which optimizes its long-term reward is designed to satisfy the user in a long run. One can also include the company’s benefit to the reward, but we will focus on users’ satisfaction.
Policy : will correspond to a recommendation strategy which returns the probability of displaying a subset of in user state .
Remark. In the above mapping, Environment, State and State Transition are associated with the user, Action and Policy are associated with the recommendation system, and Reward Function is associated with both of them. Here we use the notation to emphasize the dependency of the reward on the recommendation action, as the user can only choose from the display set. However, the value of the reward is determined by the user’s state and the clicked item once the item occurs in the display set . In fact, . Thus, in section 4.1 where we discuss the user model, we simply denote and assume is true.
Since both the reward function and the state transition model are unknown, we need to learn them from data. Once they are learned, the optimal policy in Eq. (1) can be estimated by repeated querying the model using algorithms such as Q-learning (Watkins, 1989). In the next two sections, we will explain our formulation for the user behavior model and the reward function, and design an efficient algorithm for learning the RL recommendation policy.
4 Generative Adversarial User Model
We propose a model to imitate users’ sequential choices and discuss its parameterization and estimation. The formulation is inspired by imitation learning, a powerful tool for learning sequential decision-making policies from expert demonstrations (Abbeel & Ng, 2004; Ho et al., 2016; Ho & Ermon, 2016; Torabi et al., 2018). We formulate a unified mini-max optimization to learn user behavior model and reward function simultaneously based on sample trajectories.
4.1 User Behavior As Reward Maximization
We model user behavior based on two realistic assumptions. (i) Users are not passive. Instead, given a set of items, a user will make a choice to maximize her own reward. The reward measures how much she will be satisfied with or interested in an item. Alternatively, the user can choose not to click on any items. Then she will receive the reward of not wasting time on boring items. (ii) The reward depends not only on the selected item but also on the user’s history. For example, a user may not be interested in Taylor Swift’s song at the beginning, but once she happens to listen to it, she may like it and then becomes interested in her other songs. Also, a user can get bored after listening to Taylor Swift’s songs repeatedly. In other words, a user’s evaluation of the items varies in accordance with her personal experience.
To formalize the model, we consider both the clicked item and the state of the user as the inputs to the reward function , where the clicked item is the user’s action and the user’s history is captured in her state (non-click is treated as a special item/action). Suppose in session , the user is presented with items and their associated features by the system. She will take an action according to a strategy which can maximize her expected reward. More specifically, this strategy is a probability distribution over , which is the result of the optimization problem below
Generative User Model: (2)
where is the probability simplex, and is a convex regularization function to encourage exploration, and controls the strength of the regularization.
Model interpretation. If we use the negative Shannon entropy as the regularizer, we can obtain an interpretation of our user model from the perspective of exploration-exploitation trade-off (See Appendix A for a proof).
This lemma makes it clear that the user greedily picks an item according to the reward function (exploitation), and yet the Gumbel noise allows the user to explore other less rewarding items. Similar models have appeared in econometric choice models (Manski, 1975; McFadden, 1973). The regularization parameter is an exploration-exploitation trade-off parameter. It can be easily seen that with a smaller , the user is more exploratory. Thus, reveals a part of users’ character. In practice, we simply set the value in our experiments, since it is implicitly learned in the reward , which is a function of various features of a user.
Remark. (i) Other regularization can also be used in our framework, which may induce different user behaviors. In these cases, the relations between and are also different, and may not appear in the closed form. (ii) The case where the user does not click any items can be regarded as a special item which is always in the display set . It can be defined as an item with zero feature vector, or, alternatively, its reward value can be defined as a constant to be learned.
4.2 Model Parameterization
We will represent the state as an embedding of the historical sequence of items clicked by the user before session , and then we will define the reward function based on the state and the embedding of the current action .
First, we will define the state of the user as , where each is the feature vector of the clicked item at session and is an embedding function. One can also define a truncated -step sequence as . For the state embedding function , we propose a simple and effective position weighting scheme. Let be a matrix where the number of rows corresponds to a fixed number of historical steps, and each column corresponds to one set of importance weights on positions. Then the embedding function can be designed as
where is a bias matrix, and is a nonlinear activation such as ReLU and ELU, and turns the input matrix into a long vector by concatenating the matrix columns. Alternatively, one can also use an LSTM to capture the history. However, the advantage of the position weighting scheme is that the history embedding is produced by a shallow network. It is more efficient for forward-computation and gradient backpropagation.
Next, we define the reward function and the user behavior model. A user’s choice will correspond to an item with feature , which will be used to parameterize the reward function and user behavior model as
where are weight matrices, are bias vectors , and are the final regression parameters. See Figure 2 for an illustration of the overall parameterization. For simplicity of notation, we will denote the set of all parameters in the reward function as and the set of all parameters in the user model as , and hence the notation and respectively.
4.3 Generative Adversarial Training
Both the reward function and the behavior model are unknown and need to be estimated from the data. The behavior model tries to mimic the action sequences provided by a real user who acts to maximize her reward function . In analogy to generative adversarial networks, (i) acts as a generator which generates the user’s next action based on her history, and (ii) acts as a discriminator which tries to differentiate the user’s actual actions from those generated by the behavior model . Thus, inspired by the GAN framework, we estimate and simultaneously via a mini-max formulation. More precisely, given a trajectory of observed actions of a user and the corresponding clicked item features , we learn and jointly by solving the following mini-max optimization
Generative Adversarial Training: (5)
where we use to emphasize that this is observed in the data. From the above optimization, one can see that the reward will extract some statistics from both real user actions and model user actions, and try to magnify their difference (or make their negative gap larger). In contrast, the user model will try to make the difference smaller, and hence more similar to the real user behavior. Alternatively, the mini-max optimization can also be interpreted as a game between an adversary and a learner where the adversary tries to minimize the reward of the learner by adjusting , while the learner tries to maximize its reward by adjusting to counteract the adversarial moves. This gives the user behavior training process a large-margin training flavor, where we want to learn the best model even for the worst scenario.
For general regularization function , the optimal solution in Eq. (4.3) does not have a closed form, and typically needs to be solved by alternatively updating and , e.g.
The process may be unstable due to the non-convexity nature of the problem. To stabilize the training process, we will leverage a special regularization for initializing the training process. More specifically, for entropy regularization, we can obtain a closed form solution to the inner-maximization for user behavior model, which makes the learning of reward function easy (See lemma 2 below and Appendix A for a proof). Once the reward function is learned for entropy regularization, it can be used to initialize the learning in the case of other regularization functions which may induce different user behavior models and final rewards.
When , the optimization problem in Eq. (4.3) is equivalent to the following maximum likelihood estimation
5 Cascading RL Policy for Recommendation
Using the estimated user behavior model and the corresponding reward function as the simulation environment, we can then use reinforcement learning to obtain a recommendation policy. The recommendation policy needs to choose from a combinatorial action space , where each action is a subset of items chosen from a larger set of candidates. Two challenges associated with this problem include the potentially high computational complexity of the combinatorial action space and the development of a framework for estimating the long-term reward (the Q function) from a combination of items. Our contribution is designing a novel cascading Q-networks to handle the combinatorial action space, and an algorithm to estimate the parameters by interacting with the GAN user model.
5.1 Cascading Q-Networks
We will use the Q-learning framework where an optimal action-value function will be learned and satisfies , . Once the action-value function is learned, an optimal policy for recommendation can be obtained as
where is the set of items available at time . The action space contains many choices, which can be very large even for moderate (e.g. 1,000) and (e.g. 5). Furthermore, an item put in different combinations can have different probabilities of being clicked, which is indicated by the user model and is in line with reality. For instance, interesting items may compete with each other for a user’s attention. Thus, the policy in Eq. (7) will be very expensive to compute. To address this challenge, we will design not just one but a set of related Q-functions which will be used in a cascading fashion for finding the maximum in Eq. (7).
Denote the recommender actions as and the optimal action as . Our cascading Q-networks are inspired by the key fact:
Also, there is a set of mutually consistent :
Thus, we can obtain an optimal action in computations by applying these functions in a cascading manner. See Algorithm 1 and Figure 2(c) for a summary. However, this cascade of functions are usually not available and need to be estimated from the data.
5.2 Parameterization and Estimation
Each function is parameterized by a neural network
where , and are the set of parameters, and we use the same embedding for the state as in Eq. (4). Now the problem left is how we can estimate these functions. Note that the set of functions need to satisfy a large set of constraints. At the optimal point, the value of is the same as for all , i.e.,
It is not easy to strictly enforce these constraints, we take them into account in a soft and approximate way. That is, we define the loss as
All networks are fitting against the same target . Then the parameters can be updated by performing gradient steps over the above loss. We note that in our experiments the set of learned networks satisfies the constraints nicely with a small error (Figure 5).
We conduct three sets of experiments to evaluate our generative adversarial user model (called GAN user model) and the resulting RL recommendation policy. Our experiments are designed to investigate the following questions: (1) Can GAN user model lead to better user behavior prediction? (2) Can GAN user model lead to higher user reward and click rate? and (3) Can GAN user model help reduce the sample complexity of reinforcement learning?
Dataset and Feature Description. We use 6 real-world datasets: (1) MovieLens contains a large number of movie ratings, from which we randomly sample 1,000 active users. Each display set is simulated by collecting 39 movies released near the time the movie is rated. Movie features are collected from IMDB. Categorical and descriptive features are encoded as sparse and dense vectors respectively; (2) Last.fm contains listening records from 359,347 users. Each display set is simulated by collecting 9 songs with the nearest time-stamp. (3) Yelp contains users’ reviews to various businesses. Each display set is simulated by collecting 9 businesses with the nearest location. (4) Taobao contains the clicking and buying records of users in 22 days. We consider the buying records as positive events. (5) RecSys15 YooChoose contains click-streams that sometimes end with purchase events. (6) Ant Financial News dataset contains clicks records from 50,000 users for one month, involving dozens of thousands of news. On average each display set contains 5 news articles. It also contains user-item cross features which are widely used in this online platform.(More details in Appendix C)
6.1 Predictive Performance of User Model
To assess the predictive accuracy of GAN user model with position weight (GAN-PW) and LSTM (GAN-LSTM), we choose a series of most widely used or state-of-the-arts as the baselines, including: (1) W&D-LR (Cheng et al., 2016), a wide & deep model with logistic regression loss function; (2) CCF (Yang et al., 2011), an advanced collaborative filtering model which takes into account the context information in the loss function; we further augment it with wide & deep feature layer (W&D-CCF); (3) IKNN (Hidasi et al., 2015), one of the most popular item-to-item solutions, which calculates items similarly according to the number of co-occurrences in sessions; (4) S-RNN (Hidasi et al., 2016), a session-based RNN model with a pairwise ranking loss; (5) SCKNNC (Jannach & Ludewig, 2017), a strong methods which unify session based RNN and KNN by cascading combination; (6) XGBOOST (Chen & Guestrin, 2016), a parallel tree boosting; (7) DFM (Guo et al., 2017) is a deep neural factorization-machine based on wide & deep features. S-RNN (Hidasi et al., 2016), a session-based RNN model with a pairwise ranking loss; (8) SCKNNW (Jannach & Ludewig, 2017) and (9) SCKNNC (Jannach & Ludewig, 2017), two methods which unify S-RNN and CKNN by weighted combination and cascading combination respectively; (10) XGBOOST (Chen & Guestrin, 2016), a parallel tree boosting, which is also known as GBDT and GBM.
Top- precision ([email protected]) is employed as the evaluation metric. It is the proportion of top- ranked items at each page view that are actually clicked by the user, averaged across test page views and users. Users are randomly divided into train(50%), validation(12.5%) and test(37.5%) sets. The results in Table 1 show that GAN model performs significantly better than baselines. Moreover, GAN-PW performs nearly as well as GAN-LSTM, but it is more efficient to train. Thus we use GAN-PW for later experiments and refer to it as GAN.
|(1) MovieLens||(2) LastFM||(3) Yelp||(4) Taobao||(5) YooChoose||(6) Ant Financial|
We also tested different types of regularization (Table 2). In general, Shannon entropy performs well and it is also favored for its closed form solution. However, on the Yelp dataset, we find that regularization leads to a better user model. It is noteworthy that the user model with regularization is trained with Shannon entropy initialization scheme proposed in section 4.3.
|Split 1||Split 2||Split 3|
|Model||[email protected]||[email protected]||[email protected]||[email protected]||[email protected]||[email protected]|
An interesting result on Movielens is shown in Figure 3 (see Appendix D.1 for similar figures). It shows GAN performs much better as time goes by, while the items predicted by W&D-CCF are concentrated on several categories. This indicates a drawback of static models — it fails to capture user interest evolution.
6.2 Recommendation Policy Based on User Model
With a learned user model, we can immediately derive a greedy policy to recommend items with the highest estimated likelihood. We will compare the strongest baseline methods W&D-LR, W&D-CCF and GAN-Greedy in this setting. Furthermore, we will learn an RL policy using the cascading Q-networks from section 5 (GAN-CDQN) and compare it with two RL methods: a cascading Q-network trained with reward (GAN-RWD1), and an additive Q-network policy (He et al., 2016), trained with learned reward (GAN-GDQN).
Since we cannot perform online experiments at this moment, we use collected data from the online news platform to fit a user model, and then use it as a test environment. To make the experimental results trustful and solid, we fit the test model based on a randomly sampled test set of 1,000 users and keep this set isolated. The RL policies are learned from another set of 2,500 users without overlapping the test set. The performances are evaluated by two metrics: (1) Cumulative reward: For each recommendation action, we can observe a user’s behavior and compute her reward using the test model. Note that we never use the reward of test users when we train the RL policy. The numbers shown in Table 3 are the cumulative rewards averaged over time horizon first and then averaged over all users. It can be formulated as , where is the reward received by user at time . (2) CTR (click through rate): it is the ratio of the number of clicks and the number of steps it is run. The values displayed in Table 3 are also averaged over 1,000 test users.
Experiments with different numbers of items in each page view are conducted and the results are summarized in Table 3. Since users’ behaviors are not deterministic, each policy is evaluated repeatedly for 50 times on test users. The results show that: (1) Greedy policy built on GAN model is significantly better than the policies built on other models. (2) RL policy learned from GAN is better than the greedy policy. (3) Although GAN-CDQN is trained to optimize the cumulative reward, the recommendation policy also achieves a higher CTR compared to GAN-RWD1 which directly optimizes reward. The learning of GAN-CDQN may have benefited from the well-known reward shaping effects of the learned continuous reward (Mataric, 1994; Ng et al., 1999; Matignon et al., 2006). (4) While the computational cost of GAN-CDQN is about the same as that of GAN-GDQN (both are linear in the total number of items), our proposed GAN-CDQN is a more flexible parametrization and achieved better results.
Since Table 3 only shows average values taken over test users, we compare the policies in user level and the results are shown in figure 4. GAN-CDQN policy results in higher averaged cumulative reward for most users. A similar figure which compares the CTR is deferred to Appendix D. Figure 5 shows that the learned cascading Q-networks satisfy constraints in Eq. (10) well when .
6.3 User Model Assisted Policy Adaptation
Former results in section 6.1 and 6.2 have demonstrated that GAN is a better user model and RL policy based on it can achieve higher CTR compared to other user models, but this user model may be misspecified. In this section, we show that our GAN model can help an RL policy to quickly adapt to a new user. The RL policy assisted by GAN user model is compared with other policies that are learned from and adapted to online users: (1) CDQN with GAN: cascading Q-networks which are first trained using the learned GAN user model from other users and then adapted online to a new user using MAML (Finn et al., 2017). (2) CDQN model free: cascading Q-networks without pre-trained by the GAN model. It interacts with and adapts to online users directly. (3) LinUCB: a classic contextual bandit algorithm which assumes adversarial user behavior. We choose its stronger version - LinUCB with hybrid linear models (Li et al., 2010) - to compare with.
The experiment setting is similar to section 6.2. All policies are evaluated on a set of 1,000 test users associated with a test model. Two sets of results corresponding to different sizes of display set are plotted in Figure 6. It shows how the CTR increases as each policy interacts with and adapts to users over time. In fact, the performances of users’ cumulative reward according to different policies are also similar, and the corresponding figure is deferred to Appendix D.3.
The results show that the CDQN policy pre-trained over a GAN user model can quickly achieve a high CTR even when it is applied to a new set of users (Figure 6). Without the user model, CDQN can also adapt to the users during its interaction with them. However, it takes around 1,000 iterations (i.e., 100,000 interactive data points) to achieve similar performance as the CDQN policy assisted by GAN user model. LinUCB(hybrid) is also capturing users’ interests during its interaction with users. Similarly, it takes too many interactions. In Appendix D.3, another figure is attached to compare the cumulative reward received by the user instead of CTR. Generally speaking, GAN user model provides a dynamical environment for RL policies to interact with. It helps the policy achieve a more satisfying status before applying to online users.
7 Conclusion and Future Work
We proposed a novel model-based reinforcement learning framework for recommendation systems, where we developed a GAN formulation to model user behavior dynamics and her associated reward function. Using this user model as the simulation environment, we develop a novel cascading Q-network for combinatorial recommendation policy which can handle a large number of candidate items efficiently. Although the experiments show clear benefits of our method in an offline and realistic simulation setting, even stronger results could be obtained via future online A/B testing.
This project was supported in part by NSF IIS-1218749, NIH BIGDATA 1R01GM108341, NSF CAREER IIS-1350983, NSF IIS-1639792 EAGER, NSF IIS-1841351 EAGER, NSF CNS-1704701, ONR N00014-15-1-2340, Intel ISTC, NVIDIA, Google and Amazon AWS.
- Abbeel & Ng (2004) Abbeel, P. and Ng, A. Y. Apprenticeship learning via inverse reinforcement learning. In ICML, 2004.
- Chen & Guestrin (2016) Chen, T. and Guestrin, C. Xgboost: A scalable tree boosting system. In Proceedings of the 22Nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 785–794. ACM, 2016.
- Cheng et al. (2016) Cheng, H.-T., Koc, L., Harmsen, J., Shaked, T., Chandra, T., Aradhye, H., Anderson, G., Corrado, G., Chai, W., Ispir, M., Anil, R., Haque, Z., Hong, L., Jain, V., Liu, X., and Shah, H. Wide & deep learning for recommender systems. In Proceedings of the 1st Workshop on Deep Learning for Recommender Systems. ACM, 2016.
- Clavera et al. (2018) Clavera, I., Nagabandi, A., Fearing, R. S., Abbeel, P., Levine, S., and Finn, C. Learning to adapt: Meta-learning for model-based control. arXiv preprint arXiv:1803.11347, 2018.
- Deisenroth et al. (2015) Deisenroth, M. P., Fox, D., and Rasmussen, C. E. Gaussian processes for data-efficient learning in robotics and control. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2015. ISSN 0162-8828.
- Finn et al. (2017) Finn, C., Abbeel, P., and Levine, S. Model-agnostic meta-learning for fast adaptation of deep networks. arXiv preprint arXiv:1703.03400, 2017.
- Guo et al. (2017) Guo, H., Tang, R., Ye, Y., Li, Z., and He, X. Deepfm: a factorization-machine based neural network for ctr prediction. arXiv preprint arXiv:1703.04247, 2017.
- He et al. (2016) He, J., Ostendorf, M., He, X., Chen, J., Gao, J., Li, L., and Deng, L. Deep reinforcement learning with a combinatorial action space for predicting popular reddit threads. In EMNLP, 2016.
- Hidasi et al. (2015) Hidasi, B., Karatzoglou, A., Baltrunas, L., and Tikk, D. Session-based recommendations with recurrent neural networks. arXiv preprint arXiv:1511.06939, 2015.
- Hidasi et al. (2016) Hidasi, B., Karatzoglou, A., Baltrunas, L., and Tikk, D. Session-based recommendations with recurrent neural networks. In ICLR, 2016.
- Ho & Ermon (2016) Ho, J. and Ermon, S. Generative adversarial imitation learning. In NIPS, 2016.
- Ho et al. (2016) Ho, J., Gupta, J. K., and Ermon, S. Model-free imitation learning with policy optimization. In ICML, 2016.
- Jannach & Ludewig (2017) Jannach, D. and Ludewig, M. When recurrent neural networks meet the neighborhood for session-based recommendation. In Proceedings of the Eleventh ACM Conference on Recommender Systems, pp. 306–310. ACM, 2017.
- Li et al. (2010) Li, L., Chu, W., Langford, J., and Schapire, R. E. A contextual-bandit approach to personalized news article recommendation. In Proceedings of the 19th international conference on World wide web, pp. 661–670. ACM, 2010.
- Manski (1975) Manski, C. F. Maximum score estimation of the stochastic utility model of choice. Journal of Econometrics, pp. 205 – 228, 1975. ISSN 0304-4076.
- Mataric (1994) Mataric, M. J. Reward functions for accelerated learning. In ICML, 1994.
- Matignon et al. (2006) Matignon, L., Laurent, G. J., and Fort-Piat, N. L. Reward function and initial values: Better choices for accelerated goal-directed reinforcement learning. In ICANN, 2006.
- McFadden (1973) McFadden, D. Conditional logit analysis of qualitative choice behaviour. In Zarembka, P. (ed.), Frontiers in Econometrics, pp. 105–142. Academic Press New York, 1973.
- Mnih et al. (2013) Mnih, V., Kavukcuoglu, K., Silver, D., Graves, A., Antonoglou, I., Wierstra, D., and Riedmiller, M. Playing atari with deep reinforcement learning. arXiv preprint arXiv:1312.5602, 2013.
- Nagabandi et al. (2017) Nagabandi, A., Kahn, G., Fearing, R. S., and Levine, S. Neural network dynamics for model-based deep reinforcement learning with model-free fine-tuning. arXiv preprint arXiv:1708.02596, 2017.
- Ng et al. (1999) Ng, A. Y., Harada, D., and Russell, S. J. Policy invariance under reward transformations: Theory and application to reward shaping. In ICML, 1999.
- Shi et al. (2019) Shi, J.-C., Yu, Y., Da, Q., Chen, S.-Y., and Zeng, A.-X. Virtual-taobao: Virtualizing real-world online retail environment for reinforcement learning. In Thirty-Third AAAI Conference on Artificial Intelligence, 2019.
- Torabi et al. (2018) Torabi, F., Warnell, G., and Stone, P. Behavioral cloning from observation. In IJCAI, 2018.
- Watkins (1989) Watkins, C. J. C. H. Learning from Delayed Rewards. PhD thesis, King’s College, Oxford, May 1989. (To be reprinted by MIT Press.).
- Yang et al. (2011) Yang, S.-H., Long, B., Smola, A. J., Zha, H., and Zheng, Z. Collaborative competitive filtering: learning recommender using context of user choice. In Proceedings of the 34th international ACM SIGIR conference on Research and development in Information Retrieval, pp. 295–304. ACM, 2011.
- Zhao et al. (2018a) Zhao, X., Xia, L., Zhang, L., Ding, Z., Yin, D., and Tang, J. Deep reinforcement learning for page-wise recommendations. 2018a.
- Zhao et al. (2018b) Zhao, X., Zhang, L., Ding, Z., Yin, D., Zhao, Y., and Tang, J. Deep reinforcement learning for list-wise recommendations. CoRR, 2018b.
- Zheng et al. (2018) Zheng, G., Zhang, F., Zheng, Z., Xiang, Y., Yuan, N. J., Xie, X., and Li, Z. Drn: A deep reinforcement learning framework for news recommendation. 2018.
Appendix A Lemma
A.1 Proof of lemma 1
First, recall the problem defined in Eq. (2):
Denote . Since can be an arbitrary mapping (i.e., is not limited in a specific parameter space), can be an arbitrary vector in . Recall the notation . Then the expectation taken over random variable can be written as
By simple computation, the optimal vector which maximizes Eq. (12) is
The cumulative distribution function for the Gumbel distribution is and the probability density is . Using the definition of the Gumbel distribution, the probability of the event where is defined in Eq. (3) is
Suppose we know the random variable . Then we can compute the choice probability conditioned on this information. Let and be the conditional probability; then we have
In fact, we only know the density of . Hence, using the Bayes theorem, we can express as
Now, let us look at the product itself.
Next, we make a change of variable . The Jacobian of the inverse transform is . Since , the absolute of Jacobian is . Therefore,
A.2 Proof of lemma 2
We make a assumption that there is no repeated pair in Eq. (4.3). This is a very soft assumption because is updated overtime, and is in fact representing its feature vector , which is in space . With this assumption, we can let map each pair to the optimal vector which maximize since there is no repeated pair. Using Eq. (13), we have
Eq. (4.3) can then be written as
which is the negative log-likelihood function and is equivalent to lemma 2. ∎
Appendix B Alogrithm box
The following is the algorithm of learning the cascading deep Q-networks. We employ the cascading functions to search the optimal action efficiently (line 2). Besides, both the experience replay (Mnih et al., 2013) and -exploration techniques are applied. The system’s experiences at each time-step are stored in a replay memory set (line 2) and then a minibatch of data will be sampled from the replay memory to update (line 2 and 2). An exploration to the action space is executed with probability (line 2).
Appendix C Dataset description
(1) MovieLens public dataset11 1 https://grouplens.org/datasets/movielens/ contains large amounts of movie ratings collected from their website. We randomly sample 1,000 active users from this dataset. On average, each of these active users rated more than 500 movies (including short films), so we assume they rated almost every movie that they watched and thus equate their rating behavior with watching behavior. MovieLens dataset is the most suitable public dataset for our experiments, but it is still not perfect. In fact, none of the public datasets provides the context in which a user’s choice is made. Thus, we simulate this missing information in a reasonable way. For each movie watched(rated) on the date , we collect a list of movies released within a month before that day . On average, movies run for about four weeks in theater. Even though we don’t know the actual context of user’s choice, at least the user decided to watch the rated movie instead of other movies in theater. Besides, we control the maximal size of each displayed set by 40. Features: In MovieLens dataset, only titles and IDs of the movies are given, so we collect detailed movie information from Internet Movie Database(IMDB). Categorical features as encoded as sparse vectors and descriptive features are encoded as dense vectors. The combination of such two types of vectors produces 722 dimensional raw feature vectors. To further reduce dimensionality, we use logistic regression to fit a wide&deep networks (Cheng et al., 2016) and use the learned input and hidden layers to reduce the feature to 10 dimension.
(2) An online news article recommendation dataset from Ant Financial is anonymously collected from Ant Financial news article online platform. It consists of 50,000 users’ clicks and impression logs for one month, involving dozens of thousands of news. It is a time-stamped dataset which contains user features, news article features and the context where the user clicks the articles. The size of the display set is not fixed, since a user can browse the news article platform as she likes. On average a display set contains 5 new articles, but it actually various from 2 to 10. Features: The news article raw features are approximately of dimension 100 million because it summarizes the key words in the article. Apparently it is too expensive to use these raw features in practice. The features we use in the experiments are 20 dimensional dense vector embedding produced from the raw feature by wide&deep networks. The reduced 20 dimensional features are widely used in this online platform and revealed to be effective in practice.
(3) Last.fm22 2 https://www.last.fm/api contains listening records from 359,347 users. Each display set is simulated by collecting 9 songs with nearest time-stamp.
(4) Yelp33 3 https://www.yelp.com/dataset/ contains users’ reviews to various businesses. Each display set is simulated by collecting 9 businesses with nearest location.
(5) RecSys1544 4 https://2015.recsyschallenge.com/ contains click-streams that sometimes end with purchase events.
(6) Taobao55 5 https://tianchi.aliyun.com/datalab contains the clicking behavior and buying behavior of users in 22 days. We consider the buying behaviors as positive events.
Appendix D More figures for experimental results
D.1 Figures for section 6.1
An interesting comparison is shown in Figure 3 and more similar figures are provided here. The blue curve is the trajectory of a user’s actual choices of movies over time. The orange curves are simulated trajectories predicted by GAN and CCF, respectively. Similar to what we conclude in section 6.1, these figures reveal the good performances of GAN user model in terms of capturing the evolution of users’ interest.
D.2 Figures for section 6.2
We demonstrate the policy performance in user level in figure 4 by comparing the cumulative reward. Here we attach the figure which compares the click rate. In each sub-figure, red curve represents GAN-DQN policy and blue curve represents the other. GAN-DQN policy contributes higher averaged click rate for most users.
D.3 Figures for section 6.3
This figure shows three sets of results corresponding to different sizes of display set. It reveals how users’ cumulative reward(averaged over 1,000 users) increases as each policy interacts with and adapts to 1,000 users over time. It can be easily that the CDQN policy pre-trained over a GAN user model can adapt to online users much faster then other model-free policies and can reduce the risk of losing the user at the beginning. The experiment setting is similar to section 6.2. All policies are evaluated on a separated set of 1,000 users associated with a test model. We need to emphasize that the GAN model which assists the CDQN policy is learned from a training set of users without overlapping test users. It is different from the test model which fits the 1,000 test users.