MMJul, 2023

使用优化的负采样和损失函数对基于会话的 Transformer 推荐进行扩展

TL;DRTRON 是一个可扩展的基于会话的 Transformer 推荐系统,使用了优化的负采样。通过整合前沿模型如 SASRec 和 GRU4Rec + 的 top-k 负采样和列表损失函数,TRON 在保持与 SASRec 相似的训练速度的同时,提高了推荐准确性。在相关大规模电子商务数据集的评估中,TRON 提升了当前方法的推荐质量。实时的 A/B 测试显示与 SASRec 相比点击率提高了 18.14%,突显了 TRON 在实际环境中的潜力。我们还提供了源代码和匿名数据集,供进一步研究使用。