paper-with-me

홈 › Papers

Deciphering the Interplay of Parametric and Non-parametric Memory in Retrieval-augmented Language Models

2024-10-07 · Mehrdad Farahani, Richard Johansson

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.

📄 PDF Abstract BibTeX arXiv:2410.05162

Code (1)

m3hrdadfi/rag-memory-interplay 공식 구현 pytorch

Tasks

RAGRetrievalRetrieval-augmented Generation

Methods 이 논문이 사용한 방법론

Refunds@Expedia|||How do I get a full refund from Expedia? “How do I get a full refund from Expedia? How do I get a full refund from Expedia? – Call ☎️ +1-(888) 829 (0881) or +1-805-330-4056 or +1-805-330-4056 for Quick Help &…
Attention 설명 없음
WordPiece 설명 없음
Attention Dropout Attention Dropout is a type of dropout used in attention-based architectures, where elements are randomly dropped out of the…
Linear Layer A Linear Layer is a projection $\mathbf{XW + b}$.
Weight Decay 설명 없음
Linear Warmup With Linear Decay Linear Warmup With Linear Decay is a learning rate schedule in which we increase the learning rate linearly for $n$ updates and then linearly decay afterwards.
Dropout Dropout is a regularization technique for neural networks that drops a unit (along with connections) at training time with a specified probability $p$ (a common value is…

Similar Papers 제목 키워드 기반

Rethinking Memory in AI: Taxonomy, Operations, Topics, and Future Directions

2025-05-01 · Yiming Du, WenYu Huang, Danna Zheng, Zhaowei Wang 외

Memory is a fundamental component of AI systems, underpinning large language models (LLMs) based agents. While prior surveys have focused on memory applications with LLMs, they often overlook the atomic operations that u…

Survey

Nonparametric Decoding for Generative Retrieval

2022-10-05 · Hyunji Lee, Jaeyoung Kim, Hoyeon Chang, Hanseok Oh 외

The generative retrieval model depends solely on the information encoded in its model parameters without external memory, its information capacity is limited and fixed. To overcome the limitation, we propose Nonparametri…

DecoderLanguage ModellingRetrievalText Retrieval

Retro-li: Small-Scale Retrieval Augmented Generation Supporting Noisy Similarity Searches and Domain Shift Generalization

2024-09-12 · Gentiana Rashiti, Geethan Karunaratne, Mrinmaya Sachan, Abu Sebastian 외

The retrieval augmented generation (RAG) system such as Retro has been shown to improve language modeling capabilities and reduce toxicity and hallucinations by retrieving from a database of non-parametric memory contain…

Language ModelingLanguage ModellingRAGRetrieval+3

MLP Memory: A Retriever-Pretrained Memory for Large Language Models

2025-08-03 · Rubin Wei, Jiaqi Cao, Jiarui Wang, Jushi Kai 외 arxiv

Modern approaches to enhancing Large Language Models' factual accuracy and knowledge utilization face a fundamental trade-off: non-parametric retrieval-augmented generation (RAG) provides flexible access to external know…

Taming Knowledge Conflicts in Language Models

2025-03-14 · Gaotang Li, Yuzhong Chen, Hanghang Tong

Language Models (LMs) often encounter knowledge conflicts when parametric memory contradicts contextual knowledge. Previous works attribute this conflict to the interplay between "memory heads" and "context heads", atten…

Attribute