arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.06347cs.IR
收藏

基于决策Transformer的面向用户保留的推荐

User Retention-oriented Recommendation with Decision Transformer

Kesen Zhao, Lixin Zou, Xiangyu Zhao, Maolin Wang, Dawei yin

摘要:

利用强化学习~(RL)提高用户留存率因其在提高用户参与度方面的重要性而受到越来越多的关注。然而,由于试错搜索的要求,从头开始训练RL策略而不伤害用户的体验是不可避免的。此外,离线的方法,旨在优化的政策,没有在线的互动,遭受臭名昭著的稳定性问题的价值估计或无界方差的反事实的政策评估。为此,我们建议使用Decision Transformer~(DT)优化用户保留,通过将RL转换为自回归问题来避免离线困难。然而,在推荐中部署DT是一个不平凡的问题,因为存在以下挑战:(1)对数值奖励值建模的不足;(2)策略学习和推荐生成之间的数据差异;(3)不可靠的离线性能评估。因此,在这项工作中,我们提出了一系列解决暴露问题的战略。我们首先阐明了一个有效的奖励提示信息的Meta嵌入的加权聚合。然后,我们赋予一个加权的对比学习方法,以解决训练和推理之间的差异。此外,我们设计了两个强大的离线指标来衡量用户保留。最后,在基准数据集上的显著改善证明了所提出的方法的优越性。

英文摘要:

Improving user retention with reinforcement learning~(RL) has attracted increasing attention due to its significant importance in boosting user engagement. However, training the RL policy from scratch without hurting users' experience is unavoidable due to the requirement of trial-and-error searches. Furthermore, the offline methods, which aim to optimize the policy without online interactions, suffer from the notorious stability problem in value estimation or unbounded variance in counterfactual policy evaluation. To this end, we propose optimizing user retention with Decision Transformer~(DT), which avoids the offline difficulty by translating the RL as an autoregressive problem. However, deploying the DT in recommendation is a non-trivial problem because of the following challenges: (1) deficiency in modeling the numerical reward value; (2) data discrepancy between the policy learning and recommendation generation; (3) unreliable offline performance evaluation. In this work, we, therefore, contribute a series of strategies for tackling the exposed issues. We first articulate an efficient reward prompt by weighted aggregation of meta embeddings for informative reward embedding. Then, we endow a weighted contrastive learning method to solve the discrepancy between training and inference. Furthermore, we design two robust offline metrics to measure user retention. Finally, the significant improvement in the benchmark datasets demonstrates the superiority of the proposed method.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