news 2026/8/1 20:10:40

端到端可学习的物品分词与生成式推荐

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
端到端可学习的物品分词与生成式推荐

摘要

现有生成式推荐多采用"两阶段"范式:先离线构造语义标识(Semantic ID),再以固定标识训练生成器,导致分词器与生成器目标失配、分布不一致。ETEGRec 将 RQ-VAE 物品分词器与 T5 式生成器统一到一个端到端框架中,并引入两类推荐导向对齐(SIA、PSA),通过交替优化稳定地联动两模块学习。在 Amazon-2023 三个子集上,ETEGRec 相比传统序列模型与主流生成式基线均取得稳定提升,消融实验验证了对齐与交替优化的有效性。

1. 背景与问题定义

1.1 两阶段生成式推荐的局限

解耦训练:分词器(构造语义 ID)与生成器(自回归预测)相互独立,优化目标不一致,难以互相适配。

分布失配与模式单一:固定的 token 序列使生成器面对单一的模式,易过拟合;语义空间与偏好空间缺乏耦合。

长度偏置:不等长标识会引入预测偏好偏差,需固定长度缓解。

1.2 任务设定与符号

给定用户历史交互序列

,目标是预测下一物品

。每个物品以固定长度

的层级 token 表示:

生成式推荐将任务转化为自回归地生成下一物品的语义 ID:

2. 方法总览

image

双 Encoder–Decoder 架构:

分词器

采用 RQ-VAE,将连续语义嵌入量化为

层 token;生成器

采用 T5 式 Seq2Seq,对历史 token 进行编码并自回归生成目标 token。

两类推荐导向对齐:

SIA(Sequence-Item Alignment)对齐编码器的"下一物品语义预测"与真实物品在码本空间上的分布;

PSA(Preference-Semantic Alignment)对齐解码器的"用户总体偏好表示"与物品重构语义。

交替优化:周期性冻结/更新

的参数,稳定推进端到端联合学习。

3. 方法细节(输入→运算→输出)

3.1 物品协同嵌入来源

以已训练的 SASRec 的物品嵌入作为协同语义输入

。该

兼具协同信号与部分语义信息,是分词器

的输入。

3.2 物品分词器

:RQ-VAE

输入:

编码:

多层残差量化(码本层数

,每层大小

):

设第

层码本

,令

。逐层选择:

残差更新:

量化表示:

解码与重构:

损失:

其中

为停止梯度,

为平衡系数。

输出:层级语义 ID

、量化向量

、重构语义

直观:第 1 层捕获粗粒度语义,后续层在残差上逐层细化;固定长度

减少长度偏置。

3.3 生成式推荐器

:T5 式 Seq2Seq

输入:将历史序列中的每个物品经

token 化,得到

查表嵌入并加位置编码,得

编码器:多层自注意力与前馈后输出

解码器:以 [BOS] 起始,掩蔽自注意力保证自回归;跨注意力以

为键值(K、V)。输出隐藏态

聚合了对

的全局关注,可视为"用户总体偏好摘要"。

逐层预测:第

个解码位置对应第

层码本分类

训练目标(Teacher Forcing):

推理:固定步长

的 beam search,自回归生成

,经唯一映射反查物品。

3.4 推荐导向对齐(核心创新)

SIA:序列-物品分布对齐

汇聚得到

分别送入同一分词器,得到各层分布

,用对称 KL 对齐:

作用:迫使编码器在"码本空间"直接预测与目标物品一致的分布,避免解码器绕过编码器。

PSA:偏好-语义向量对齐

为正样本对,采用双向 InfoNCE(批内负样本):

其中

为余弦相似度,

为温度,

为 mini-batch。

作用:将"用户总体偏好"锚定到"物品语义空间",保持两空间同构。

3.5 交替优化与"损失—参数—梯度"映射

阶段 A(优化分词器,冻结生成器):

更新

及 SIA 的 MLP。

仅影响分词器;

通过

的分布与

的分布对齐,反向至码本与分词器;

拉向

的方向,更新分词器(使重构语义更贴合偏好)。

阶段 B(优化生成器,冻结分词器):

更新

及 SIA 的 MLP。

反向经解码器→编码器;

更新编码器与 MLP(使

的分布靠近

的分布);

拉向

,主要更新解码器。

收敛策略:循环数个周期,直至分词器收敛后固定,再充分训练生成器。

4. 复杂度与推理

分词器:单物品 token 化复杂度约为

生成器:序列建模主耗时为自注意力

与前馈

;逐层分类与 SIA 为

;PSA 为

为批内负样本数)。

总体量级:与 TIGER、LETTER 相当。推理阶段可预缓存语义 ID,生成步长固定为

5. 实验与结果要点

image

image

image

image

5.1 整体表现(Overall Performance)

我们在三个公共推荐基准上评估了 ETEGRec。表 3 给出了总体结果,主要观察如下:

传统序列推荐模型。 FDSA 在三个数据集上整体表现更优,主要得益于其引入了额外的文本特征嵌入。FMLP-Rec 与 SASRec、BERT4Rec 的表现大致相当,说明全 MLP 架构同样能够有效建模行为序列。

