arXiv:2409.17745cs.IRcs.CL2024-09EMNLP被引 16

用少量示例提升大模型排序能力,无需复杂训练即可接近监督模型效果。

Few-shot Prompting for Pairwise Ranking: An Effective Non-Parametric Retrieval Model

  • 基于相似查询的少量示例增强,实现非参数化排序。
  • 在TREC DL和BEIR上均显著优于零样本基线,接近监督模型性能。
  • 适合追求简单高效且需跨域泛化的检索场景使用。

监督排序模型虽有效,但通常需多阶段任务特定预训练与微调,流程复杂。为简化流程,研究者尝试利用大语言模型实现零样本推理。然而,零样本方法因缺乏查询-文档配对训练数据,性能普遍低于监督模型。受现有发现启发——训练样本能提升零样本表现,本文探索其在排序任务中的适用性。具体而言,给定一个查询及一对文档,通过从训练集中引入相似查询的偏好示例,增强偏好预测能力。所提成对少样本排序器在域内(TREC DL)和域外(BEIR子集)检索基准上均持续优于零样本基线,且性能接近监督模型,无需复杂训练流程。

原文摘要 · Abstract (English)

A supervised ranking model, despite its advantage of being effective, usually involves complex processing - typically multiple stages of task-specific pre-training and fine-tuning. This has motivated researchers to explore simpler pipelines leveraging large language models (LLMs) that are capable of working in a zero-shot manner. However, since zero-shot inference does not make use of a training set of pairs of queries and their relevant documents, its performance is mostly worse than that of supervised models, which are trained on such example pairs. Motivated by the existing findings that training examples generally improve zero-shot performance, in our work, we explore if this also applies to ranking models. More specifically, given a query and a pair of documents, the preference prediction task is improved by augmenting examples of preferences for similar queries from a training set. Our proposed pairwise few-shot ranker demonstrates consistent improvements over the zero-shot baseline on both in-domain (TREC DL) and out-domain (BEIR subset) retrieval benchmarks. Our method also achieves a close performance to that of a supervised model without requiring any complex training pipeline.

少样本学习排序模型大模型检索

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