第 95 题:S3-Rec的自监督预训练,互信息最大化的代理任务设计?
题目
S3-Rec的自监督预训练,互信息最大化的代理任务设计?
完整讲解
一、S3-Rec 与自监督预训练
S3-Rec 通过自监督预训练学习序列表征,再在下游点击/转化等任务上微调。核心思想是设计代理任务(pretext),使模型从无标注行为序列中学习有用的结构,其中 互信息最大化(MIM) 是常用目标之一。
二、互信息最大化的代理任务
互信息 $I(X;Y)$ 衡量两个随机变量的关联程度。在序列推荐中,常令 $X$ 为局部/掩码后的序列视图,$Y$ 为全局序列表征或另一视图,目标为最大化 $I(X;Y)$,使局部与全局一致、或不同增强视图一致。
- 全局-局部:用 encoder 得到全局表征 $\boldsymbol{h}_g$,再对局部片段(如连续子序列)得到 $\boldsymbol{h}_l$,通过 MIM 目标(如 InfoNCE)拉近正对 $(\boldsymbol{h}_l, \boldsymbol{h}_g)$、推远负对。
- 多视图:对同一序列做不同增强(掩码、裁剪、重排等)得到多视图,最大化各视图表征间的互信息。
- Item-Attribute:物品 ID 与属性(类目、描述)的互信息,使嵌入对齐多模态信息。
三、代理任务设计要点
- 与下游一致:若下游是 next-item,代理任务宜包含序列连续性/预测性;若下游是多行为,可包含行为类型预测。
- 负样本:InfoNCE 需负样本,常用 in-batch 负例或采样负例;负例质量影响对比学习效果。
- 预训练+微调:预训练阶段只优化 MIM,微调阶段接任务头(如 next-item 交叉熵),可冻结或全量微调 encoder。
面试要点
- 能说明 S3-Rec 用自监督预训练 + 互信息最大化学习序列表征。
- 能解释 MIM 在序列中的典型形式:全局-局部、多视图、item-attribute 对齐。
- 能说清代理任务与下游一致、负样本设计、预训练与微调两阶段。
记忆要点
- S3-Rec:自监督预训练,代理任务常用互信息最大化(MIM)。
- MIM 形式:全局-局部、多视图、item-属性对齐;拉近正对、推远负对(如 InfoNCE)。
- 设计要点:代理与下游一致、负样本质量、预训练后微调。