生成式推荐模型。 TIGER 与 TIGER-SAS 在三个数据集上稳定优于 CID 和 SID,即便后者采用了参数规模更大的预训练 T5。其性能差异主要来自物品分词策略的不同:SID 使用数值型 token 索引物品,缺乏语义信息;CID 基于物品共现图的启发式分词难以有效捕获物品相似性。相比之下,TIGER 与 TIGER-SAS 通过 RQ-VAE 从粗到细地学习层级文本或协同语义,更有利于推荐任务。值得注意的是,TIGER-SAS 与 TIGER 表现接近,表明协同与文本语义均对效果有重要贡献。LETTER 在多数场景下最佳,因为其有效融合了协同与文本语义信息。

我们的方法。 与所有基线相比,ETEGRec 在全部数据集上持续取得最优结果,验证了方法的有效性。我们将性能提升归因于通过面向推荐的对齐实现的物品分词器与生成式推荐器的相互增强。

SIA 使编码器在码本空间的预测与目标一致;

PSA 将解码器的偏好表示锚定到物品语义;

交替优化缓解两模块的相互干扰并稳定收敛。

5.2 消融实验(Ablation Study)

为评估 ETEGRec 中各个技术组件的影响,我们在三个数据集上进行了消融研究,四个变体的结果见表:

w/o LSIA: 去除序列—物品对齐(SIA,式(15))。该变体在各数据集上均劣于完整 ETEGRec,表明在码本空间中对齐序列表示与物品表示有助于生成式推荐。

w/o LPSA: 去除偏好—语义对齐(PSA,式(16)),同样导致性能下降,说明 PSA 损失能够增强用户偏好建模。

w/o LSIA & LPSA: 同时去除两种对齐,效果劣于只去除其中一种,进一步表明 SIA 与 PSA 均有正向贡献,且二者叠加可带来更好的性能。

w/o AT: 直接联合优化框架内所有目标,不采用交替训练策略。结果显示性能显著下降,说明在训练过程中对分词器的频繁更新会干扰推荐器的学习;采用交替训练可在保持组件协同对齐的同时实现稳定且有效的训练。

w/o ETE: 不进行端到端联合优化,而是使用 ETEGRec 最终得到的物品 token 重新训练一个生成式推荐器。结果表明,ETEGRec 的改进不仅来源于更优的物品标识符,还得益于将分词器中编码的先验知识与生成式推荐器进行一体化融合。

6. 小结

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/8/1 21:55:39

49、Bash编程:模式匹配、命令操作与示例代码详解

Bash编程:模式匹配、命令操作与示例代码详解 1. extglob扩展模式匹配运算符 在使用 shopt -s extglob 时,以下运算符会生效。匹配默认区分大小写,但可以使用 shopt -s nocasematch (bash 3.1+)来改变这一特性,该选项会影响 case 和 [[ 命令。 分组 含义 @( …

作者头像 李华
网站建设 2026/8/1 23:32:34

2、深入探索Bash编程:从基础到实用技巧

深入探索Bash编程:从基础到实用技巧 代码获取与结构 代码可从网站(http://www.bashcookbook.com )下载,下载格式为 .tgz 或 .zip 。代码文件通常位于类似 ./chXX/snippet_name 的路径下,其中 chXX 代表章节, snippet_name 是文件名。 “无用的cat使用”探讨…

作者头像 李华
网站建设 2026/8/1 11:36:07

40、计算机日常维护与管理任务实用指南

计算机日常维护与管理任务实用指南 在计算机使用和管理过程中,我们常常会遇到各种任务和问题。本文将为大家介绍一些常见问题的解决方案,涵盖文件重命名、文档查看、文件解压、会话恢复、会话共享、日志记录以及屏幕清理等方面。 1. 批量重命名文件 在实际操作中,我们可能…

作者头像 李华
网站建设 2026/8/1 2:37:49

Kotaemon重排序模型(Re-Ranker)集成教程

Kotaemon重排序模型集成深度指南 在构建企业级智能问答系统时,一个常见的痛点是:即便使用了强大的大语言模型(LLM),系统仍可能给出看似合理却与实际政策或知识不符的回答。这种“幻觉”问题在金融、医疗、人力资源等高…

作者头像 李华
网站建设 2026/8/1 9:54:50

Unity学习笔记(二十)PlayerPrefs(一)

目录 PlayerPrefs是什么 存储原理 读取相关 删除数据 PlayerPrefs数据唯一性 PlayerPrefs是什么 是Unity提供的可以用于存储读取玩家数据的公共类 存储原理 PlayerPrefs的数据存储,类似键值对存储,一个键对应一个值 提供了存储3种数据的方法 &am…

作者头像 李华
网站建设 2026/7/31 5:42:33

书籍-严羽《沧浪诗话》

严羽《沧浪诗话》详细介绍 书籍基本信息 书名:沧浪诗话 作者:严羽(南宋) 成书时间:南宋理宗时期(约公元1230-1250年) 卷数:1卷(分五章) 类别:诗歌…

作者头像 李华