大模型增强推荐

Large Language Models Enhanced Collaborative Filtering

LLM-CF 离线生成推理示例,在推荐时检索相关示例,并通过上下文学习模块增强协同过滤,无需在线调用大模型。

CIKM 2024 共同第一作者 Zhongxiang Sun, Zihua Si, Xiaoxue Zang, Kai Zheng, Yang Song, Xiao Zhang, Jun Xu
LLM-CF

通过离线生成推理示例与上下文学习,将大模型的世界知识和推理能力融入协同过滤。

怎样把大模型的世界知识与推理能力用于历史用户交互中的协同过滤,同时将大模型文本生成留在离线阶段?

从研究问题到机制与证据

下面按论文的图表顺序展开。每一节都说明这张图回答了什么问题,又支持了哪一步结论。

1

问题

把大模型知识接入协同过滤

LLM-CF 用离线大模型生成可复用的推理示例,再检索相关历史交互,帮助常规推荐模型判断当前用户对候选物品的兴趣。

LLM-CF 把大模型知识接入协同过滤图示

协同过滤从用户与物品的交互中学习偏好规律。大模型则可以理解产品属性、用户兴趣,以及历史行为和候选物品之间的关系。 只把一个用户和一个物品放进提示词,能用到的交互信息有限。LLM-CF 将相似的历史交互及其推理说明一起检索出来, 让这些语言信息参与当前预测。

图中的虚线划分了离线生成和在线服务。下半部分用 RecGen-LLaMA 构建推理示例库;上半部分把检索结果转成协同过滤特征, 交给原有推荐模型。在线仍需执行特征编码、检索、ICT 和骨干模型计算,但每次请求都不再需要大模型重新生成一段回答。

读图要点
  • 沿下方流程看推荐数据怎样经过 RecGen-LLaMA,形成带推理内容的历史示例库。
  • 沿上方流程看检索出的示例怎样变成新特征,与原有用户、物品特征共同参与推荐。
2

方法

从检索示例到 ICT 协同过滤特征

完整流程包括离线生成推理、检索 top-K 相似示例,以及通过短序列 Transformer Decoder 学习增强特征。

LLM-CF 从检索示例到 ICT 协同过滤特征图示

每条历史示例由推荐特征、生成的推理和历史交互标签组成。右侧提示词让教师模型分析用户历史、候选物品属性、 两者的匹配关系和可能的多样性需求。离线生成时会提供该条历史交互的真实标签。论文从训练交互中抽取十分之一, 构建这份示例库。

当前请求先用 BGE 将用户和物品的文本特征编码,再按余弦相似度检索历史示例。检索排除当前时间之后的交互, 并控制返回示例中的正负标签数量相等,减少标签比例带来的捷径。当前请求待预测的标签不会输入 ICT。

右上角说明每条示例怎样编码:推荐特征、推理文本和历史标签各形成一个向量。ICT 将这些三元组排在当前请求的特征向量之前, 输入 Transformer Decoder。检索四条示例时,序列共有十三个位置;当前请求位置的最终隐状态就是增强后的协同过滤特征。 论文采用两层、每层两个注意力头的解码器。

训练时,在原推荐损失之外加入推理向量的余弦重建损失,促使特征表征保留历史推理中的信息。排序模型将新特征与原特征拼接后做交互; 召回模型将它加到用户向量上。用于候选召回时,当前目标物品的信息会被遮蔽,避免把待召回答案暴露给模型。

读图要点
  • 左侧从下往上依次是生成示例、检索近邻、编码上下文和预测。
  • 重建分支在训练中约束推理信息;在线推荐使用学到的向量,不生成推理文本。
3

方法

让教师模型学会推荐,同时保留生成推理的能力

RecGen-LLaMA 将推荐数据与通用指令数据混合训练,并同时用推荐 AUC 和 MMLU 准确率检查两类能力的变化。

LLM-CF 让教师模型学会推荐,同时保留生成推理的能力 Amazon Beauty子图

