arXiv:2409.16497cs.AI2024-09中稿 · DCAI24 workshop@CI…被引 3

用指令微调大模型生成合成查询,无监督提升文本检索效果

Unsupervised Text Representation Learning via Instruction-Tuning for Zero-Shot Dense Retrieval

  • 通过指令微调语言模型生成相关合成查询,增强文档表征
  • 在多语言数据集上零样本检索性能显著提升,最高增益9.52%
  • 无需标注数据,模型更小但效果超越多个主流检索器

密集检索系统广泛用于信息检索,通常依赖带标签数据进行监督建模,但这类数据获取成本高或不可得。本文提出一种新型无监督文本表示学习方法:在双编码器框架下,对预训练的编码器-解码器大语言模型(LLM)进行指令微调。基于Rao-Blackwell定理,我们利用指令微调后的LLM生成相关合成查询,以增强语料库的表示能力。进一步通过自指令微调实现查询与语料库表示的一致性对齐。具体流程为:先用开源预训练LLM根据指定指令(如问题生成、关键词摘要)生成合成查询;再对通过质量筛选的生成查询进行指令微调;最后使用微调后的模型为每个语料项生成合成查询,通过加权平均合成查询与原始语料嵌入来表示语料。我们在三个英文和一个德文检索数据集上评估该方法,在低资源设置下使用NDCG@10、MRR@100、Recall@100作为指标,结果表明所有指标均有显著提升,相较于开放源码FLAN-T5模型变体绝对提升[3.34%, 3.50%],且在模型规模至少减少38%的前提下,超越mDPR、T-Systems、mBART-Large三个主流密集检索器,在NDCG@10上分别提升1.96%、4.62%、9.52%。

原文摘要 · Abstract (English)

Dense retrieval systems are commonly used for information retrieval (IR). They rely on learning text representations through an encoder and usually require supervised modeling via labelled data which can be costly to obtain or simply unavailable. In this study, we introduce a novel unsupervised text representation learning technique via instruction-tuning the pre-trained encoder-decoder large language models (LLM) under the dual-encoder retrieval framework. We demonstrate the corpus representation can be augmented by the representations of relevant synthetic queries generated by the instruct-tuned LLM founded on the Rao-Blackwell theorem. Furthermore, we effectively align the query and corpus text representation with self-instructed-tuning. Specifically, we first prompt an open-box pre-trained LLM to follow defined instructions (i.e. question generation and keyword summarization) to generate synthetic queries. Next, we fine-tune the pre-trained LLM with defined instructions and the generated queries that passed quality check. Finally, we generate synthetic queries with the instruction-tuned LLM for each corpora and represent each corpora by weighted averaging the synthetic queries and original corpora embeddings. We evaluate our proposed method under low-resource settings on three English and one German retrieval datasets measuring NDCG@10, MRR@100, Recall@100. We significantly improve the average zero-shot retrieval performance on all metrics, increasing open-box FLAN-T5 model variations by [3.34%, 3.50%] in absolute and exceeding three competitive dense retrievers (i.e. mDPR, T-Systems, mBART-Large), with model of size at least 38% smaller, by 1.96%, 4.62%, 9.52% absolute on NDCG@10.

无监督学习文本检索指令微调大模型应用

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