1. 问题与动机
生成式推荐(Generative Recommendation, GR)已成为推荐系统的重要范式,其核心思路是将推荐问题转化为基于 Semantic ID(SID)的序列生成任务。然而,随着 SID 粒度提高(每个物品由多个 token 表示),系统面临两个严重问题:(1) 训练资源消耗急剧上升——序列长度从单 token 膨胀至 L×N(L 为 SID 长度,N 为序列项数),导致 Transformer 注意力计算呈二次增长,显存和训练时间均大幅增加;(2) 性能不稳定——SID 粒度与推荐精度之间呈非单调关系,粒度增加并不总是带来收益,在不同模型和数据集上表现不一致。
作者将问题根因归结为"语义稀释效应"(Semantic Dilution Effect):高粒度 SID 序列中存在大量冗余 token,用户意图实际上仅由少数关键 token 驱动,而冗余部分不仅浪费计算资源,还稀释了本就稀疏的学习信号(SID 中真正的监督目标仅占序列末尾少数位置)。这一发现统一解释了训练效率低下和性能波动两大问题,为后续的双端优化策略奠定了理论基础。
2. 方法详解
STAMP(Semantic Trimming and Auxiliary Multi-step Prediction)采用"输入端压缩 + 输出端增密"的双端优化框架,包含两个核心模块:
语义自适应裁剪(Semantic Adaptive Pruning, SAP):在 Transformer 的指定层 l_prune 处插入动态 token 裁剪模块。对每个 token 计算重要性分数 I_i = S_sem(i) × S_attn(i),其中 S_sem 为隐状态的 L1 范数(衡量信息密度),S_attn 为该 token 从所有其他 token 接收到的注意力权重总和(衡量结构重要性)。保留重要性最高的 α·N 个 token(α 为保留比例),同时通过保护窗口 W 确保最近时间步的 token 不被剪枝(维护 Decoder-only 架构的因果性)。整个过程无需额外可学习参数。
多步辅助预测(Multi-step Auxiliary Prediction, MAP):在主任务预测下一个 token y_{t+1} 的同时,并行预测更远的 token y_{t+2}。具体做法是将当前隐状态 h_t 与下一步 token 的嵌入 E(y_{t+1}) 拼接后经 MLP 生成前瞻表示 h_t^{mtp},再通过与主模型共享的 LM_Head 投影得到 y_{t+2} 的预测分布。MAP 模块在推理时完全丢弃,不增加任何延迟开销。
训练目标为 L_total = L_NTP + λ·L_MAP,其中 λ=0.3 平衡两个损失。T5 架构使用 Adam 优化器(lr=1e-3),Qwen 架构使用 AdamW(lr 未明确给出),分别在单卡/双卡 A100 上训练,每个实验重复 5 次取平均。
3. 核心技术贡献
(1) 语义稀释效应的统一解释:首次将 SID-based GR 中的训练低效和性能不稳定统一归因为"语义稀释",揭示了 SID 冗余 token 同时影响计算效率(输入端)和学习信号(输出端)两个维度。
(2) 无参数动态裁剪机制:SAP 结合语义幅度(L1-norm)和注意力中心度双指标评估 token 重要性,在不引入额外参数的情况下实现自适应序列压缩。通过注意力热力图可视化,论文揭示了"脚手架 token"现象——部分 token 在浅层具有高注意力但在深层逐渐衰减,其语义价值已被关键 token 吸收,因此在较深层裁剪是安全的。与 max/average pooling 聚合或仅用 L2-norm/仅用 attention 的替代策略相比,SAP 效果最优。
(3) 共享权重的前瞻增密:MAP 通过多步预测将稀疏的 next-token 监督扩展为更密集的监督信号,且使用共享 LM_Head 避免参数膨胀。实验显示 MAP 作为独立模块即可稳定提升基线性能,与 SAP 结合后更进一步弥补裁剪可能带来的信息损失。
(4) 跨架构验证:在 Encoder-Decoder(T5/GRID)和 Decoder-only(Qwen/FORGE)两种主流 GR 框架上均验证了 STAMP 的有效性,证明了方法的通用性。
4. 实验结果分析
论文在四个数据集上进行了全面实验:三个 Amazon 学术数据集(Beauty/Sports/Toys)和一个淘宝工业级数据集 AL-GR-Tiny(131M 用户、251M 物品、14B 交互)。
主要结果:在 GRID 框架(T5)上,STAMP 在 Beauty 数据集上将 Recall@5 从 0.0426 提升至 0.0444(+4.2%),同时训练速度提升 1.25×-1.38×,显存降低 37.7%-53.3%。在 Sports 和 Toys 上保持性能持平的同时获得类似的效率提升。在 FORGE 框架(Qwen-0.5B)的工业数据集上,STAMP 实现 1.34× 加速和 17.2% 显存降低,Hit@1000 从 0.1400 提升至 0.1405。
消融分析:单独使用 MAP 可稳定提升性能(验证监督增密有效性);单独使用 SAP 在 Beauty/Toys 上甚至因去噪效应提升了准确率,在其他数据集上仅有轻微性能折损但大幅降低资源消耗;两者结合在效率和精度之间达到最佳平衡。
裁剪策略对比:SAP 优于 max pooling、average pooling、仅 L2-norm、仅 attention 等替代方案。聚合方法破坏语义细节,单一指标则遗漏不同类型的重要 token(如 L2-norm 忽略低幅度但高注意力的 token)。
超参数敏感性:保留比例 α 在中等压缩率下表现稳健,超过临界阈值后性能急剧下降;裁剪层深度需与保留比例协调——浅层保留基础特征,深层更适合裁剪。
5. 局限与未来方向
论文未设专门的局限性章节,但从实验中可识别以下局限:(1) 最优超参数因架构而异(T5 上 α=1/3,Qwen 上 α=0.6),部署时需针对不同骨干网络调参;(2) 工业数据集上显存降低幅度(17.2%)远低于学术数据集(36-54%),因为大模型的静态参数显存占比高,动态 token 显存的压缩空间相对有限;(3) 仅在序列推荐任务上验证,未涉及多模态推荐或对话式推荐场景。作者未明确讨论 SAP 对长尾物品或冷启动物品 SID 表示的潜在影响。
6. 工业实践意义
直接可用性:STAMP 在淘宝级数据(2.5 亿物品、140 亿交互)上验证了可行性,是首个在工业级 SID 基准上展示效率提升的框架。对于日常模型重训场景,1.34× 的加速直接转化为训练成本节约。
SID 设计启示:语义稀释效应的发现表明,工业系统中的 SID 不应盲目追求高粒度。推荐团队应先评估实际 SID 序列中的信息分布(注意力热力图),在粒度与效率之间找到最优平衡点。
即插即用部署:SAP 模块无额外参数,MAP 推理时丢弃,二者对在线 serving 无任何延迟影响。可直接集成到现有 T5 或 Qwen 基座的 GR 系统中。
注意事项:在显存受限但模型本身较大的场景(如 7B+ LLM-based GR),SAP 的显存收益有限,应更多关注 MAP 带来的精度稳定性。在 SID 信息密度较高的场景(如多模态 SID),SAP 的压缩比应保守设置以避免过度裁剪关键信号。