用Transformer生成购物篮推荐,更准更懂用户习惯。
T-REX: Transformer-Based Category Sequence Generation for Grocery Basket Recommendation
- 基于动态分段采样和因果掩码,捕捉购物短时与长时依赖。
- 在亚马逊真实数据上,推荐准确率显著优于现有系统。
- 适合做生鲜电商、零售类个性化推荐的工程师参考。
在线生鲜购物对序列推荐提出独特挑战:购买模式重复且商品间关系复杂。与传统电商不同,生鲜推荐需同时捕捉互补商品关联与跨会话的时间依赖。为解决亚马逊在线生鲜业务中的这一问题,我们提出T-REX,一种新型Transformer架构,通过学习短期购物篮依赖与长期用户偏好,生成个性化的品类级推荐。该方法引入三项创新:(1) 针对稀疏购物模式的动态序列分段采样策略;(2) 适应时间模式的自适应位置编码;(3) 降低维度的同时保持推荐质量的品类级建模。尽管BERT4Rec等掩码语言模型在捕捉商品关系上表现优异,但其信息泄露问题使其不适用于下一篮子预测。相比之下,T-REX采用因果掩码设计,更契合篮子生成的顺序特性,实现更高精度的预测。大规模离线数据实验与线上A/B测试均显示,T-REX显著优于现有系统。
原文摘要 · Abstract (English)
Online grocery shopping presents unique challenges for sequential recommendations due to repetitive purchase patterns and complex item relationships within the baskets. Unlike traditional e-commerce, grocery recommendations must capture both complementary item associations and temporal dependencies across shopping sessions. To address these challenges in Amazon's online grocery business, we propose T-REX, a novel transformer architecture that generates personalized category-level suggestions by learning both short-term basket dependencies and long-term user preferences. Our approach introduces three key innovations: (1) an efficient sampling strategy utilizing dynamic sequence splitting for sparse shopping patterns, (2) an adaptive positional encoding scheme for temporal patterns, and (3) a category-level modeling approach that reduces dimensionality while maintaining recommendation quality. Although masked language modeling techniques like BERT4Rec excel at capturing item relations, they prove less suitable for next basket generation due to information leakage issues. In contrast, T-REX's causal masking approach better aligns with the sequential nature of basket generation, enabling more accurate next-basket predictions. Experiments on large-scale grocery offline data and online A/B tests show significant improvement over existing systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。