利用未来互动信息提升序列推荐模型性能
Learning from the Future: Privileged Self-Distillation for Sequential Recommendation

- 用未来交互作为训练时特权信息,不改变推理结构
- 在多个数据集上显著提升推荐准确率,最高提升4.2%
- 适合需要高精度推荐的电商与内容平台使用
序列推荐模型通常在推理时仅依赖历史行为(前缀)进行预测,训练也采用仅基于目标项的独热标签。然而,这种监督方式难以捕捉非目标项间的相对偏好。本文发现,目标项之后的用户交互能反映意图演化,可作为训练时的额外监督信号。为此提出特权自蒸馏(PSD)框架:共享同一骨干网络,一个视图结合过去和未来交互生成特权教师分布,另一个视图仅用过去交互生成学生分布;通过蒸馏将未来信息转化为训练阶段的监督信号,而非推理输入。引入优势可达门控机制,仅蒸馏由当前前缀支持的教师信号,并采用动量平均教师以保持稳定性。整个过程端到端单阶段优化,部署模型与推理开销不变。在多个公开基准和不同骨干网络上均实现一致性能提升。
原文摘要 · Abstract (English)
Sequential recommenders are commonly trained with one-hot next-item labels under a causal (prefix-only) objective aligned with inference. While deployment-compatible, this supervision offers little insight into relative preferences among non-target items. Yet logged interaction sequences contain an additional supervisory source: interactions following the target often reveal how user intent evolves, making the target easier to interpret. We treat these future interactions as training-only privileged information, available during learning but not at inference. This raises a natural question: can future interactions provide richer supervision while keeping training aligned with inference-time prediction? We propose Privileged Self-Distillation (PSD), a framework that separates learning-time information from inference-time information. PSD applies two attention masks to the same backbone: a future-aware view yields a privileged teacher distribution conditioned on past and future interactions, while a prefix-only view yields the student distribution used for deployment. Distilling the privileged distribution converts future interactions into training-only supervision rather than inference-time inputs. Since both views share a backbone, the teacher's advantage is purely informational, not architectural, removing the need for a separately pretrained teacher and letting its supervision adapt as the student evolves. PSD further uses an advantage-reachability gate to focus distillation on teacher signals likely supported by the observed prefix, along with a momentum-averaged teacher for stable targets. The framework is optimized end-to-end in a single stage, leaving the deployed model and inference cost unchanged. Experiments across public benchmarks and diverse backbones show consistent improvements.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。