端到端生成推荐系统,让物品编码与推荐协同优化。
Generative Recommender with End-to-End Learnable Item Tokenization
- 构建双编码器-解码器框架,统一物品编码与生成推荐过程。
- 在多个数据集上超越传统序列推荐模型和现有生成推荐基线。
- 适合对生成式推荐、端到端训练感兴趣的科研与工程人员。
生成式推荐系统作为一项创新方法,可直接生成物品标识符用于推荐任务。尽管潜力巨大,但有效构建与推荐系统匹配的物品标识符仍是主要挑战。当前方法常将物品编码与生成推荐训练分开进行,导致性能不佳。为此,我们提出ETEGRec——一种全新的端到端生成式推荐系统,将物品编码与生成推荐整合为统一框架。基于双编码器-解码器架构,ETEGRec包含物品编码器与生成式推荐器。为实现两者的协同互动,我们设计了一种面向推荐任务的对齐策略,包含序列-物品对齐与偏好-语义对齐两个关键优化目标,紧密耦合编码器与推荐器的学习过程,促进相互提升。此外,我们提出交替优化技术,保障整个框架稳定高效的端到端训练。大量实验表明,该方法在多个数据集上显著优于传统序列推荐模型及现有生成推荐基线。代码已开源:https://github.com/RUCAIBox/ETEGRec。
原文摘要 · Abstract (English)
Generative recommendation systems have gained increasing attention as an innovative approach that directly generates item identifiers for recommendation tasks. Despite their potential, a major challenge is the effective construction of item identifiers that align well with recommender systems. Current approaches often treat item tokenization and generative recommendation training as separate processes, which can lead to suboptimal performance. To overcome this issue, we introduce ETEGRec, a novel End-To-End Generative Recommender that unifies item tokenization and generative recommendation into a cohesive framework. Built on a dual encoder-decoder architecture, ETEGRec consists of an item tokenizer and a generative recommender. To enable synergistic interaction between these components, we propose a recommendation-oriented alignment strategy, which includes two key optimization objectives: sequence-item alignment and preference-semantic alignment. These objectives tightly couple the learning processes of the item tokenizer and the generative recommender, fostering mutual enhancement. Additionally, we develop an alternating optimization technique to ensure stable and efficient end-to-end training of the entire framework. Extensive experiments demonstrate the superior performance of our approach compared to traditional sequential recommendation models and existing generative recommendation baselines. Our code is available at https://github.com/RUCAIBox/ETEGRec.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。