Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

When Not to Trust Language Models: Investigating Effectiveness of Parametric and Non-Parametric Memories

About

Despite their impressive performance on diverse tasks, large language models (LMs) still struggle with tasks requiring rich world knowledge, implying the limitations of relying solely on their parameters to encode a wealth of world knowledge. This paper aims to understand LMs' strengths and limitations in memorizing factual knowledge, by conducting large-scale knowledge probing experiments of 10 models and 4 augmentation methods on PopQA, our new open-domain QA dataset with 14k questions. We find that LMs struggle with less popular factual knowledge, and that scaling fails to appreciably improve memorization of factual knowledge in the long tail. We then show that retrieval-augmented LMs largely outperform orders of magnitude larger LMs, while unassisted LMs remain competitive in questions about high-popularity entities. Based on those findings, we devise a simple, yet effective, method for powerful and efficient retrieval-augmented LMs, which retrieves non-parametric memories only when necessary. Experimental results show that this significantly improves models' performance while reducing the inference costs.

Alex Mallen, Akari Asai, Victor Zhong, Rajarshi Das, Daniel Khashabi, Hannaneh Hajishirzi• 2022

Related benchmarks

TaskDatasetResultRank
Question AnsweringPopQA
Accuracy43.3
186
Factuality DetectionShort-form QA (Average of NQ, PopQA, TriviaQA, SimpleQA) (test)
PR-AUC55.6
60
Long-form QA Factuality DetectionLong-form QA benchmark factuality target
PR-AUC11.2
48
Generative Commonsense ReasoningCommonGen V1.0 (test)
BLEU-426.68
16
Question AnsweringQASC
Spearman's rho0.2727
12
Question AnsweringMuSiQue
Spearman's rho0.2818
12
Question AnsweringNQ
Spearman's rho0.2636
12
Question AnsweringTriviaQA
Spearman's rho0.1727
12
Question AnsweringMuSiQue
Cohen's d-0.0275
12
Question AnsweringNQ
Cohen's d0.1535
12
Showing 10 of 16 rows

Other info

Code

Follow for update