用改进的蒸馏方法让对话搜索更高效,速度翻倍且效果几乎不变。
Improving the Efficiency and Effectiveness of LLM Knowledge Distillation for Conversational Search
- 在KLD基础上加对比损失,提升检索精度
- 控制采样数量避免收益递减,优化效率
- 用正则化增强稀疏性,推理快两倍
对话搜索(CS)依赖对话上下文检索相关文档。大语言模型(LLM)通过有效查询重写显著提升了性能,但推理时效率低。知识蒸馏是平衡效果与效率的关键方法。现有工作使用相对熵(KLD)进行蒸馏,放宽了与教师信号的对齐要求。本文研究发现,在KLD目标中加入对比损失可显著提升精度导向的排序指标;对比采样策略对结果影响显著,需谨慎选择。尽管理论认为样本越多越准,实验显示收益随样本数增加而递减。此外,长对话中表示稀疏性下降限制了稀疏检索效率。我们发现,对基于KLD蒸馏的模型施加正则化损失,可大幅增强表示稀疏性,显著提升推理效率,且不影响检索效果:在TopiOCQA上实现2倍FLOPS降低,召回率@100仅下降≤2%。本工作为学习型稀疏对话检索器的蒸馏目标提供了新见解,并给出实用优化指南。
原文摘要 · Abstract (English)
Conversational Search (CS) considers retrieval of relevant documents based on conversational context. Large Language Models (LLMs) have significantly enhanced CS by enabling effective query rewriting. However, employing LLMs during inference poses efficiency challenges. A method to balance effectiveness and efficiency is the use of knowledge distillation from LLM-based query rewriting. Recent work applies the Kullback-Leibler Divergence (KLD) for distillation, relaxing the alignment with the teacher signal compared to previous methods. Despite these gains, several aspects of KLD-based distillation for conversational search remain understudied, and we investigate them in this work. Prior work in related fields suggests that adding a contrastive loss to the KLD objective can improve performance; we confirm this and observe significant gains in precision-oriented ranking metrics. We also find that contrastive sampling strategies for the KLD loss have a non-trivial impact and must be chosen carefully. Although theory suggests that more samples improve the KLD estimate, experiments show diminishing returns on the number of used samples. Finally, we address the phenomenon of decreased sparsity in longer conversations, which limits computational efficiency across sparse retrieval methods. We find that the representations from the model distilled with the KLD loss can be strongly regularized with a regularization loss, substantially improving sparsity and inference efficiency without significantly harming retrieval effectiveness. We achieve a $2\times$ decrease in FLOPS on TopiOCQA with negligible loss in effectiveness, corresponding to a $\leq 2%$ drop in Recall@100. Our results provide insights into distillation objectives for learned sparse conversational retrievers and offer practical guidelines for improving effectiveness and efficiency in first-stage retrieval.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。