arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.04875cs.IRcs.LG
收藏

通过强化学习使GPTRec与超越准确率的目标对齐

Aligning GPTRec with Beyond-Accuracy Goals with Reinforcement Learning

University of Glasgow(格拉斯哥大学)

Aleksandr Petrov, Craig Macdonald

AI总结:

针对传统Top-K推荐模型难优化超越准确率指标的问题,提出两阶段训练GPTRec的方法:先以师生法模拟Top-K模型,再用强化学习对齐多样性、降流行度偏差等目标,实验显示多数情况下其在准确率与次要指标间权衡更优。

摘要:

Transformer模型的适配版本(如BERT4Rec和SASRec)在序列推荐任务中,基于准确率的指标(如NDCG)上取得了当前最优性能。这类模型将物品视为词元,随后采用评分-排序方法(Top-K策略):模型先计算物品的分数,再根据该分数对物品排序。尽管该方法在基于准确率的指标上表现优异,但很难用于优化更复杂的超越准确率的指标,例如多样性。\n近期,采用不同Next-K策略的GPTRec模型被提出,作为Top-K模型的替代方案。与传统Top-K推荐不同,Next-K逐物品生成推荐结果,因此能够考虑对超越准确率指标至关重要的复杂物品间依赖关系。然而,GPTRec的原始论文仅在实验中关注准确率,并未解决如何针对复杂的超越准确率指标优化模型的问题。实际上,为实现超越准确率的目标训练GPTRec颇具挑战,因为用于训练推荐系统的交互训练数据通常难以与超越准确率的推荐目标对齐。\n为解决这一不对齐问题,我们采用两阶段方法训练GPTRec:第一阶段,我们使用师生方法训练GPTRec,模拟传统Top-K模型的行为;第二阶段,我们使用强化学习(Reinforcement Learning)使模型与超越准确率的目标对齐。我们专门针对提升推荐多样性和降低流行度偏差开展了实验。在两个数据集上的实验表明,在4种情况中的3种里,GPTRec的Next-K生成方法相比经典贪心重排序技术,能在准确率和次要指标之间实现更好的权衡。

英文摘要:

Adaptations of Transformer models, such as BERT4Rec and SASRec, achieve state-of-the-art performance in the sequential recommendation task according to accuracy-based metrics, such as NDCG. These models treat items as tokens and then utilise a score-and-rank approach (Top-K strategy), where the model first computes item scores and then ranks them according to this score. While this approach works well for accuracy-based metrics, it is hard to use it for optimising more complex beyond-accuracy metrics such as diversity. Recently, the GPTRec model, which uses a different Next-K strategy, has been proposed as an alternative to the Top-K models. In contrast with traditional Top-K recommendations, Next-K generates recommendations item-by-item and, therefore, can account for complex item-to-item interdependencies important for the beyond-accuracy measures. However, the original GPTRec paper focused only on accuracy in experiments and needed to address how to optimise the model for complex beyond-accuracy metrics. Indeed, training GPTRec for beyond-accuracy goals is challenging because the interaction training data available for training recommender systems typically needs to be aligned with beyond-accuracy recommendation goals. To solve the misalignment problem, we train GPTRec using a 2-stage approach: in the first stage, we use a teacher-student approach to train GPTRec, mimicking the behaviour of traditional Top-K models; in the second stage, we use Reinforcement Learning to align the model for beyond-accuracy goals. In particular, we experiment with increasing recommendation diversity and reducing popularity bias. Our experiments on two datasets show that in 3 out of 4 cases, GPTRec's Next-K generation approach offers a better tradeoff between accuracy and secondary metrics than classic greedy re-ranking techniques.

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

↑