arXiv:2601.21759cs.IRcs.CL2026-01

用强化学习动态调整文本检索训练数据采样,提升性能且更省显存。

Influence Guided Sampling for Domain Adaptation of Text Retrievers

  • 基于影响度设计奖励信号,自动优化训练数据采样权重。
  • 在多语言和单语模型上分别提升5.03和0.94的NDCG@10。
  • 比现有方法节省1.5到4倍显存,适合资源受限场景。

通用开放域稠密检索系统通常使用大量异构语料和搜索任务进行训练。如何采样这些多样化的数据集?传统方法采用均匀采样、按实例数量比例采样或依赖人工专家监督。众所周知,采样策略对模型性能影响巨大,但针对嵌入模型的最优采样策略尚未得到充分研究。本文提出Inf-DDS,一种由强化学习驱动的采样框架,通过基于影响度的奖励信号自适应重加权训练数据,显著降低显存消耗。该方法迭代优化采样策略,优先选择能提升目标开发集表现的数据集。我们在多种文本检索任务上验证了该策略的有效性,结果表明其在检索性能和领域适应性上均优于现有基于梯度的采样方法,同时显存成本降低1.5至4倍。在训练多语言bge-m3模型时,绝对提升NDCG@10达5.03;训练all-MiniLM-L6-v2模型时,提升0.94,即使从专家设定的初始权重开始亦然。

原文摘要 · Abstract (English)

General-purpose open-domain dense retrieval systems are usually trained with a large, eclectic mix of corpora and search tasks. How should these diverse corpora and tasks be sampled for training? Conventional approaches sample them uniformly, proportional to their instance population sizes, or depend on human-level expert supervision. It is well known that the training data sampling strategy can greatly impact model performance. However, how to find the optimal strategy has not been adequately studied in the context of embedding models. We propose Inf-DDS, a novel reinforcement learning driven sampling framework that adaptively reweighs training datasets guided by influence-based reward signals and is much more lightweight with respect to GPU consumption. Our technique iteratively refines the sampling policy, prioritizing datasets that maximize model performance on a target development set. We evaluate the efficacy of our sampling strategy on a wide range of text retrieval tasks, demonstrating strong improvements in retrieval performance and better adaptation compared to existing gradient-based sampling methods, while also being 1.5x to 4x cheaper in GPU compute. Our sampling strategy achieves a 5.03 absolute NDCG@10 improvement while training a multilingual bge-m3 model and an absolute NDCG@10 improvement of 0.94 while training all-MiniLM-L6-v2, even when starting from expert-assigned weights on a large pool of training datasets.

文本检索领域自适应强化学习采样优化

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