行为统计写入LLM重排涨13.3%,双样本dropout保住长尾

Robust Fusion of Semantic and Behavioural Signals for LLM Reranking in Personalised Search

Aleksandr V. Petrov, Nathan Stein, Erik Lybecker, Emma Schüldt, Daniel Lazarovski, Hugues Bouchard, Mounia Lalmas

cs.IR

2026-09-22

Spotify把查询–候选历史成功率QSS写入0.6B交叉编码器提示,全特征NDCG@7涨13.3%;双样本特征dropout在去掉QSS时比始终注入高4.0%,线上搜索成功率约+2%。

这篇在解决什么

个性化搜索同时要满足查询意图和用户偏好。工业界排序一直靠点击率、成功率这类行为统计吃饭,LLM交叉编码器又把这些数字变成提示词里的文本token。方便的代价是捷径学习:历史统计太好用,模型就不学语义和用户上下文,长尾和冷启动一旦缺统计就会掉点。

Spotify在音频流媒体搜索里把这个问题钉死在一个特征上。Query Slice Stats(QSS)是查询–候选–国家切片上的历史搜索成功率,对头部query–item对很强,对罕见查询和新内容经常缺失。论文要回答的不是「统计能不能进提示词」,而是「进了之后模型会不会把语义学废」。

方法

重排器是0.6B开源语言模型,词表扩了Semantic ID,对检索栈送来的音乐、播客、有声书候选做pointwise打分。提示词拼三组证据:查询和候选的标题、创作者、类型、Semantic ID;用户的市场和近期收听;以及分桶后的流行度和QSS。QSS连续成功率按历史分位切成verylow到veryhigh五档,缺失则整段省略。模型在决策位生成Y或N,推理时用Y的对数概率当排序分。

训练用标准SFT,损失只算在标签token上。双样本特征dropout把每条有QSS的样本做成一对:x+保留全部特征,x−只删QSS,标签不变。混合损失是α·L(x+)+(1−α)·L(x−)。α=1是始终看见QSS的朴素注入,α=0是训练时完全不看QSS。随机dropout作为对照消融,期望暴露量可以和1−α对齐,但论文主推确定性配对,因为两个视图一一对应。服务时仍是一次前向、同一套提示接口,多出来的计算只在训练。

结果

数据来自两周生产搜索日志。训练集100万session、45.2万查询、4610万候选;评估1万session。每session平均514个候选、1.3个正例。评估集34.7%的查询训练时没见过,97.8%属于出现次数≤500的长尾。指标是NDCG@7和Recall@7,截断对应产品里不滚动能看见的条数。

方法全特征NDCG@7去掉QSS
BM25 / monoT50.105 / 0.104—
仅QSS0.494—
α=0(训练不见QSS)0.5490.550
α=0.5(双样本)0.6220.547
α=1.0(始终注入)0.6180.526

α=0.5相对α=0全特征涨13.3%。α=1.0全特征涨12.6%、Recall@7涨9.5%,但去掉QSS后NDCG@7掉到0.526,显著低于从不看QSS的0.550。α=0.5去掉QSS时是0.547,与0.550统计上分不开(p=0.15),相对朴素注入高4.0%。Recall@7同样:α=0.5全特征0.756,去掉QSS为0.684,基线0.686。仅QSS已经0.494,远强于文本锚点,说明捷径本身就很硬;α=0.5仍比仅QSS高0.128。

按频率切片,双样本在冷启动查询上相对α=0大约+0.061,始终注入约+0.055,优势贯穿到频率500的长尾,只在短头(>500)反转。相对始终注入,长尾查询NDCG@7 +0.44%(95% CI 0.12%–0.75%),短头点估计−0.55%但不显著。

线上A/B/C:双样本搜索成功率+2.05%,始终注入+2.33%,两者1σ区间重叠,总盘分不出谁更好。冷启动切片上双样本相对始终注入+0.75%,1σ边界贴着零,方向和离线一致;实验不是为细切片打的功效。

为什么重要

把强行为特征写进LLM提示词,会改变模型学什么,不只是多一个token。生产上要用「特征在」和「特征被拿掉」两套评估,再按头、尾、冷启动切开。双样本把鲁棒性成本挪到训练,服务架构不用改,也不用另做一套降级排序器。

α=0.5已经吃到绝大部分QSS红利。始终看见统计在短头并不差,但会在缺统计时把语义学薄。线上总盘约2%的成功率和离线13.3%不是同一量纲,总盘也分不出两种QSS训练;真正要跟的是冷启动切片和诊断性的去特征评估。

局限与存疑

完全删除QSS是压力测试,不是线上每一种部分缺失、过期或噪声的再现。用户上下文被当作提示词的一部分,没有单独拆出个性化贡献,也没有对照可学习的融合基线。线上切片功效不足。QSS只是一个平台、一个重排器上的一条统计,推广到多条相关行为特征还没做。

术语

原文与代码

社区讨论

相关论文

全部论文解读