The Case Against Generation for Retrieval: Discriminative Language Models as Effective Retrievers
判别式LLM反击:用双塔模型替代生成式推荐检索
🧭 LLM双塔检索 · 判别式推荐 ★ 深度解读
Meta
将LLM作为共享背板的双塔检索器,配合蒸馏与连续思考,性能匹敌生成式推荐且可规模化部署
📌 工作参考价值(4 点)
- 双塔+LLM的务实路线:放弃生成式检索的Token解码,改用LLM编码器生成用户和物品的向量,通过ANN检索。对于物料量大的电商/广告系统,这种方案更易落地,无需担心Token到物品ID的映射错误和自回归延迟。
- 蒸馏技巧:从Cross-Encoder教师转移动态排序知识:用Cross-Encoder对同一候选集计算更准确的分数分布,然后以KL散度蒸馏到双塔学生。实操中可使用“yes/no”输出头计算logit差作为分数,同时加一个以用户历史为条件的Token预测辅助任务来增强教师,提升蒸馏上限。
- Token级特征压缩与层剪枝:将连续特征用有限标量量化(FSQ)离散化为少量Token,减少输入长度,提升QPS 19.7%;对Transformer进行层剪枝(搜索代理指标评估重要性)再知识迁移,可再提速10%,几乎无损。
- 数据效率与模型时效性:LLM双塔在仅用0.5%训练数据下即可匹敌生产DLRM基线;模型冻结后连续数天性能几乎不降,而传统DLRM因物品ID分布漂移损失严重,该特性可大幅降低重训成本,适合推荐系统频繁更新的场景。