已获取论文核心内容,以下是深度解读。
深度解读:Accelerating Generative Recommendation via Simple Categorical User Sequence Compression (CAUSE)
会议:WSDM 2026 | 作者:Qijiong Liu, Lu Fan, Zhongzhou Liu 等
1. 问题与动机
核心问题:生成式推荐(Generative Recommendation)在处理长用户交互序列时面临严重的计算效率瓶颈。
生成式推荐系统(如 Meta 提出的 HSTU)直接通过 Transformer 对用户行为序列建模并生成推荐 item,避免了传统"召回→粗排→精排"多阶段流水线的信息损失。然而,这类方法的注意力机制带来 O(HL²D) 的计算复杂度(H 为注意力头数,L 为序列长度,D 为维度),序列越长计算成本呈二次增长。
为什么重要:
- 工业场景中用户的历史行为序列通常长达数百甚至数千,长序列对推荐精度至关重要,但计算成本难以承受
- 现有序列压缩方法(如 GenRank)虽然能缩短序列,但往往导致推荐精度显著下降
- 生成式推荐要在工业级系统中落地,必须解决"长序列高精度"与"低延迟低成本"之间的矛盾
论文的关键洞察是:用户长期历史中存在大量冗余信号,同一类别下的多次交互所携带的兴趣信息高度重叠。与其用复杂方法压缩序列,不如利用 item 天然自带的类别属性进行分组聚合,既简单又有效。
2. 方法概述
论文提出 CAUSE(CAtegory-based User Sequence comprEssion) 框架,核心思路是将用户的长交互序列分为两部分,分别处理:
- 近期序列(Recent Sequence, r_u):保留最近的交互记录,完整输入 Transformer,捕捉用户当前兴趣
- 长期历史(Long-term History, h_u):通过类别分组压缩为少量 compact token,作为历史兴趣的紧凑表示
压缩三步骤:
| 步骤 | 操作 | 说明 |
|---|---|---|
| Grouping | 按 item 类别特征分桶 | 利用 item 已有的 categorical feature(如视频标签、电影类型)将历史交互分组 |
| Selection | 保留最近 V 个桶,每桶最多 G 个 item | 基于时间戳的双重截断,控制压缩后的 token 数量 |
| Aggregation | 桶内 embedding 平均池化 | 通过对齐变换 + 平均 + 桶位置编码,将每个桶压缩为单个 token |
聚合公式:
其中 是可学习的对齐参数,$E(b)[j]$ 是桶的位置编码。
最终模型输入为:[用户 ID] + [V 个压缩历史 token] + [近期序列 token],大幅缩短了 Transformer 的有效输入长度。
3. 核心技术贡献
1. "简单即有效"的序列压缩范式
与追求复杂注意力压缩机制的研究路线不同,CAUSE 证明了一个反直觉的结论:利用 item 已有的类别属性做最朴素的分组平均,就能取得甚至超越复杂方法的效果。这是该论文最核心的贡献——它不是在方法上做创新,而是在认知上提出了重要洞察。
2. 对齐变换机制(Alignment Transformation)
在聚合前对 item embedding 施加线性变换 ,将原始 embedding 映射到更适合聚合的空间。消融实验证明去掉该变换后性能显著下降,说明直接平均原始 embedding 是不够的。
3. 与动作预测的联合训练
除了 item 预测(使用 InfoNCE 对比学习损失),还联合训练 action 预测(点击/购买/dislike 等行为分类),总损失为 。action 预测为 embedding 提供了额外的监督信号。
4. 模型无关性(Model-agnostic)
CAUSE 不绑定特定模型架构,在 HSTU 和 GenRank 两种不同的生成式推荐模型上都取得了改进,说明该压缩策略具有通用性。
与现有方法的关键区别:
- 对比 HSTU:HSTU 直接处理全长序列,计算成本高;CAUSE 将历史部分压缩为少量 token
- 对比 GenRank:GenRank 也做序列缩减,但采用更粗暴的截断/采样策略,导致精度下降;CAUSE 通过类别分组保留了兴趣多样性
4. 实验结果分析
数据集
| 数据集 | 场景 | Item 数 | 用户数 |
|---|---|---|---|
| KuaiRand-27K | 短视频推荐 | 220K | 27K |
| MovieLens-20M | 电影推荐 | 21K | 130K |
主要结果
KuaiRand-27K(序列长度 512):
- nDCG@1 相比 HSTU 提升 11.48%
- 推理延迟仅 1.87ms,与 HSTU 相当
- 在相同计算成本下精度大幅领先
MovieLens-20M(序列长度 256):
- nDCG@1 相比 HSTU 提升 39.14%
- 保持可比的计算成本
效率维度:
- 整体可实现最高 6 倍计算成本缩减
- 在相同成本约束下,精度可提升最高 39%
消融实验(关键发现)
| 消融变体 | 影响 |
|---|---|
| 去掉对齐变换 () | 性能显著下降 |
| 去掉 action 预测任务 | 性能下降 |
| 去掉历史压缩(仅用近期序列) | 性能下降,证明长期历史有价值 |
| 用 K-Means 聚类替代显式类别 | 可行但劣于显式类别特征 |
最后一项消融尤为重要:当 item 没有显式类别特征时,K-Means 聚类可作为退路方案,使方法适用面更广。
超参数设置
默认配置:V=8 个桶,每桶最多 G=32 个 item,即长期历史最多压缩为 8 个 token。
5. 局限与未来方向
当前局限
- 对类别特征质量的依赖:方法效果很大程度上取决于 item 类别体系的合理性。如果类别划分过粗或过细,分组效果会打折扣。K-Means 替代方案虽可行但效果不如显式类别。
- 固定压缩粒度:V=8、G=32 是全局超参数,对所有用户一视同仁。但不同用户的兴趣广度不同——兴趣广泛的用户可能需要更多桶,而兴趣集中的用户可能只需要少数桶。
- 评估范围有限:仅在两个数据集上验证,且均为中等规模。在真正的工业级大规模数据(亿级用户、千万级 item)上的表现有待验证。
- 简单平均聚合的信息损失:桶内使用均值池化,不可避免地丢失了交互时间顺序、频率分布等细粒度信息。对于序列模式高度依赖时序的场景,可能不够。
- 类别层级单一:只使用一级类别特征。实际中 item 往往有多级类目体系(如"电子产品→手机→华为"),如何利用层级结构是开放问题。
未来方向
- 自适应压缩:根据用户兴趣分布动态确定桶数和每桶容量
- 加权聚合:用注意力机制或时间衰减替代简单平均,保留更多时序信息
- 多粒度类别融合:同时利用多级类目或多个类别特征维度进行分组
- 与其他序列压缩方法结合:CAUSE 可作为第一级粗压缩,再叠加更精细的压缩手段
- 在线增量更新:如何在用户产生新交互时高效更新压缩 token,避免重新计算
6. 工业实践意义
直接可落地的价值
1. 生成式推荐的上线成本降低
这是论文最大的工业价值。生成式推荐(如 HSTU)在学术上已证明有效,但在工业部署中最大的痛点就是长序列带来的计算成本。CAUSE 提供了一条低成本改造路径:不需要改模型架构,只需要在数据预处理阶段加入类别分组压缩,就能大幅降低推理成本。
2. 充分利用已有类目体系
工业系统中 item 普遍已有完善的类目标签体系(电商的三级类目、视频的内容标签、新闻的频道分类等),CAUSE 的思路可以直接复用这些现成资源,无需额外构建。
3. 用户长期兴趣画像的新思路
传统推荐系统中的用户画像(user profile)通常是离线计算的统计特征。CAUSE 的"类别桶"本质上是一种端到端学习的用户兴趣画像——每个压缩 token 代表用户在某个类别上的兴趣向量。这为统一用户画像和序列建模提供了桥梁。
架构层面的启发
4. 近期行为 + 长期压缩的双通道架构
CAUSE 的"近期序列完整保留 + 历史分组压缩"模式,与工业中常见的"短期兴趣模型 + 长期兴趣模型"双塔设计理念一致,但在统一框架内端到端实现,减少了系统复杂度。
5. 序列冗余的启示
论文最深刻的启示是:用户行为序列中大量交互是冗余的。一个用户连续看了 50 个科幻电影,第 2-50 个所提供的边际信息量在递减。这提醒工业实践者:与其一味追求更长序列,不如思考如何高效去冗余。这个思路不限于生成式推荐,对任何基于序列建模的推荐/搜索系统都有参考价值。
6. 工程简洁性优势
方法极其简单(分组→截断→平均),无需复杂的注意力压缩、记忆网络或蒸馏机制,工程实现门槛低,适合快速实验验证和线上 A/B 测试。在"复杂方法提升有限"的搜推场景中,简单有效的方案往往是最佳选择。
总结:CAUSE 论文的核心价值不在于方法的复杂度,而在于它揭示了一个被忽视的事实——用户行为序列中的类别级冗余是生成式推荐效率的主要瓶颈。通过最朴素的类别分组压缩,就能实现 6 倍加速和精度同步提升。这种"少即是多"的设计哲学对工业系统的模型优化和成本控制具有很强的指导意义。