arXiv:2504.10816cs.IRcs.CL2025-04被引 15

用大模型实现高效可解释的稀疏检索,索引更小性能不降。

CSPLADE: Learned Sparse Retrieval with Causal Language Models

  • 通过轻量适配训练和双向结构改进,让大模型稳定做稀疏检索
  • 80亿参数模型在稀疏检索上表现接近密集模型,索引体积更小
  • 首次分析大模型量化对检索效率的影响,适合关注部署优化的研究者

近年来,密集检索成为信息检索研究热点,但存在向量不可解释、索引体积大的问题。学习型稀疏检索(LSR)作为替代方案,在保持高效倒排索引结构的同时,取得了有竞争力的检索性能。然而,现有研究大多局限于BERT规模,难以扩展至大语言模型(LLM)。本文识别出两个关键挑战:(1)对比学习初期训练不稳定;(2)预训练模型单向注意力导致性能不佳。为此,我们提出两项技术:(1)轻量级适配训练阶段消除早期训练不稳定性;(2)设计两种模型变体实现双向信息传播。基于此,我们成功训练出80亿参数级别的LSR模型,在检索性能上与密集模型相当,同时显著减小索引规模。此外,我们首次从模型量化视角系统分析了基于LLM的LSR模型在性能与效率间的权衡,为大模型高效检索建模提供了新洞见。

原文摘要 · Abstract (English)

In recent years, dense retrieval has been the focus of information retrieval (IR) research. While effective, dense retrieval produces uninterpretable dense vectors, and suffers from the drawback of large index size. Learned sparse retrieval (LSR) has emerged as promising alternative, achieving competitive retrieval performance while also being able to leverage the classical inverted index data structure for efficient retrieval. However, limited works have explored scaling LSR beyond BERT scale. In this work, we identify two challenges in training large language models (LLM) for LSR: (1) training instability during the early stage of contrastive training; (2) suboptimal performance due to pre-trained LLM's unidirectional attention. To address these challenges, we propose two corresponding techniques: (1) a lightweight adaptation training phase to eliminate training instability; (2) two model variants to enable bidirectional information. With these techniques, we are able to train LSR models with 8B scale LLM, and achieve competitive retrieval performance with reduced index size. Furthermore, we are among the first to analyze the performance-efficiency tradeoff of LLM-based LSR model through the lens of model quantization. Our findings provide insights into adapting LLMs for efficient retrieval modeling.

稀疏检索大模型高效检索量化

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