生成式推荐的泛化能力如何? How Well Does Generative Recommendation Generalize?
揭示生成式推荐通过token级记忆实现泛化的机制
前置知识
序列推荐
序列推荐是根据用户的历史交互序列预测下一个可能交互的物品。例如用户依次购买了A、B、C物品,模型需要预测用户接下来会购买什么。这是推荐系统中的一个经典问题,因为用户行为具有时序依赖性,最近的交互往往更能反映用户当前的兴趣。
本文研究的就是序列推荐任务中的两种范式对比,理解序列推荐是读懂本文的基础。
生成式推荐
生成式推荐将每个物品表示为一系列子物品token(例如语义ID),而不是唯一的物品ID。模型像语言模型一样在token序列上进行自回归生成。例如物品可能被编码为[token1, token2, token3],模型通过预测这些token序列来生成下一个物品。
本文的核心研究对象就是生成式推荐模型,理解其工作原理对理解论文贡献至关重要。
语义ID
语义ID是一种将物品映射到离散token序列的方法,通过聚类或量化技术将相似的物品分配相近的ID。例如使用分层聚类将物品编码为多层token,每层代表不同的语义粒度,这样相似的物品会共享相同的前缀token。
本文的机制分析核心就是基于语义ID的token级分析,理解语义ID是理解GR模型泛化机制的关键。
记忆与泛化
记忆指模型直接复用训练数据中观察到的模式,泛化指模型组合已知模式推断未见过的模式。在推荐中,记忆可能是用户买了A后通常会买B(训练中见过),泛化可能是用户买了A后喜欢和A相似的C(A到C的训练样本很少,但能从相似性推断)。
本文的分析框架就是基于记忆vs泛化的区分,理解这对概念是理解整篇论文的前提。
研究动机
生成式推荐模型在实践中表现出优于传统物品ID模型的性能,学术界广泛认为这是因为GR模型具有更好的泛化能力。然而,这种假设缺乏系统的验证,现有的对比大多停留在整体性能层面(如总体NDCG指标),而没有深入分析GR模型在什么类型的数据实例上表现更好,什么类型上表现更差。例如,GR模型在Sports数据集上总体表现可能不如SASRec,但这可能是因为它在某些类型的实例上表现很好,而在另一些类型上表现很差,两者相互抵消导致的整体效果不明朗。缺乏这种细粒度的分析使得我们无法真正理解GR模型的优势来源,也无法有效地改进模型设计。
本文的目标是本文的目标是建立一套分析框架,系统性地研究生成式推荐模型与传统物品ID模型在记忆和泛化能力上的差异。具体来说,作者希望回答以下问题:(1) GR模型是否真的在需要泛化的数据实例上表现更好?(2) GR模型的泛化能力背后的机制是什么?(3) 能否结合两种范式的优势来提升整体性能?通过在7个真实数据集上的实验对比,本文试图揭示GR模型泛化更好这一假设的真相,并深入探索其内在机制。
与已有工作不同的是,本文的独特切入角度在于将分析从物品级别转移到token级别,并提出了基于物品转移模式的数据实例分类框架。与以往工作只关注目标物品(如冷启动物品)不同,本文关注的是从历史物品到目标物品的转移模式。即使目标物品很热门,从给定历史到该物品的转移可能在训练数据中很罕见,这种情况下预测仍然需要泛化能力。此外,本文不是通过计算昂贵的反事实记忆(需要重新训练模型)或表征相似性(推荐任务有明确ground truth,不需要间接度量)来分类实例,而是基于训练数据中是否观察到了所需的物品转移模式来进行简单有效的分类,这使得分析可以扩展到大规模推荐场景。
核心方法
本文的方法分为三个主要部分:首先是建立记忆与泛化的分析框架,将测试数据实例分类为记忆相关、泛化相关或未分类三类;其次是进行token级别的机制分析,探索GR模型的泛化能力背后的机制;最后是提出基于记忆感知的适应性集成方法,结合两种范式的优势。整体思路是:先通过定义清晰的分类标准,区分不同类型的数据实例,然后在每个子集上分别评估GR模型和物品ID模型的性能,找出各自的优势领域,接着深入分析GR模型在泛化子集上表现更好的原因,最后利用这些发现来设计更好的集成策略。
本文的核心创新点在于提出了一套基于物品转移模式的数据实例分类框架,以及token级别的分析视角。与以往工作关注单个目标物品不同,本文关注的是从历史物品到目标物品的转移,并基于这些转移模式在训练数据中的出现情况来分类实例。如果一个实例所需的1-hop转移在训练数据中出现过,则定义为记忆相关;如果没有,但可以通过观察到的模式推断或组合得到(如传递性、对称性等),则定义为泛化相关。这种分类方法简单、有效,不需要重新训练模型或复杂的计算,可以大规模应用。此外,本文的另一个关键创新是从物品级别转移到token级别分析,发现GR模型的物品级泛化往往可以解释为token级的记忆,这一发现揭示了GR模型泛化的真相。
方法步骤详情
方法的第一步是定义数据分类标准。对于序列推荐任务,用户表示为历史物品序列u = [i1, i2, ..., it-1],目标是预测下一个物品it。作者定义物品转移为定向物品对[is 到 it],其中is, it属于u且s小于t。一个数据实例被分类为记忆相关当且仅当1-hop转移[it-1到it]在训练数据中出现过。如果不满足记忆条件,但满足泛化条件(如传递性:存在中间物品x使得[it-1到x]和[x到it]都在训练中;对称性:[it到it-1]在训练中;二阶对称性:通过中间物品的更复杂关系),则分类为泛化相关。对于最大跳数之外(本文设为4)既不属于记忆也不属于泛化的实例,标记为未分类。第二步是在七个真实数据集上训练两个代表性模型:TIGER(语义ID基础的GR模型)和SASRec(物品ID基础的传统模型),然后按照上述分类标准将测试数据分成记忆、泛化和未分类三个子集,分别评估两个模型在每个子集上的性能。第三步是token级别分析,作者引入前缀n-gram记忆的概念,定义一个实例满足n-gram前缀记忆如果训练数据中存在一个转移,其源物品和目标物品的前n个token与测试实例中的匹配。通过统计各个泛化类别的实例中有多大比例可以解释为不同长度的token级记忆,揭示GR模型泛化能力的机制。第四步是提出适应性集成方法,使用物品ID模型的预测置信度(MSP:最大softmax概率)作为记忆可能性的代理,通过公式将置信度转换为集成权重,动态调整两个模型的贡献。
技术新颖性
本文的技术新颖性体现在多个方面。首先,提出了基于物品转移模式的分类框架,这是首次在推荐系统中系统性地区分记忆和泛化能力,而不是仅仅观察整体性能。其次,从物品级别转移到token级别分析,引入前缀n-gram记忆的概念,揭示了GR模型泛化的真相——很多看似物品级泛化的实例实际上是token级的记忆。这一发现不仅解释了GR模型的优势来源,也解释了为什么GR模型在记忆任务上表现不如物品ID模型(token级记忆会导致概率质量分散,稀释了对特定物品转移的记忆)。第三,通过控制词表大小实验验证了token记忆比例与记忆-泛化权衡的因果关系:更小的词表(更高的token记忆比例)导致更好的泛化但更差的记忆。第四,提出了基于记忆感知的适应性集成策略,利用简单的MSP指标作为记忆可能性的代理,在两个范式之间动态调整权重,实现了优于固定权重集成的性能。这些发现和方法的结合使本文不仅提供了新的分析工具和理论洞察,还带来了实用的改进策略。
实验结果
本文的核心发现是生成式推荐模型和物品ID模型在不同类型的数据实例上表现出互补的优势。在7个真实数据集(Sports、Beauty、Science、Music、Office、Steam、Yelp)上的实验显示,TIGER(GR模型)在泛化相关子集上一致优于SASRec(物品ID模型),例如在Office数据集上提升高达+58.8%,在Beauty数据集上提升+56.7%,在Sports数据集上提升+39.8%。相反,SASRec在记忆相关子集上表现更好,例如在Yelp数据集上TIGER落后-43.6%,在Sports数据集上落后-41.2%,在Beauty数据集上落后-35.2%。两个模型在记忆子集上的性能都明显高于泛化子集,反映了超越观察转移进行泛化的内在难度。token级别分析显示,大量物品级泛化实例可以解释为token级记忆。例如,在Yelp数据集上,对称性类别中有46.6%的实例可以解释为4-gram前缀记忆,20.5%可以解释为3-gram前缀记忆。传递性类别中有24.9%可以解释为3-gram前缀记忆,31.5%可以解释为2-gram前缀记忆。二阶对称性也有类似模式。几乎所有测试实例(超过99%)都至少有1-gram前缀记忆支持。更深入的token记忆支持分析显示,TIGER的NDCG@10性能与前缀转移计数正相关,token记忆支持越多,TIGER相对SASRec的优势越大。相反,当物品转移概率高但前缀转移概率低时,TIGER会因为概率质量分散在共享相同前缀的多个物品上而表现不如SASRec。控制词表大小的实验验证了token记忆比例与记忆-泛化权衡的因果关系:更小的词表(更高的token记忆比例)平均提升泛化性能+10.24%,但降低记忆性能-7.62%。适应性集成实验显示,在7个数据集上,基于MSP指标的动态权重集成优于固定权重集成和单独使用任一模型。例如在Sports数据集上,适应性集成的N@10为0.0296,优于SASRec的0.0253、TIGER的0.0237和固定权重集成的0.0291。在Yelp数据集上,适应性集成的N@10为0.0352,优于SASRec的0.0321、TIGER的0.0257和固定权重集成的0.0351。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 序列推荐(需要泛化的实例) | NDCG@10 | TIGER模型 | SASRec模型 | 在Office数据集上+58.8%,在Beauty数据集上+56.7%,在Sports数据集上+39.8% |
| 序列推荐(需要记忆的实例) | NDCG@10 | SASRec模型 | TIGER模型 | 在Yelp数据集上+43.6%,在Sports数据集上+41.2%,在Beauty数据集上+35.2% |
| 序列推荐(整体性能) | N@10 | 适应性集成 | 固定权重集成 | 在Sports数据集上从0.0291提升到0.0296,在Beauty数据集上从0.0471提升到0.0476,在Yelp数据集上从0.0351提升到0.0352 |
| 序列推荐(整体性能) | R@10 | 适应性集成 | SASRec | 在Sports数据集上从0.0318提升到0.0537,在Beauty数据集上从0.0566提升到0.0841,在Yelp数据集上从0.0362提升到0.0637 |
局限与改进
作者承认的局限性包括:本文的分类框架依赖于训练数据中观察到的物品转移模式,可能无法捕捉所有类型的泛化。例如,未分类实例(通常少于10%)可能涉及训练中未见过的物品、更高阶的转移模式,或需要超越记忆和泛化的能力,本文框架无法解释这些情况。此外,本文的token记忆分析基于特定的语义ID量化方案(256乘3量化加一个标识符),其他tokenization方法可能有不同的行为。作者也提到本文只研究了两个代表性模型(TIGER和SASRec),其他模型可能有不同的记忆-泛化权衡。观察到的局限性还包括:MSP指标虽然简单有效,但可能不是估计记忆可能性的最优方法,特别是在模型校准不准确的情况下。另外,适应性集成的性能提升在某些数据集上相对较小(如Steam数据集上几乎没有提升),这可能是因为在这些数据集上两个模型的优势领域重叠较少或分类器的效果有限。
独立分析的弱点
独立分析的弱点包括:(1) 本文的分类框架只考虑了有限的泛化类型(传递性、对称性、二阶对称性、可替代性),可能遗漏其他重要的泛化模式。例如,基于内容相似性的泛化、跨用户的知识迁移等类型没有考虑。改进方向可以扩展泛化类型的定义,纳入更多维度的分析。(2) token记忆分析虽然揭示了GR模型的机制,但只关注前缀n-gram,可能忽略了更复杂的token模式(如token的组合效应、上下文依赖等)。改进方向可以引入更细粒度的token分析工具,如注意力权重分析、token激活模式分析等。(3) MSP指标作为记忆可能性的代理可能不够准确,特别是在模型不确定或训练不充分的情况下。改进方向可以设计更复杂的记忆估计指标,如基于模型不确定性的度量、基于训练数据分布距离的度量等。(4) 适应性集成只考虑了两个模型,可以扩展到更多模型或更复杂的集成策略,如基于学习器的权重预测、多粒度的集成等。(5) 实验只考虑了离线评估,在线A/B测试可能得出不同的结论,因为在线环境涉及探索-利用权衡、延迟、成本等额外因素。
未来方向
作者提出的未来工作方向包括探索明确针对本文刻画的能力(记忆和泛化)的tokenization方法。例如,可以设计能够平衡token级记忆和物品级记忆的tokenization方案,或者能够区分不同泛化类型的token表示。基于本文成果可以延伸的方向包括:(1) 将分析框架扩展到更多模型和任务,如图推荐、会话推荐、多行为推荐等,验证记忆-泛化权衡是否是普遍现象。(2) 开发更精细的记忆和泛化估计方法,不仅基于转移模式的出现频率,还考虑转移质量、上下文、用户特征等因素。(3) 研究如何设计tokenization方法来控制记忆-泛化权衡,例如通过正则化、约束优化或元学习来自动调整token共享程度。(4) 将适应性集成扩展到在线学习场景,根据实时反馈动态调整权重,实现持续改进。(5) 探索记忆和泛化在推荐系统中的其他应用,如冷启动推荐、长尾物品推荐、跨域推荐等,这些场景可能对泛化有更高要求。
复现评估
本文的复现性评估如下:(1) 开源情况:作者提供了代码仓库(https://github.com/Jamesding000/MemGen-GR),这是一个积极的信号,表明作者支持复现。(2) 数据:本文使用的7个数据集都是公开数据集,包括Amazon Reviews 2014和2023集合、Steam和Yelp,这些数据集可以从官方来源获取,数据复现难度低。(3) 算力:论文提到实验使用UC San Diego提供的计算资源,但没有给出具体的GPU型号、训练时间等细节。从模型规模(TIGER和SASRec都是中等规模模型)和数据集规模(最大Office数据集有223K用户、77K物品、1.8M交互)来看,训练应该可以在单张或几张现代GPU上完成,算力需求中等。(4) 难度:实验设置相对标准(leave-last-out划分、NDCG@10和Recall@10评估),模型实现应该不复杂。主要挑战可能在于数据分类和token记忆分析的实现细节,但代码仓库应该提供了这些实现。总体而言,本文的复现难度中等偏下,代码开源和数据公开大大降低了复现门槛。
论文图表
这张图展示了3个数据集(Sports、Beauty、Office)上,TIGER相对于SASRec的NDCG增益如何分解为两个维度:物品转移概率(X轴分位数)和前缀转移概率(Y轴分位数)。每个位置的颜色和数字表示该分组中TIGER相对于SASRec的NDCG差值。正值表示TIGER更好,负值表示SASRec更好。
这张图对理解token记忆如何稀释物品记忆很关键,它显示当物品转移概率高但前缀转移概率低时,TIGER因为概率质量分散而表现不如SASRec。这解释了为什么GR模型在记忆任务上表现更差,支持了token记忆稀释效应的论点。