用大模型蒸馏对话检索相似度,提升效率与效果。
DiSCo: LLM Knowledge Distillation for Efficient Sparse Retrieval in Conversational Search
- 蒸馏对话与文档的相似度而非嵌入,统一检索与上下文建模。
- 在五个数据集上实现最高6%的召回率提升,跨域效果显著。
- 支持多教师蒸馏,可灵活控制模型稀疏性,适合部署优化。
对话搜索(CS)需结合对话上下文从语料库中检索相关文档,近年来大语言模型(LLMs)通过基于上下文重写查询显著提升了性能,但推理时使用LLM带来效率问题。现有方法通过蒸馏人类重写查询的嵌入来缓解,但通常将对比检索任务与蒸馏过程分离。为此,我们提出DiSCo(稀疏对话检索的蒸馏),通过松弛的蒸馏目标统一检索与上下文建模。不同于仅依赖表示学习,该方法蒸馏对话与文档间的相似度得分,释放表示空间自由度并更好利用文档相关性的对比特性。在五个对话搜索数据集上的实验表明,DiSCo在领域内和跨域检索任务中均有显著提升,跨域召回率最高提升达6个百分点。此外,采用多教师蒸馏策略,使用多个LLM作为教师,进一步增强性能,在领域内设置下超越单个教师。模型稀疏性分析显示,DiSCo能更有效地控制模型稀疏程度。
原文摘要 · Abstract (English)
Conversational Search (CS) involves retrieving relevant documents from a corpus while considering the conversational context, integrating retrieval with context modeling. Recent advancements in Large Language Models (LLMs) have significantly enhanced CS by enabling query rewriting based on conversational context. However, employing LLMs during inference poses efficiency challenges. Existing solutions mitigate this issue by distilling embeddings derived from human-rewritten queries, focusing primarily on learning the context modeling task. These methods, however, often separate the contrastive retrieval task from the distillation process, treating it as an independent loss term. To overcome these limitations, we introduce DiSCo (Distillation of Sparse Conversational retrieval), a novel approach that unifies retrieval and context modeling through a relaxed distillation objective. Instead of relying exclusively on representation learning, our method distills similarity scores between conversations and documents, providing more freedom in the representation space and better leveraging the contrastive nature of document relevance. Extensive experiments on Learned Sparse Retrieval (LSR) across five CS datasets demonstrate that DiSCo achieves substantial improvements in both in-domain and out-of-domain retrieval tasks, achieving up to a six-point gain in recall for out-of-domain datasets over state-of-the-art methods. Additionally, DiSCo employs a multi-teacher distillation strategy, using multiple LLMs as teachers, further enhancing performance and surpassing the individual teachers in in-domain settings. Furthermore, analysis of model sparsity reveals that DiSCo allows for more effective control over the sparsity of the trained models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。