1. 问题与动机
论文要解决电商赞助搜索(sponsored search)召回阶段稠密检索模型的训练数据问题:如何在不依赖点击信号、也不做人工标注的前提下规模化产出高质量训练数据。作者把现有做法的失败拆成三点并给出量化证据:(1) 点击有位置偏置,P(click|q,i,pos) ≠ P(rel|q,i),用户无论相关与否都点高位结果,点击训练的模型只是复刻现有排序而非学真实相关性;(2) 点击稀疏,长尾 query 的期望点击趋近 0,新品和低频 query(占相当大流量)拿不到任何训练信号;(3) 人工标注不可扩展,单对成本 $0.10–0.50,而他们需要 4M query 上的 240M+ 样本,一次刷新要 $24M–$120M,对"每周重训"完全不可行。检索是赞助搜索管线第一环,决定用户能看到哪些广告,下游排序、出价、收入都依赖它,Walmart 这种体量上召回相关性的小幅提升都带来可观收益。这构成了"既便宜又能每周刷新的训练数据"的动机。
2. 方法详解
核心洞察:三套已在线运行的异构检索系统——基于词典的词面系统 R_D(BM25 变体,索引挂了历史相关 query)、标准 BM25 系统 R_B、当前 ANN embedding 模型 R_A(即 Embedding Model V2,两塔 BERT)——对召回 item 大量分歧。top-500 处 dictionary-ANN 重合度仅 13.4%、dictionary-BM25 仅 15.05%,这种低重合不是噪声而是结构化信号。
流水线四阶段:采集(每 query 取三路各 top-K=500 及 channel 内 rank)→ 标注(校准级联给所有 query-item 对打 5 档相关性)→ 抽取(按一致/分歧分难度等级)→ 课程训练学生。五档相关性 0=Embarrassing、1=Bad、2=Okay、3=Good、4=Excellent,rel≥3 视为相关。五类样本:Easy Positive(三路都召回且 cascade 判 rel≥3);Hard Positive(ANN 漏召但至少一路词面系统 rank≤50 召回且 rel≥3,正是稠密模型该补的失败样本);Hard Negative(恰好一路 top-100 召回但 rel≤2,并用多 channel 一致性过滤剔除被多路召回的边界相关项);Semi-Hard/Token-Similar Negative(catalog 中 TF-IDF 相似度 ∈[0.2,0.6] 但任何 channel 都没召回且 cascade 判不相关——词面像但语义不相关,逼模型超越关键词匹配);Easy/Random Negative。经质量控制(删除无 rel≥3 的 query 约 8%、近重复去重、rank top-50 正/top-100 负、每 query 上限 50 正 50 负),理论上限约 400M 降到 240M+(158M+ 正、80M+ 负)。
标注级联 f_T 串联三个递增规模的领域微调分类器:184M cross-encoder → LoRA 微调 2B LLM → LoRA 微调 8B LLM。每模型输出 5 类预测加校准置信度,超本模型阈值即接受、否则下传更大模型,三个都未定的用多数投票+max 预测打破平局。关键是每类单独做 isotonic 校准(per-class isotonic)而非全局校准——模型在极端类(0、4)过自信、中间类(1–3)欠自信。这是 distant supervision(远监督)而非知识蒸馏:cascade 离线跑一次产出硬标签(argmax 非软概率),学生用标准监督目标训练,无软标签、无 teacher 梯度、无 co-training,好处是标注可缓存复用、cascade 与学生可独立升级、在线零额外延迟。
学生是两塔 sentence-transformers:6 层 22.7M 参数 encoder、384 维输出,query 与 item 双塔共享全部 transformer 层(Siamese);query 塔吃原始 query 串,item 塔只吃商品标题。三阶段渐进课程,每阶段从上一阶段最优 checkpoint 初始化,损失与样本难度匹配:Stage1 BCE(Easy Pos + Random Neg,学粗粒度相关),温度 τ 初始化 20 且可学;Stage2 MNR(多负样本排序损失,用 Hard Pos/Neg、in-batch 负样本);Stage3 Triplet(Token-Similar Neg,margin α=0.3,学细粒度语义区分)。训练 AdamW,lr 1e-5,5% warmup,单卡 batch 512、4 卡有效 2048,float16+torch.compile,导出 ONNX 用 GPU+ANN 在线服务。
3. 核心技术贡献
(1) 把多 channel 检索分歧转成五个难度等级的结构化训练数据挖掘管线——让标准两塔 BERT 不改架构就达生产级检索的关键上游创新;分歧区天然提供 hard positive(词面找到、ANN 漏)和 hard negative(单 channel 误召),比随机负样本信息量大得多。
(2) 三模型校准级联 + per-class isotonic 校准:5 类分级任务对人工标注达 89.1% 一致,cross-encoder 独立处理 74.5% 的对、把大 LLM 留给疑难样本,相比所有对都过三个模型省约 50% 算力;对照零样本 GPT-4(68.2%)、Claude-3(70.1%)说明领域微调级联远优于通用零样本。
(3) 三阶段渐进课程:相比同样 240M 数据单阶段训练,NDCG@10 从 0.843 提升到 0.923(+9.5%)。
(4) 生产部署 + Walmart 规模在线 A/B 验证:证明离线相关性增益能传导到可量化业务指标。
4. 实验结果分析
离线评测用 30,303 条第三方人工标注 query(Head 1439 / Torso 21185 / Tail 7679),标注与训练用的 cascade 完全独立,10K 对审计 Cohen's κ=0.79。主结果(V3 vs 上代 V2,二者同架构同服务栈、只差训练监督):NDCG@10 0.878→0.923(+5.1%,CI [+0.042,+0.051]);Avg Relevance 2.413→2.687;Precision 0.526→0.624;Recall 0.676→0.801;MAP 0.722→0.820;MRR 0.901→0.945。预训练基座 NDCG@10 0.771,说明 V2 点击训练已贡献 +13.9%,V3 的 LLM 标注再叠加 +5.1%。相关性分布上,最有害的 Embarrassing(0 分)从 8.7% 降到 3.5%(-5.2pp),Good+Excellent 从 52.7% 升到 62.4%。
分段看,尾部 query 增益最大:Tail 0.830→0.886(+6.8%),Head +5.1%、Torso +4.6%——印证 cascade 标注不依赖历史点击量,尾部与头部拿到同等标注质量,补上点击训练的数据稀疏短板。分品类六大品类全涨,Food 增益最大(0.837→0.897,+7.1%)但绝对值最低。在线 A/B(两周、每臂数千万广告请求):Ad Spend +2.80%、Ad CTR +1.4%、eCPM +2.8%、ROAS +2.4%、点击转化 +2.9%、Rev/Click +3.9%,均 p<0.05 显著。
消融(全量 V3=0.923)最有信息量:课程训练是最大贡献(+9.5%),且阶段顺序极关键——任何以 BCE 结尾的配置都塌回近单阶段水平(MNR→BCE 0.843、Triplet→BCE 0.844),因为粗粒度 BCE 放在排序损失之后会摧毁已学到的细粒度结构;MNR→Triplet(0.903)优于 Triplet→MNR(0.897),BCE 基座再叠加 +2.2%。数据侧多 channel 抽取贡献第二(单 ANN 通道 0.875,-5.2%),token-similar 负样本第三(-4.0%);一致性过滤(-0.8%)和 hard positive(-0.9%)贡献小但稳定。标注侧校准方法比模型选择更重要:Platt 校准 0.904(-2.1%)明显差于 per-class isotonic;只用 cross-encoder -1.0%;只用 8B LLM(无路由、约 3× 成本)0.921(-0.2%),说明级联价值主要在经济性而非精度。
5. 局限与未来方向
作者明确列出的局限:(a) 数据专有,无法在其他语料复现,只能复现方法;(b) 训练标签约 11% 与人工判断不一致,误差集中在边界类 1–3;(c) 对比面窄,只比生产 V2,没和微调过的通用检索器比,难以完全归因增益来源;(d) 多 channel 依赖,管线需多套异构检索系统提供分歧,单通道平台得自造合成分歧;(e) 单次部署,只有一个两周 A/B,需更多市场和更长周期验证。作者还讨论硬标签 vs 软标签取舍——argmax 硬标签解耦了 cascade 与学生,但丢掉边界对(1–3 分)的不确定性,可能传播系统性偏差;并指出 Food 品类最弱是因为只用标题输入,缺品牌/类目/描述等结构化属性。一个未正式刻画的风险是每周重训的反馈回路:ANN 通道就是当前生产模型,hard positive 集合每周随 ANN 改进而漂移,实践中观察到回路稳定(每轮 hard-positive 集随 ANN 追平词面系统而收缩),但收敛性未形式化证明,存在病态漂移的理论风险。未来方向:用结构化商品属性和多模态信号丰富输入、用同一 cascade 做软标签蒸馏对比硬标签量化信息损失、扩展到跨市场和多语言。
6. 工业实践意义
对搜索/推荐工程师最直接的可借鉴点:(1) 当你已有多路异构召回(词典/BM25/向量)时,它们的分歧本身就是免费且结构化的训练信号——一致即 easy positive、词面命中而向量漏即 hard positive、单路误召即 hard negative,可绕开点击偏置和昂贵人工标注冷启稠密召回,对社区 UGC 短文本搜索(点击稀疏、长尾多)尤其契合。(2) LLM 标注用级联+per-class isotonic 校准而非单个大模型:cross-encoder 扛 74.5% 简单对、疑难才上 8B,省约 50% 算力且只比纯 8B 低 0.2% NDCG——标注成本敏感时这是关键取舍。(3) 课程训练顺序不能随意:BCE→MNR→Triplet(由粗到细)比同数据单阶段多 9.5% NDCG,且千万别把 BCE 放最后。(4) 这是离线 distant supervision(标注一次缓存复用)而非在线蒸馏,在线零额外延迟,学生与标注器可独立升级,便于维护。(5) 部署上学生是 22.7M 小模型、384 维、ONNX+GPU ANN,整条重训周期 24–48 小时支持每周刷新;输入只用标题,想进一步提升需补结构化属性。注意它需要多路召回才能造分歧,单一召回栈要先解决"分歧从哪来"。