让检索和生成用不同知识表示,提升大模型效率与效果
HeteRAG: A Heterogeneous Retrieval-augmented Generation Framework with Decoupled Knowledge Representations
- 检索用带上下文的长片段,生成用精简短片段
- 在多个数据集上显著优于基线方法
- 适合需要高效精准生成的问答与写作场景
检索增强生成(RAG)通过将检索到的知识块融入生成过程来提升大模型性能。通常,检索和生成对知识块的要求不同:检索需完整信息以提高准确性,而过长的块会引入冗余上下文,降低生成效率与效果。现有RAG方法对两阶段使用相同的知识表示,导致性能受限。本文提出异构检索增强生成框架HeteRAG,解耦知识块在检索与生成中的表示:生成阶段采用短知识块,检索阶段则结合多粒度上下文信息的完整块,并引入自适应提示调优方法优化检索模型。大量实验表明,该方法在多个基准上均取得显著提升。
原文摘要 · Abstract (English)
Retrieval-augmented generation (RAG) methods can enhance the performance of LLMs by incorporating retrieved knowledge chunks into the generation process. In general, the retrieval and generation steps usually have different requirements for these knowledge chunks. The retrieval step benefits from comprehensive information to improve retrieval accuracy, whereas excessively long chunks may introduce redundant contextual information, thereby diminishing both the effectiveness and efficiency of the generation process. However, existing RAG methods typically employ identical representations of knowledge chunks for both retrieval and generation, resulting in suboptimal performance. In this paper, we propose a heterogeneous RAG framework (\myname) that decouples the representations of knowledge chunks for retrieval and generation, thereby enhancing the LLMs in both effectiveness and efficiency. Specifically, we utilize short chunks to represent knowledge to adapt the generation step and utilize the corresponding chunk with its contextual information from multi-granular views to enhance retrieval accuracy. We further introduce an adaptive prompt tuning method for the retrieval model to adapt the heterogeneous retrieval augmented generation process. Extensive experiments demonstrate that \myname achieves significant improvements compared to baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。