用大模型增强检索数据,发现小模型也能高效提效。
Evaluating the Effectiveness and Scalability of LLM-Based Data Augmentation for Retrieval
- 用大模型生成多样化数据提升检索性能。
- 增广规模超过阈值后效果反而下降。
- 小模型增广可媲美大模型,适合资源有限场景。
紧凑型双编码器模型因高效和可扩展性被广泛用于检索,但其性能常逊于基于大语言模型(LLM)的检索系统,可能源于知识储备有限。尽管已有研究提出使用LLM进行数据增强以缩小差距,但对其实效性和可扩展性在真实检索任务中的理解仍不足。现有工作未系统探索关键因素,如最优增广规模、是否必须使用大型增广模型,以及多样增广是否提升泛化能力,尤其在分布外(OOD)场景下。本研究对检索中LLM增广的有效性进行了全面评估,涵盖超过100种不同的检索模型、增广模型与策略组合。结果表明,尽管增广能提升检索性能,但其收益在达到一定规模后会减弱,即使采用多样化策略亦然。令人意外的是,使用较小的LLM进行增广即可达到与大型增广模型相当的性能。此外,我们发现增广对预训练不充分的检索模型提升最大。这些洞察有助于制定更明智、高效的增广策略,实现性能最大化的同时降低成本。代码与增广数据集已公开于 https://aka.ms/DAGR。
原文摘要 · Abstract (English)
Compact dual-encoder models are widely used for retrieval owing to their efficiency and scalability. However, such models often underperform compared to their Large Language Model (LLM)-based retrieval counterparts, likely due to their limited world knowledge. While LLM-based data augmentation has been proposed as a strategy to bridge this performance gap, there is insufficient understanding of its effectiveness and scalability to real-world retrieval problems. Existing research does not systematically explore key factors such as the optimal augmentation scale, the necessity of using large augmentation models, and whether diverse augmentations improve generalization, particularly in out-of-distribution (OOD) settings. This work presents a comprehensive study of the effectiveness of LLM augmentation for retrieval, comprising over 100 distinct experimental settings of retrieval models, augmentation models and augmentation strategies. We find that, while augmentation enhances retrieval performance, its benefits diminish beyond a certain augmentation scale, even with diverse augmentation strategies. Surprisingly, we observe that augmentation with smaller LLMs can achieve performance competitive with larger augmentation models. Moreover, we examine how augmentation effectiveness varies with retrieval model pre-training, revealing that augmentation provides the most benefit to models which are not well pre-trained. Our insights pave the way for more judicious and efficient augmentation strategies, thus enabling informed decisions and maximizing retrieval performance while being more cost-effective. Code and augmented datasets accompanying this work are publicly available at https://aka.ms/DAGR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。