arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.04522cs.CLcs.AIcs.IRcs.LG
收藏

Q-PEFT: 面向大语言模型文本重排序的查询依赖参数高效微调

Q-PEFT: Query-dependent Parameter Efficient Fine-tuning for Text Reranking with Large Language Models

Santa Clara University(圣克拉拉大学);Meta AI;Walmart Global Tech(沃尔玛全球科技)

Zhiyuan Peng, Xuyang Wu, Qifan Wang, Sravanthi Rajanala, Yi Fang

AI总结:

针对现有PEFT用于文本重排序时提示固定、过拟合、适配性差的问题,提出查询依赖的Q-PEFT方法,通过查询提取文档关键token并引入多头注意力增强,在四个公开数据集上验证了重排序性能提升效果。

摘要:

参数高效微调(Parameter Efficient Fine-Tuning, PEFT)方法已被广泛应用于大语言模型(Large Language Models, LLMs)中,无需微调整个LLM即可提升下游任务性能。现有研究已证明PEFT可有效用于排序任务的LLM微调并取得了令人信服的效果,但仍存在一些局限,包括针对不同文档学习到的提示固定、对特定任务过拟合以及适配能力较低等问题。本文提出一种面向文本重排序的查询依赖参数高效微调(Q-PEFT)方法,将真实查询的信息注入LLM,从而使LLM从输入文档生成真实查询的过程更简单。具体而言,我们利用查询从拼接后的文档中提取前k个token作为上下文线索。我们进一步用多头注意力层替代检索机制来增强Q-PEFT,实现端到端训练并覆盖文档中的所有token,引导LLM生成更具文档特异性的合成查询,从而进一步提升重排序性能。我们在四个公开数据集上开展了大量实验,验证了所提方法的有效性。

英文摘要:

Parameter Efficient Fine-Tuning (PEFT) methods have been extensively utilized in Large Language Models (LLMs) to improve the down-streaming tasks without the cost of fine-tuing the whole LLMs. Recent studies have shown how to effectively use PEFT for fine-tuning LLMs in ranking tasks with convincing performance; there are some limitations, including the learned prompt being fixed for different documents, overfitting to specific tasks, and low adaptation ability. In this paper, we introduce a query-dependent parameter efficient fine-tuning (Q-PEFT) approach for text reranking to leak the information of the true queries to LLMs and then make the generation of true queries from input documents much easier. Specifically, we utilize the query to extract the top-$k$ tokens from concatenated documents, serving as contextual clues. We further augment Q-PEFT by substituting the retrieval mechanism with a multi-head attention layer to achieve end-to-end training and cover all the tokens in the documents, guiding the LLMs to generate more document-specific synthetic queries, thereby further improving the reranking performance. Extensive experiments are conducted on four public datasets, demonstrating the effectiveness of our proposed approach.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