用大模型解决非相似文档的推理型检索难题
Large Language Model Can Be a Foundation for Hidden Rationale-Based Retrieval
- 将检索任务转为生成式二选一问答,利用指令微调
- 在情感支持对话上零样本和微调性能均超越现有方法
- 适合需要逻辑推理的复杂检索场景
尽管检索增强生成(RAG)系统取得进展,但多数检索方法针对事实型检索,假设查询与正样本文档语义相近。本文提出并研究更具挑战性的隐藏推理型检索任务:查询与文档不相似,但可通过推理链、逻辑关系或经验推断关联。为此,我们采用指令微调的大语言模型(LLM)结合交叉编码器架构,设计特殊提示将检索转化为生成式二选一问题,并通过直接偏好优化(DPO)进行微调。该框架在保持计算效率的同时无性能损失。我们命名为RaHoRe,在情感支持对话(ESC)数据集上验证了其零样本及微调性能优于先前工作。研究证明大模型可作为更广泛检索任务的基础。代码、模型与数据集已开源。
原文摘要 · Abstract (English)
Despite the recent advancement in Retrieval-Augmented Generation (RAG) systems, most retrieval methodologies are often developed for factual retrieval, which assumes query and positive documents are semantically similar. In this paper, we instead propose and study a more challenging type of retrieval task, called hidden rationale retrieval, in which query and document are not similar but can be inferred by reasoning chains, logic relationships, or empirical experiences. To address such problems, an instruction-tuned Large language model (LLM) with a cross-encoder architecture could be a reasonable choice. To further strengthen pioneering LLM-based retrievers, we design a special instruction that transforms the retrieval task into a generative task by prompting LLM to answer a binary-choice question. The model can be fine-tuned with direct preference optimization (DPO). The framework is also optimized for computational efficiency with no performance degradation. We name this retrieval framework by RaHoRe and verify its zero-shot and fine-tuned performance superiority on Emotional Support Conversation (ESC), compared with previous retrieval works. Our study suggests the potential to employ LLM as a foundation for a wider scope of retrieval tasks. Our codes, models, and datasets are available on https://github.com/flyfree5/LaHoRe.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。