为上下文学习设计新评估框架,用推荐思维提升检索效果。
ICLERB: In-Context Learning Embedding and Reranker Benchmark
- 将ICL检索视为推荐问题,优先选择对任务最有用的文档。
- 新基准ICLERB显示,小模型经AI反馈微调后超越大模型性能。
- 适合关注LLM检索优化与高效训练策略的研究者。
上下文学习(ICL)使大语言模型通过提示词中的相关信息执行新任务。检索增强生成(RAG)通过在查询时将检索到的文档纳入语言模型上下文来增强ICL。然而,传统检索方法仅关注语义相关性,将检索视为搜索问题。本文提出将ICL的检索重新定义为推荐问题,目标是选择能最大化任务效用的文档。我们引入了上下文学习嵌入与重排序基准(ICLERB),一个新颖的评估框架,用于比较不同检索器在提升语言模型在ICL任务中准确率方面的表现。此外,我们提出一种基于人工智能反馈的强化学习排序算法(RLRAIF),旨在使用少量来自语言模型的反馈微调检索模型。实验结果揭示了ICLERB与现有基准之间的显著差异,并表明经过该算法微调的小模型性能优于大型先进检索模型。这些发现突显了现有评估方法的局限性,强调了针对ICL专用的基准和训练策略的必要性。
原文摘要 · Abstract (English)
In-Context Learning (ICL) enables Large Language Models (LLMs) to perform new tasks by conditioning on prompts with relevant information. Retrieval-Augmented Generation (RAG) enhances ICL by incorporating retrieved documents into the LLM's context at query time. However, traditional retrieval methods focus on semantic relevance, treating retrieval as a search problem. In this paper, we propose reframing retrieval for ICL as a recommendation problem, aiming to select documents that maximize utility in ICL tasks. We introduce the In-Context Learning Embedding and Reranker Benchmark (ICLERB), a novel evaluation framework that compares retrievers based on their ability to enhance LLM accuracy in ICL settings. Additionally, we propose a novel Reinforcement Learning-to-Rank from AI Feedback (RLRAIF) algorithm, designed to fine-tune retrieval models using minimal feedback from the LLM. Our experimental results reveal notable differences between ICLERB and existing benchmarks, and demonstrate that small models fine-tuned with our RLRAIF algorithm outperform large state-of-the-art retrieval models. These findings highlight the limitations of existing evaluation methods and the need for specialized benchmarks and training strategies adapted to ICL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。