Amazon Beauty

LLM-CF 让教师模型学会推荐,同时保留生成推理的能力 Amazon Sports子图

Amazon Sports

横轴是推荐 AUC,纵轴是 MMLU 准确率。仅用推荐数据做全参数微调,虽然提高了推荐效果,却明显降低了 MMLU 表现。 这会影响后续教师的选择,因为离线阶段还需要它生成有信息量的推理示例。

论文比较了混合数据、LoRA、加噪微调、表征正则化和权重插值等设置。RecGen 将推荐数据与 LIMA、alpaca-gpt4 指令数据混合, 在这两项评估中取得较好的平衡。这里对通用能力的判断来自 MMLU,图中的证据也应按这两个坐标来理解。

读图要点
  • 横、纵轴要一起看:只看推荐 AUC,无法判断教师是否适合继续生成推理。
  • RecGen 混合数据设置的表现,是选择它构建推理示例库的实验依据。
4

结果

检索多少条示例更合适?

这组实验比较 0、2、4、6、8 条上下文示例,以及取消正负标签比例约束的四示例设置。

LLM-CF 检索多少条示例更合适?图示

每个子图固定一个骨干模型,三条曲线对应三个数据集。若干曲线从两条增加到四条示例后继续改善, 但不同设置的峰值位置并不相同,也有两条、六条或八条示例表现更好的情况。论文采用四条示例作为效果与计算量之间的实用选择, 并非每条曲线的最优点都在四条。

n4_noc 保持四条示例,但取消正负标签比例约束。它在不同数据集和骨干上的变化方向并不相同。 主实验保留标签平衡设置,以减少检索结果中正标签占比对预测的影响。下面的表格进一步给出排序、召回与消融实验的主要证据。

读图要点
  • 在同一子图中比较 n2、n4、n6、n8,观察上下文长度对该骨干的影响。
  • n4_noc 改变的是标签比例,不能把它与增加示例数量当作同一个变量。

主实验:排序、召回与消融

实验使用 2019 年 Amazon Sports & Outdoors、Beauty、Toys & Games 数据,采用 leave-one-out 划分。 CTR 评估为每条正交互随机采样一个负例。下面三张表保留六个骨干模型的主指标 AUC,越高越好。 KD、KAR 与 LLM-CF 使用相同的 7B LLaMA2 系列模型,并保持骨干结构一致。LLM-CF 提高了全部 18 个原模型的 AUC, 并在四种框架对比中的 16 个组合上最高;完整 LogLoss 结果见论文表 3。

排序 AUC:Amazon Sports

骨干模型原模型KDKARLLM-CF
DeepFM0.79900.80430.79910.8137
xDeepFM0.81580.81690.81610.8196
AutoInt0.80030.80120.80390.8088
DCNv10.80230.80400.80240.8092
DCNv20.81100.81120.80870.8131
DIN0.79860.80230.79710.8089

取自论文表 3。Sports 上六个骨干模型的最高 AUC 均来自 LLM-CF。

排序 AUC:Amazon Beauty

骨干模型原模型KDKARLLM-CF
DeepFM0.78530.79590.78700.8044
xDeepFM0.80650.81040.81010.8113
AutoInt0.79490.79610.79390.8090
DCNv10.81460.81470.81650.8182
DCNv20.80280.80570.80030.8033
DIN0.78610.79340.78610.7967

取自论文表 3。DCNv2 是例外:KD 为 0.8057,高于 LLM-CF 的 0.8033。

排序 AUC:Amazon Toys

骨干模型原模型KDKARLLM-CF
DeepFM0.76810.77130.76980.7881
xDeepFM0.78360.78650.78980.7947
AutoInt0.76300.76350.76830.7754
DCNv10.76210.76520.76510.7702
DCNv20.77740.78270.77590.7812
DIN0.75860.76520.76200.7783

