arXiv:2502.15526cs.IR2025-02被引 27

对比稀疏与稠密检索在大模型中的扩展性,发现稀疏检索更优且可融合训练实现最佳效果。

Scaling Sparse and Dense Retrieval in Decoder-Only LLMs

  • 比较稀疏/稠密检索与对比损失/知识蒸馏的组合在不同模型规模下的表现
  • 80亿参数下融合两种损失的稀疏检索达到当前最优,超越所有基线
  • 稀疏检索在跨领域任务中更稳定,对标注数据不敏感

大规模语言模型在提升检索性能方面展现出巨大潜力,但以往研究主要聚焦于使用对比损失(CL)训练的稠密检索,忽略了其他检索范式和优化技术(如稀疏检索与知识蒸馏,KD)的扩展行为。本工作系统比较了不同检索范式(稀疏与稠密)及微调目标(CL、KD及其组合)在不同模型规模下的影响。基于MSMARCO段落数据集,在固定计算预算下,使用解码器仅有的模型(Llama-3系列:1B、3B、8B),在域内(MSMARCO、TREC DL)和域外(BEIR)基准上评估多种训练配置。关键发现:(1) 仅在使用对比损失时出现清晰的缩放效应,大模型显著提升性能;而知识蒸馏训练的模型在1B、3B、8B规模下表现相近,提升有限。(2) 稀疏检索在所有基准上均优于稠密检索,且对不完美监督信号更具鲁棒性。(3) 成功在8B规模上通过结合对比损失与知识蒸馏实现了稀疏检索的规模化,所有评测集均达到当前最优(SOTA)结果。

原文摘要 · Abstract (English)

Scaling large language models (LLMs) has shown great potential for improving retrieval model performance; however, previous studies have mainly focused on dense retrieval trained with contrastive loss (CL), neglecting the scaling behavior of other retrieval paradigms and optimization techniques, such as sparse retrieval and knowledge distillation (KD). In this work, we conduct a systematic comparative study on how different retrieval paradigms (sparse vs. dense) and fine-tuning objectives (CL vs. KD vs. their combination) affect retrieval performance across different model scales. Using MSMARCO passages as the training dataset, decoder-only LLMs (Llama-3 series: 1B, 3B, 8B), and a fixed compute budget, we evaluate various training configurations on both in-domain (MSMARCO, TREC DL) and out-of-domain (BEIR) benchmarks. Our key findings reveal that: (1) Scaling behaviors emerge clearly only with CL, where larger models achieve significant performance gains, whereas KD-trained models show minimal improvement, performing similarly across the 1B, 3B, and 8B scales. (2) Sparse retrieval models consistently outperform dense retrieval across both in-domain (MSMARCO, TREC DL) and out-of-domain (BEIR) benchmarks, and they demonstrate greater robustness to imperfect supervised signals. (3) We successfully scale sparse retrieval models with the combination of CL and KD losses at 8B scale, achieving state-of-the-art (SOTA) results in all evaluation sets.

检索增强大模型稀疏检索知识蒸馏

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。