arXiv:2412.18232cs.IR2024-12ACL被引 4

用压缩技术提升长文本模型的检索效率,性能提升6%且内容缩减近一半。

Efficient Long Context Language Model Retrieval with Compression

  • 针对长文本检索设计新压缩方法,训练时兼顾效果与简洁性。
  • 在9个数据集上实现6%的性能提升,上下文长度压缩1.91倍。
  • 适合需要高效处理长文档的检索系统开发者使用。

长上下文语言模型(LCLMs)已成为信息检索的新范式,可直接将整个语料库作为单一上下文处理,具备超越传统稀疏和密集检索方法的潜力。然而,在上下文中处理大量段落计算成本高,推理时表征处理进一步加剧延迟。为此,我们提出一种专为LCLM检索设计的压缩方法,旨在最大化检索性能的同时最小化压缩后段落长度。通过生成合成数据,自动创建压缩段落并根据查询检索成功率标注为优选或次选,采用偏好优化训练压缩模型CoLoR,同时加入长度正则化损失以强化简洁性。在9个数据集上的实验表明,CoLoR在检索性能上提升6%,同时使上下文规模压缩1.91倍。代码已开源:https://github.com/going-doer/CoLoR。

原文摘要 · Abstract (English)

Long Context Language Models (LCLMs) have emerged as a new paradigm to perform Information Retrieval (IR), which enables the direct ingestion and retrieval of information by processing an entire corpus in their single context, showcasing the potential to surpass traditional sparse and dense retrieval methods. However, processing a large number of passages within in-context for retrieval is computationally expensive, and handling their representations during inference further exacerbates the processing time; thus, we aim to make LCLM retrieval more efficient and potentially more effective with passage compression. Specifically, we propose a new compression approach tailored for LCLM retrieval, which is trained to maximize the retrieval performance while minimizing the length of the compressed passages. To accomplish this, we generate the synthetic data, where compressed passages are automatically created and labeled as chosen or rejected according to their retrieval success for a given query, and we train the proposed Compression model for Long context Retrieval (CoLoR) with this data via preference optimization while adding the length regularization loss on top of it to enforce brevity. Through extensive experiments on 9 datasets, we show that CoLoR improves the retrieval performance by 6% while compressing the in-context size by a factor of 1.91. Our code is available at: https://github.com/going-doer/CoLoR.

长文本检索模型压缩信息检索

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