取自论文表 3。DCNv2 仍由 KD 取得更高 AUC。DeepFM 的 LogLoss 则随 AUC 改善,从 0.5770 降至 0.5581。

候选召回:HIT@10 / NDCG@10

骨干模型设置SportsBeautyToys
SASREC原模型0.0203 / 0.01010.0648 / 0.02930.0646 / 0.0251
SASREC+ LLM-CF0.0349 / 0.01540.0652 / 0.02930.0723 / 0.0321
YoutubeDNN原模型0.0177 / 0.00910.0272 / 0.01280.0229 / 0.0110
YoutubeDNN+ LLM-CF0.0191 / 0.00940.0291 / 0.01390.0285 / 0.0138
GRU4REC原模型0.0268 / 0.01350.0470 / 0.02410.0411 / 0.0202
GRU4REC+ LLM-CF0.0317 / 0.01640.0504 / 0.02500.0380 / 0.0192
SRGNN原模型0.0146 / 0.00730.0341 / 0.01690.0284 / 0.0137
SRGNN+ LLM-CF0.0210 / 0.01010.0359 / 0.01740.0302 / 0.0144

取自论文表 4。每格依次为 HIT@10 / NDCG@10,均越高越好;训练时每条交互采样 128 个负例。GRU4REC 在 Toys 上的两项指标都下降。

哪些部分有效?DeepFM 消融 AUC

变体SportsBeautyToys
原模型0.79900.78530.7681
去掉文本特征0.81370.80300.7880
去掉 CoT 与重建损失0.80090.79800.7766
用原始 LLaMA2 生成示例0.81320.79730.7806
均值池化替代解码器0.80710.79860.7739
LLM-CF0.81370.80440.7881

取自论文表 5 的 DeepFM 部分。去掉 CoT 时也去掉了重建损失,因此这项消融不能分离两者各自的贡献。其他骨干的个别指标存在变体优于完整模型的情况。

结果说明了什么

  • 与原模型相比,18 个骨干模型与数据集组合的排序 AUC 全部提高;同时考虑 KD 和 KAR 时,LLM-CF 在其中 16 个组合上最高。两处例外都是 DCNv2,分别位于 Beauty 和 Toys。
  • 候选召回的 HIT@10 在 12 个组合中的 11 个上升。GRU4REC 在 Toys 上出现负迁移,说明接入同一套增强特征并不保证所有骨干都获益。
  • DeepFM 消融支持推理示例和 ICT 解码器的作用。这些指标衡量的是推荐效果,并未评估教师生成推理的忠实性。

这项工作得到了什么

1

用历史示例承载知识迁移

离线生成的推理与检索到的用户交互共同形成增强特征,使常规推荐模型能够使用大模型提供的信息。

2

排序收益广泛,也有明确的适用边界

主实验中每个原始骨干的 AUC 都得到改善;KD 和召回实验中的例外则说明,迁移效果仍取决于骨干与数据。

3

离线生成与在线服务分开

大模型生成留在离线阶段,在线通过检索和小型 ICT 模块复用已有推理,最终预测仍由骨干推荐模型完成。

在整体研究计划中的位置

LLM-CF 是我较早围绕推荐系统开展的知识迁移研究。它考察大模型生成的推理怎样与用户交互数据共同改进较小的决策模型, 并明确区分教师生成与在线服务两个阶段。

这项工作与我对模型知识获取、使用的关注相衔接,实验直接检验的是推荐准确率、检索推理示例的作用及其跨骨干迁移效果。 后续项目再进一步研究模型内部的能力组织、失效机制与元认知调控。

资源与引用

@inproceedings{sun2024large,
  title={Large Language Models Enhanced Collaborative Filtering},
  author={Sun, Zhongxiang and Si, Zihua and Zang, Xiaoxue and Zheng, Kai and Song, Yang and Zhang, Xiao and Xu, Jun},
  booktitle={Proceedings of the 33rd ACM International Conference on Information and Knowledge Management},
  doi={10.1145/3627673.3679558},
  year={2024}
}