揭秘大模型如何权衡自有知识与检索信息的优先级
Deciphering the Interplay of Parametric and Non-parametric Memory in Retrieval-augmented Language Models
- 通过因果中介分析,揭示模型在生成时对内置知识和检索内容的选择机制
- 当两种信息可选时,模型更依赖检索到的上下文而非自身参数知识
- 发现模型会先判断上下文相关性,再决定是否复制检索内容
生成式语言模型常难以处理专业或冷门知识。一种解决方案是检索增强生成(RAG)模型,其在生成前先检索信息。本研究以Atlas方法为例,探究模型在已有知识(参数化)与检索内容(非参数化)之间如何抉择。通过因果中介分析与受控实验,我们解耦了参数知识与检索上下文的影响。结果表明:当模型可选择两者时,更倾向于依赖检索上下文而非自身参数知识。进一步分析显示,模型内部存在多重计算机制:一是判断上下文是否相关,二是编码器如何生成支持复制的输出表示。
原文摘要 · Abstract (English)
Generative language models often struggle with specialized or less-discussed knowledge. A potential solution is found in Retrieval-Augmented Generation (RAG) models which act like retrieving information before generating responses. In this study, we explore how the \textsc{Atlas} approach, a RAG model, decides between what it already knows (parametric) and what it retrieves (non-parametric). We use causal mediation analysis and controlled experiments to examine how internal representations influence information processing. Our findings disentangle the effects of parametric knowledge and the retrieved context. They indicate that in cases where the model can choose between both types of information (parametric and non-parametric), it relies more on the context than the parametric knowledge. Furthermore, the analysis investigates the computations involved in \emph{how} the model uses the information from the context. We find that multiple mechanisms are active within the model and can be detected with mediation analysis: first, the decision of \emph{whether the context is relevant}, and second, how the encoder computes output representations to support copying when relevant.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。