计算机科学 > 信息检索
[提交于 2025年10月16日
(v1)
,最后修订 2025年10月17日 (此版本, v2)]
标题: 大型推理嵌入模型:迈向下一代密集检索范式
标题: Large Reasoning Embedding Models: Towards Next-Generation Dense Retrieval Paradigm
摘要: 在现代电子商务搜索系统中,密集检索已成为不可或缺的组成部分。 通过计算查询与项目(产品)嵌入之间的相似性,它能够从大规模存储库中高效地选择候选产品。 随着大语言模型(LLMs)的突破,主流嵌入模型逐渐从BERT转向LLMs,以实现更精确的文本建模。 然而,这些模型仍然采用直接嵌入方法,嵌入的语义准确性仍显不足。 因此,对比学习被大量用于实现正对之间的紧密语义对齐。 结果,此类模型倾向于捕捉训练数据中的统计共现模式,使其偏向于浅层的词汇和语义匹配。 对于与目标项目存在明显词汇差异的困难查询,性能会显著下降。 在此工作中,我们提出了大型推理嵌入模型(LREM),该模型创新性地将推理过程融入表示学习中。 对于困难查询,LREM首先进行推理以深入理解原始查询,然后生成增强推理的查询嵌入用于检索。 这一推理过程有效弥合了原始查询与目标项目之间的语义差距,显著提高了检索准确性。 具体而言,我们采用两阶段训练过程:第一阶段通过SFT和InfoNCE损失在精心策划的查询-CoT-项目三元组上优化LLM,以建立初步的推理和嵌入能力,第二阶段通过强化学习(RL)进一步精炼推理轨迹。 大量的离线和在线实验验证了LREM的有效性,导致其自2025年8月起在中国最大的电子商务平台上部署。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.