超小模型实现高效文本检索,可本地运行
Fantastic (small) Retrievers and How to Train Them: mxbai-edge-colbert-v0 Tech Report
- 基于蒸馏技术打造17M/32M参数小模型
- 在短文本任务上超越ColBERTv2,长文本效率惊人
- 适合移动端和边缘设备的实时检索应用
本文提出mxbai-edge-colbert-v0系列模型,参数量分别为17M和32M。通过大量消融实验优化检索与后期交互模型,并将其蒸馏为小型化版本作为概念验证。目标是支持从云端大规模检索到任意设备本地运行的全尺度检索能力。该模型在常见短文本基准(BEIR)上表现优于ColBERTv2,长文本任务中实现前所未有的高效性,是未来一系列小型验证模型的首个基础版本。
原文摘要 · Abstract (English)
In this work, we introduce mxbai-edge-colbert-v0 models, at two different parameter counts: 17M and 32M. As part of our research, we conduct numerous experiments to improve retrieval and late-interaction models, which we intend to distill into smaller models as proof-of-concepts. Our ultimate aim is to support retrieval at all scales, from large-scale retrieval which lives in the cloud to models that can run locally, on any device. mxbai-edge-colbert-v0 is a model that we hope will serve as a solid foundation backbone for all future experiments, representing the first version of a long series of small proof-of-concepts. As part of the development of mxbai-edge-colbert-v0, we conducted multiple ablation studies, of which we report the results. In terms of downstream performance, mxbai-edge-colbert-v0 is a particularly capable small model, outperforming ColBERTv2 on common short-text benchmarks (BEIR) and representing a large step forward in long-context tasks, with unprecedented efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。