通过动态剪枝提升检索模型微调效率与效果
OPERA: Online Data Pruning for Efficient Retrieval Model Adaptation
- 提出两阶段动态剪枝,按质量自适应筛选数据对
- 相比标准微调,排名提升1.0点,召回率提升0.4点
- 训练时间缩短至一半以下,适用于各类检索模型
领域特定微调对密集检索器至关重要,但并非所有数据对都同等贡献。本文提出OPERA框架,利用数据异质性提升检索模型适配的效率与效果。首先研究静态剪枝(SP),仅保留高相似度查询-文档对,发现存在质量与覆盖率的固有权衡:排名(NDCG)提升,但召回(Recall)因查询多样性下降而恶化。为解决此问题,提出两阶段动态剪枝(DP)策略,在训练全程自适应调节查询和文档层面的采样概率,优先选择高质量样本同时保持对全训练集的访问。在八个覆盖六个领域的数据集上评估表明:SP在标准微调基础上提升排名(NDCG@10 +0.2点);DP在排名(NDCG@10 +1.0点)和召回(Recall@20 +0.4点)上均达最优,平均排名1.38。该效果可扩展至基于大语言模型的Qwen3-Embedding,证明其架构无关性。值得注意的是,DP仅需标准微调50%以下的训练时间即可达到相当性能。
原文摘要 · Abstract (English)
Domain-specific finetuning is essential for dense retrievers, yet not all data pairs contribute equally to the learning process. We introduce OPERA, a data pruning framework that exploits this heterogeneity to improve both the effectiveness and efficiency of retrieval model adaptation. We first investigate static pruning (SP), which retains only high-similarity query-document pairs, revealing an intrinsic quality-coverage tradeoff: ranking (NDCG) improves while retrieval (Recall) can degrade due to reduced query diversity. To resolve this tradeoff, we propose a two-stage dynamic pruning (DP) strategy that adaptively modulates sampling probabilities at both query and document levels throughout training, prioritizing high-quality examples while maintaining access to the full training set. Evaluations across eight datasets spanning six domains demonstrate the effectiveness of both approaches: SP improves ranking over standard finetuning (NDCG@10 +0.2 points), while DP achieves the strongest performance on both ranking (NDCG@10 +1.0 points) and retrieval (Recall@20 +0.4 points), with an average rank of 1.38 across all methods. These findings scale to Qwen3-Embedding, an LLM-based dense retriever, confirming architecture-agnostic benefits. Notably, DP reaches comparable performance in less than 50\% of the training time required by standard finetuning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。