Making Retrieval-Augmented Language Models Robust to Irrelevant Context

By Ori Yoran, Tomer Wolfson, Ori Ram, Jonathan Berant. International Conference on Learning Representations, 2023

405 citations recorded by the source corpus.

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.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

View Paper: https://www.semanticscholar.org/paper/3aee33831e0bdea1a1eaae21c7586e4f7c0396d6

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Akari Asai
  2. Hannaneh Hajishirzi
  3. Yizhong Wang
  4. Zeqiu Wu
  5. Avirup Sil
  6. Haofen Wang
  7. Luke S. Zettlemoyer
  8. Jonathan Berant

In Computer Vision and Pattern Recognition

  1. Hannaneh Hajishirzi
  2. Akari Asai
  3. Yizhong Wang
  4. Avirup Sil
  5. Zeqiu Wu
  6. Jonathan Berant
  7. Daniel Khashabi
  8. Danqi Chen

Citations in the corpus