Fine-tune the Entire RAG Architecture (including DPR retriever) for Question-Answering

22 Jun 2021  ·  Shamane Siriwardhana, Rivindu Weerasekera, Elliott Wen, Suranga Nanayakkara ·

In this paper, we illustrate how to fine-tune the entire Retrieval Augment Generation (RAG) architecture in an end-to-end manner. We highlighted the main engineering challenges that needed to be addressed to achieve this objective. We also compare how end-to-end RAG architecture outperforms the original RAG architecture for the task of question answering. We have open-sourced our implementation in the HuggingFace Transformers library.

PDF Abstract

Datasets


Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Question Answering SQuAD RAG-end2end Exact Match 40.02 # 2

Methods