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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Farahani, Mehrdad, Johansson, Richard
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929530403291136
author Farahani, Mehrdad
Johansson, Richard
author_facet Farahani, Mehrdad
Johansson, Richard
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2410_05162
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Deciphering the Interplay of Parametric and Non-parametric Memory in Retrieval-augmented Language Models
Farahani, Mehrdad
Johansson, Richard
Computation and Language
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.
title Deciphering the Interplay of Parametric and Non-parametric Memory in Retrieval-augmented Language Models
topic Computation and Language
url https://arxiv.org/abs/2410.05162