CLAIOct 2, 2023

Making Retrieval-Augmented Language Models Robust to Irrelevant Context

DeepMind
arXiv:2310.01558v2374 citationsh-index: 59
Originality Incremental advance
AI Analysis

This addresses a critical reliability issue for RALMs in applications like question answering, though it is incremental as it builds on existing retrieval-augmentation frameworks.

The paper tackled the problem of retrieval-augmented language models (RALMs) suffering performance drops when given irrelevant retrieved contexts, especially in multi-hop reasoning, by analyzing five open-domain QA benchmarks and proposing methods to mitigate this. They showed that fine-tuning with 1,000 examples can train models to be robust to irrelevant contexts while maintaining high performance with relevant ones.

Retrieval-augmented language models (RALMs) hold promise to produce language understanding systems that are are factual, efficient, and up-to-date. An important desideratum of RALMs, is that retrieved information helps model performance when it is relevant, and does not harm performance when it is not. This is particularly important in multi-hop reasoning scenarios, where misuse of irrelevant evidence can lead to cascading errors. However, recent work has shown that retrieval augmentation can sometimes have a negative effect on performance. In this work, we present a thorough analysis on five open-domain question answering benchmarks, characterizing cases when retrieval reduces accuracy. We then propose two methods to mitigate this issue. First, a simple baseline that filters out retrieved passages that do not entail question-answer pairs according to a natural language inference (NLI) model. This is effective in preventing performance reduction, but at a cost of also discarding relevant passages. Thus, we propose a method for automatically generating data to fine-tune the language model to properly leverage retrieved passages, using a mix of relevant and irrelevant contexts at training time. We empirically show that even 1,000 examples suffice to train the model to be robust to irrelevant contexts while maintaining high performance on examples with relevant ones.

Code Implementations2 repos
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes